mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2026-10-06 20:29:59 +08:00
性能优化 (#2126)
* 性能优化 * 代码改进 * 优化浏览器代际切换逻辑 * 统一缓存与生命周期 * 添加aiomysql依赖 * 优化插件路径处理逻辑,简化条件判断;在虚拟环境包管理器中添加编码和错误处理参数以增强稳定性 * 🚨 auto fix by pre-commit hooks * 优化Windows下的关闭逻辑 * 代码优化 * bugfix:修复配置重载问题 * bugfix:修复插件加载启动竞态问题 * 收敛事件入口和权限上下文 * 优化 Windows launcher 关闭重启兜底 --------- Co-authored-by: HibiKier <775757368@qq.com> Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
This commit is contained in:
co-authored by
HibiKier
pre-commit-ci[bot]
parent
24c316cd2c
commit
5d92ccd3b0
@@ -5,12 +5,11 @@ from tortoise import fields
|
||||
from tortoise.exceptions import IntegrityError
|
||||
|
||||
from zhenxun.models.goods_info import GoodsInfo
|
||||
from zhenxun.services.buffered_writers import append_user_gold_log
|
||||
from zhenxun.services.db_context import Model
|
||||
from zhenxun.utils.enum import CacheType, GoldHandle
|
||||
from zhenxun.utils.exception import GoodsNotFound, InsufficientGold
|
||||
|
||||
from .user_gold_log import UserGoldLog
|
||||
|
||||
|
||||
class UserConsole(Model):
|
||||
id = fields.IntField(pk=True, generated=True, auto_increment=True)
|
||||
@@ -77,6 +76,17 @@ class UserConsole(Model):
|
||||
user, _ = await cls.get_or_create_user(user_id=user_id, platform=platform)
|
||||
return user
|
||||
|
||||
@classmethod
|
||||
async def _get_user_for_write(
|
||||
cls, user_id: str, platform: str | None = None
|
||||
) -> "UserConsole":
|
||||
"""获取写入用用户;已有用户不走 get_or_create,避免重复清理缓存。"""
|
||||
user = await cls.get_or_none(user_id=user_id)
|
||||
if user is not None:
|
||||
return user
|
||||
user, _ = await cls.get_or_create_user(user_id=user_id, platform=platform)
|
||||
return user
|
||||
|
||||
@classmethod
|
||||
async def get_new_uid(cls) -> int:
|
||||
"""获取最新uid
|
||||
@@ -103,10 +113,10 @@ class UserConsole(Model):
|
||||
source: 来源
|
||||
platform: 平台.
|
||||
"""
|
||||
user, _ = await cls.get_or_create_user(user_id=user_id, platform=platform)
|
||||
user = await cls._get_user_for_write(user_id=user_id, platform=platform)
|
||||
user.gold += gold
|
||||
await user.save(update_fields=["gold"])
|
||||
await UserGoldLog.create(
|
||||
await append_user_gold_log(
|
||||
user_id=user_id, gold=gold, handle=GoldHandle.GET, source=source
|
||||
)
|
||||
|
||||
@@ -131,12 +141,12 @@ class UserConsole(Model):
|
||||
异常:
|
||||
InsufficientGold: 金币不足
|
||||
"""
|
||||
user, _ = await cls.get_or_create_user(user_id=user_id, platform=platform)
|
||||
user = await cls._get_user_for_write(user_id=user_id, platform=platform)
|
||||
if user.gold < gold:
|
||||
raise InsufficientGold()
|
||||
user.gold -= gold
|
||||
await user.save(update_fields=["gold"])
|
||||
await UserGoldLog.create(
|
||||
await append_user_gold_log(
|
||||
user_id=user_id, gold=gold, handle=handle, source=plugin_module
|
||||
)
|
||||
|
||||
@@ -152,7 +162,7 @@ class UserConsole(Model):
|
||||
num: 道具数量.
|
||||
platform: 平台.
|
||||
"""
|
||||
user, _ = await cls.get_or_create_user(user_id=user_id, platform=platform)
|
||||
user = await cls._get_user_for_write(user_id=user_id, platform=platform)
|
||||
if goods_uuid not in user.props:
|
||||
user.props[goods_uuid] = 0
|
||||
user.props[goods_uuid] += num
|
||||
@@ -186,7 +196,7 @@ class UserConsole(Model):
|
||||
num: 道具数量.
|
||||
platform: 平台.
|
||||
"""
|
||||
user, _ = await cls.get_or_create_user(user_id=user_id, platform=platform)
|
||||
user = await cls._get_user_for_write(user_id=user_id, platform=platform)
|
||||
|
||||
if goods_uuid not in user.props or user.props[goods_uuid] < num:
|
||||
raise GoodsNotFound("未找到商品或道具数量不足...")
|
||||
|
||||
Reference in New Issue
Block a user