mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2026-10-17 02:00:01 +08:00
🔧 优化数据访问和数据库上下文逻辑,移除不必要的全局变量和日志信息,调整日志级别为调试,提升代码可读性和性能。
This commit is contained in:
@@ -1,7 +1,6 @@
|
||||
import asyncio
|
||||
import time
|
||||
|
||||
from zhenxun.services.cache import CacheRoot
|
||||
from zhenxun.services.log import logger
|
||||
|
||||
from .config import (
|
||||
@@ -10,13 +9,6 @@ from .config import (
|
||||
SLOW_QUERY_THRESHOLD,
|
||||
)
|
||||
|
||||
cache = CacheRoot.cache_dict("DB_TEST", 10)
|
||||
|
||||
index = 0
|
||||
|
||||
|
||||
total = {}
|
||||
|
||||
|
||||
async def with_db_timeout(
|
||||
coro,
|
||||
@@ -25,20 +17,10 @@ async def with_db_timeout(
|
||||
source: str | None = None,
|
||||
):
|
||||
"""带超时控制的数据库操作"""
|
||||
global index, total, index2
|
||||
start_time = time.time()
|
||||
try:
|
||||
if operation not in total:
|
||||
total[operation] = CacheRoot.cache_dict(f"DB_TEST_{operation}", 10)
|
||||
total[operation][str(index)] = "1"
|
||||
index += 1
|
||||
logger.info(f"当前数据库查询来源: {source}")
|
||||
logger.info(
|
||||
f"10秒内数据库查询次数 [{operation}]: {len(total[operation])}", "DB_TEST"
|
||||
)
|
||||
logger.debug(f"开始执行数据库操作: {operation} 来源: {source}")
|
||||
result = await asyncio.wait_for(coro, timeout=timeout)
|
||||
cache[str(index)] = "1"
|
||||
index += 1
|
||||
elapsed = time.time() - start_time
|
||||
if elapsed > SLOW_QUERY_THRESHOLD and operation:
|
||||
logger.warning(f"慢查询: {operation} 耗时 {elapsed:.3f}s", LOG_COMMAND)
|
||||
|
||||
Reference in New Issue
Block a user