gardel/Mofox-Core
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refactor(db): 将数据库操作异步化

Browse Source 
- 将所有 session.add() 改为 await session.add()
- 将所有 session.commit() 改为 await session.commit()
- 将 session.refresh() 改为 await session.refresh()
This commit is contained in:
雅诺狐
2025-09-20 14:35:31 +08:00
parent 816ce9805c
commit 55717669dd
20 changed files with 44 additions and 47 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/plugins/built_in/maizone_refactored/services/scheduler_service.py
View File

@@ -186,8 +186,8 @@ class SchedulerService:
story_content=content,
send_success=success,
)
session.add(new_record)
session.commit()
await session.add(new_record)
await session.commit()
logger.info(f"已更新日程处理状态: {hour_str} - {activity} - 成功: {success}")
except Exception as e:
logger.error(f"更新日程处理状态时发生数据库错误: {e}")

8
src/plugins/built_in/napcat_adapter_plugin/src/database.py
View File

@@ -83,14 +83,14 @@ class DatabaseManager:
continue
# 更新现有记录的 lift_time
existing_record.lift_time = ban_user.lift_time
session.add(existing_record)
await session.add(existing_record)
logger.debug(f"更新禁言记录: {existing_record}")
else:
# 创建新记录
db_record = DB_BanUser(
user_id=ban_user.user_id, group_id=ban_user.group_id, lift_time=ban_user.lift_time
)
session.add(db_record)
await session.add(db_record)
logger.debug(f"创建新禁言记录: {ban_user}")
# 删除不在 ban_list 中的记录
for db_record in all_records:
@@ -132,14 +132,14 @@ class DatabaseManager:
if existing_record:
# 如果记录已存在，更新 lift_time
existing_record.lift_time = ban_record.lift_time
session.add(existing_record)
await session.add(existing_record)
logger.debug(f"更新禁言记录: {ban_record}")
else:
# 如果记录不存在，创建新记录
db_record = DB_BanUser(
user_id=ban_record.user_id, group_id=ban_record.group_id, lift_time=ban_record.lift_time
)
session.add(db_record)
await session.add(db_record)
logger.debug(f"创建新禁言记录: {ban_record}")
def delete_ban_record(self, ban_record: BanUser):