mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2025-12-14 21:52:56 +08:00
e7f3c210df
* 🔧 修复和优化：调整超时设置，重构检查逻辑，简化代码结构 - 在 `chkdsk_hook.py` 中重构 `check` 方法，提取公共逻辑 - 更新 `CacheManager` 中的超时设置，使用新的 `CACHE_TIMEOUT` - 在 `utils.py` 中添加缓存逻辑，记录数据库操作的执行情况 * ✨ feat(auth): 添加并发控制，优化权限检查逻辑 * Update utils.py * 🚨 auto fix by pre-commit hooks --------- Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
35 lines
971 B
Python
35 lines
971 B
Python
import asyncio
|
|
import time
|
|
|
|
from zhenxun.services.log import logger
|
|
|
|
from .config import (
|
|
DB_TIMEOUT_SECONDS,
|
|
LOG_COMMAND,
|
|
SLOW_QUERY_THRESHOLD,
|
|
)
|
|
|
|
|
|
async def with_db_timeout(
|
|
coro,
|
|
timeout: float = DB_TIMEOUT_SECONDS,
|
|
operation: str | None = None,
|
|
source: str | None = None,
|
|
):
|
|
"""带超时控制的数据库操作"""
|
|
start_time = time.time()
|
|
try:
|
|
logger.debug(f"开始执行数据库操作: {operation} 来源: {source}")
|
|
result = await asyncio.wait_for(coro, timeout=timeout)
|
|
elapsed = time.time() - start_time
|
|
if elapsed > SLOW_QUERY_THRESHOLD and operation:
|
|
logger.warning(f"慢查询: {operation} 耗时 {elapsed:.3f}s", LOG_COMMAND)
|
|
return result
|
|
except asyncio.TimeoutError:
|
|
if operation:
|
|
logger.error(
|
|
f"数据库操作超时: {operation} (>{timeout}s) 来源: {source}",
|
|
LOG_COMMAND,
|
|
)
|
|
raise
|