在我写的模块采用更安全的 asyncio 锁写法

2025-10-19 20:27:18 +08:00
parent fd4c9302c2
commit 67382a0c0a
2 changed files with 41 additions and 31 deletions
--- a/konabot/common/longtask.py
+++ b/konabot/common/longtask.py
@ -254,11 +254,10 @@ def _save_longtask_data(data: LongTaskModuleData):

@asynccontextmanager
 async def longtask_data():
-    await longtask_lock.acquire()
-    data = _load_longtask_data()
-    yield data
-    _save_longtask_data(data)
-    longtask_lock.release()
+    async with longtask_lock:
+        data = _load_longtask_data()
+        yield data
+        _save_longtask_data(data)


 async def create_longtask(