perf: support keyed create locks and use user_id lock for UserConsole

zhenxun/models/user_console.py

@@ -1,4 +1,3 @@
-import asyncio
 from typing import ClassVar
 
 from tortoise import fields
@@ -7,7 +6,7 @@ from tortoise.exceptions import IntegrityError
 from zhenxun.models.goods_info import GoodsInfo
 from zhenxun.services.cache import CacheRoot
 from zhenxun.services.db_context import Model
-from zhenxun.utils.enum import CacheType, GoldHandle
+from zhenxun.utils.enum import CacheType, DbLockType, GoldHandle
 from zhenxun.utils.exception import GoodsNotFound, InsufficientGold
 
 from .user_gold_log import UserGoldLog
@@ -40,32 +39,18 @@ class UserConsole(Model):
     """缓存类型"""
     cache_key_field = "user_id"
     """缓存键字段"""
-    _create_locks: ClassVar[dict[str, asyncio.Lock]] = {}
+    lock_fields: ClassVar[dict[DbLockType, str]] = {DbLockType.CREATE: "user_id"}
 
     @classmethod
     async def get_user(cls, user_id: str, platform: str | None = None) -> "UserConsole":
         if user := await cls.get_or_none(user_id=user_id):
             return user
 
-        lock = cls._create_locks.setdefault(user_id, asyncio.Lock())
-        async with lock:
-            # 双重检查，避免锁等待后重复创建
-            if user := await cls.get_or_none(user_id=user_id):
-                return user
-
-            for _ in range(3):
-                try:
-                    uid = await cls.get_new_uid()
-                    return await cls.create(
-                        user_id=user_id,
-                        uid=uid,
-                        platform=platform,
-                    )
-                except IntegrityError:
-                    # 可能是 uid 或 user_id 竞争，短暂等待再重试
-                    await asyncio.sleep(0.05)
-
-            # 多次重试仍失败，按 user_id 再查一遍兜底
+        try:
+            uid = await cls.get_new_uid()
+            return await cls.create(user_id=user_id, uid=uid, platform=platform)
+        except IntegrityError:
+            # 并发竞争下，依赖 user_id 唯一约束兜底查询
             return await cls.get(user_id=user_id)
 
     @classmethod