mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2025-12-15 14:22:55 +08:00
fcb385cf01
**架构重构** - 拆分为 Service、Repository、Adapter 三层架构，提升模块化 - 统一 APScheduler Job ID 生成方式，优化 ScheduleTargeter 逻辑 **新增功能** - 支持定时任务时区配置 - 新增"运行中"任务状态显示 - 为"所有群组"任务增加随机延迟，分散并发压力 **用户体验优化** - 重构操作反馈消息，提供详细的成功提示卡片 - 优化任务查看命令的筛选逻辑 - 统一删除、暂停、恢复、执行、更新操作的响应格式 Co-authored-by: webjoin111 <455457521@qq.com>
63 lines
2.4 KiB
Python
63 lines
2.4 KiB
Python
"""
|
|
定时任务的生命周期管理
|
|
|
|
包含在机器人启动时加载和调度数据库中保存的任务的逻辑。
|
|
"""
|
|
|
|
from zhenxun.services.log import logger
|
|
from zhenxun.utils.manager.priority_manager import PriorityLifecycle
|
|
|
|
from .adapter import APSchedulerAdapter
|
|
from .repository import ScheduleRepository
|
|
from .service import scheduler_manager
|
|
|
|
|
|
@PriorityLifecycle.on_startup(priority=90)
|
|
async def _load_schedules_from_db():
|
|
"""在服务启动时从数据库加载并调度所有任务。"""
|
|
logger.info("正在从数据库加载并调度所有定时任务...")
|
|
schedules = await ScheduleRepository.get_all_enabled()
|
|
count = 0
|
|
for schedule in schedules:
|
|
if schedule.plugin_name in scheduler_manager._registered_tasks:
|
|
APSchedulerAdapter.add_or_reschedule_job(schedule)
|
|
count += 1
|
|
else:
|
|
logger.warning(f"跳过加载定时任务：插件 '{schedule.plugin_name}' 未注册。")
|
|
logger.info(f"数据库定时任务加载完成，共成功加载 {count} 个任务。")
|
|
|
|
logger.info("正在检查并注册声明式默认任务...")
|
|
declared_count = 0
|
|
for task_info in scheduler_manager._declared_tasks:
|
|
plugin_name = task_info["plugin_name"]
|
|
group_id = task_info["group_id"]
|
|
bot_id = task_info["bot_id"]
|
|
|
|
query_kwargs = {
|
|
"plugin_name": plugin_name,
|
|
"group_id": group_id,
|
|
"bot_id": bot_id,
|
|
}
|
|
exists = await ScheduleRepository.exists(**query_kwargs)
|
|
|
|
if not exists:
|
|
logger.info(f"为插件 '{plugin_name}' 注册新的默认定时任务...")
|
|
schedule = await scheduler_manager.add_schedule(
|
|
plugin_name=plugin_name,
|
|
group_id=group_id,
|
|
trigger_type=task_info["trigger_type"],
|
|
trigger_config=task_info["trigger_config"],
|
|
job_kwargs=task_info["job_kwargs"],
|
|
bot_id=bot_id,
|
|
)
|
|
if schedule:
|
|
declared_count += 1
|
|
logger.debug(f"默认任务 '{plugin_name}' 注册成功 (ID: {schedule.id})")
|
|
else:
|
|
logger.error(f"默认任务 '{plugin_name}' 注册失败")
|
|
else:
|
|
logger.debug(f"插件 '{plugin_name}' 的默认任务已存在于数据库中，跳过注册。")
|
|
|
|
if declared_count > 0:
|
|
logger.info(f"声明式任务检查完成，新注册了 {declared_count} 个默认任务。")
|