🔧 修复和优化：调整超时设置，重构检查逻辑，简化代码结构

- 在 `chkdsk_hook.py` 中重构 `check` 方法，提取公共逻辑
- 更新 `CacheManager` 中的超时设置，使用新的 `CACHE_TIMEOUT`
- 在 `utils.py` 中添加缓存逻辑，记录数据库操作的执行情况

zhenxun/builtin_plugins/hooks/chkdsk_hook.py

@@ -43,17 +43,19 @@ class BanCheckLimiter:
 
     def check(self, key: str | float) -> bool:
         if time.time() - self.mtime[key] > self.default_check_time:
-            self.mtime[key] = time.time()
-            self.mint[key] = 0
-            return False
+            return self._extracted_from_check_3(key, False)
         if (
             self.mint[key] >= self.default_count
             and time.time() - self.mtime[key] < self.default_check_time
         ):
+            return self._extracted_from_check_3(key, True)
+        return False
+
+    # TODO Rename this here and in `check`
+    def _extracted_from_check_3(self, key, arg1):
         self.mtime[key] = time.time()
         self.mint[key] = 0
-            return True
-        return False
+        return arg1
 
 
 _blmt = BanCheckLimiter(
@@ -70,7 +72,8 @@ async def _(
     module = None
     if plugin := matcher.plugin:
         module = plugin.module_name
-        if metadata := plugin.metadata:
+        if not (metadata := plugin.metadata):
+            return
         extra = metadata.extra
         if extra.get("plugin_type") in [
             PluginType.HIDDEN,
@@ -79,8 +82,6 @@ async def _(
             PluginType.SUPERUSER,
         ]:
             return
-        else:
-            return
     if matcher.type == "notice":
         return
     user_id = session.id1
@@ -88,8 +89,7 @@ async def _(
     malicious_ban_time = Config.get_config("hook", "MALICIOUS_BAN_TIME")
     if not malicious_ban_time:
         raise ValueError("模块: [hook], 配置项: [MALICIOUS_BAN_TIME] 为空或小于0")
-    if user_id:
-        if module:
+    if user_id and module:
         if _blmt.check(f"{user_id}__{module}"):
             await BanConsole.ban(
                 user_id,

zhenxun/services/cache/__init__.py

@@ -98,6 +98,7 @@ from .cache_containers import CacheDict, CacheList
 from .config import (
     CACHE_KEY_PREFIX,
     CACHE_KEY_SEPARATOR,
+    CACHE_TIMEOUT,
     DEFAULT_EXPIRE,
     LOG_COMMAND,
     SPECIAL_KEY_FORMATS,
@@ -551,7 +552,6 @@ class CacheManager:
         返回:
             Any: 缓存数据，如果不存在返回默认值
         """
-        from zhenxun.services.db_context import DB_TIMEOUT_SECONDS
 
         # 如果缓存被禁用或缓存模式为NONE，直接返回默认值
         if not self.enabled or cache_config.cache_mode == CacheMode.NONE:
@@ -561,7 +561,7 @@ class CacheManager:
             cache_key = self._build_key(cache_type, key)
             data = await asyncio.wait_for(
                 self.cache_backend.get(cache_key),  # type: ignore
-                timeout=DB_TIMEOUT_SECONDS,
+                timeout=CACHE_TIMEOUT,
             )
 
             if data is None:

zhenxun/services/cache/config.py

@@ -5,6 +5,9 @@
 # 日志标识
 LOG_COMMAND = "CacheRoot"
 
+# 缓存获取超时时间（秒）
+CACHE_TIMEOUT = 10
+
 # 默认缓存过期时间（秒）
 DEFAULT_EXPIRE = 600
 

zhenxun/services/db_context/utils.py

@@ -1,14 +1,18 @@
 import asyncio
+import random
 import time
 
 from zhenxun.services.log import logger
 
+from ..cache import CacheRoot
 from .config import (
     DB_TIMEOUT_SECONDS,
     LOG_COMMAND,
     SLOW_QUERY_THRESHOLD,
 )
 
+aaa = CacheRoot.cache_dict("AAA", 10, int)
+
 
 async def with_db_timeout(
     coro,
@@ -21,11 +25,20 @@ async def with_db_timeout(
     try:
         logger.debug(f"开始执行数据库操作: {operation} 来源: {source}")
         result = await asyncio.wait_for(coro, timeout=timeout)
+        aaa[str(random.randint(0, 100000))] = 1
         elapsed = time.time() - start_time
         if elapsed > SLOW_QUERY_THRESHOLD and operation:
             logger.warning(f"慢查询: {operation} 耗时 {elapsed:.3f}s", LOG_COMMAND)
+        logger.debug(
+            f"10s内数据库查询次数: {len(aaa)}， 开始执行数据库操作: {operation} "
+            f"来源: {source}, 当次返回数据ID: {getattr(result, 'id', None)}",
+            LOG_COMMAND,
+        )
         return result
     except asyncio.TimeoutError:
         if operation:
-            logger.error(f"数据库操作超时: {operation} (>{timeout}s)", LOG_COMMAND)
+            logger.error(
+                f"数据库操作超时: {operation} (>{timeout}s) 来源: {source}",
+                LOG_COMMAND,
+            )
         raise