refactor(db): 修正SQLAlchemy异步操作调用方式
移除session.add()方法的不必要await调用,修正异步数据库操作模式。主要变更包括: - 将 `await session.add()` 统一改为 `session.add()` - 修正部分函数调用为异步版本(如消息查询函数) - 重构SQLAlchemyTransaction为完全异步实现 - 重写napcat_adapter_plugin数据库层以符合异步规范 - 添加aiomysql和aiosqlite依赖支持
This commit is contained in:
@@ -32,7 +32,7 @@ class AntiInjectionStatistics:
|
||||
stats = session.query(AntiInjectionStats).order_by(AntiInjectionStats.id.desc()).first()
|
||||
if not stats:
|
||||
stats = AntiInjectionStats()
|
||||
await session.add(stats)
|
||||
session.add(stats)
|
||||
await session.commit()
|
||||
await session.refresh(stats)
|
||||
return stats
|
||||
@@ -48,7 +48,7 @@ class AntiInjectionStatistics:
|
||||
stats = session.query(AntiInjectionStats).order_by(AntiInjectionStats.id.desc()).first()
|
||||
if not stats:
|
||||
stats = AntiInjectionStats()
|
||||
await session.add(stats)
|
||||
session.add(stats)
|
||||
|
||||
# 更新统计字段
|
||||
for key, value in kwargs.items():
|
||||
|
||||
@@ -85,7 +85,7 @@ class UserBanManager:
|
||||
reason=f"提示词注入攻击 (置信度: {detection_result.confidence:.2f})",
|
||||
created_at=datetime.datetime.now(),
|
||||
)
|
||||
await session.add(ban_record)
|
||||
session.add(ban_record)
|
||||
|
||||
await session.commit()
|
||||
|
||||
|
||||
Reference in New Issue
Block a user