konjacpotato
f93fda8efa
All checks were successful
Gitea Actions Demo / host-commands (push) Successful in 0s
45 lines
1.6 KiB
Python
45 lines
1.6 KiB
Python
import datetime
|
||
from functools import partial
|
||
from apscheduler.schedulers.blocking import BlockingScheduler
|
||
from apscheduler.events import EVENT_JOB_ERROR
|
||
from config import config, settings
|
||
from task import manager_task
|
||
from utils import logger, MailSender
|
||
|
||
|
||
def job_error_listener(event):
|
||
if event.exception:
|
||
logger.error(f"Job {event.job_id} crashed: {str(event.exception)}")
|
||
# 可添加邮件/钉钉告警逻辑
|
||
try:
|
||
mail_sender = MailSender()
|
||
mail_sender.execute(
|
||
to_addrs=settings.ERROR_NOTIFICATION_EMAIL,
|
||
subject=f"Job {event.job_id} crashed",
|
||
body=f"Job {event.job_id} crashed with error: {str(event.exception)}"
|
||
)
|
||
except Exception as e:
|
||
logger.error(f"Failed to send error notification email: {e}")
|
||
|
||
|
||
if __name__ == '__main__':
|
||
scheduler = BlockingScheduler()
|
||
|
||
# 每隔config.scheduler_interval秒执行一次任务，同时设定第一次执行在程序启动后10秒后执行
|
||
scheduler.add_job(
|
||
partial(manager_task, scheduler),
|
||
'interval',
|
||
seconds=config.scheduler_interval,
|
||
jitter=30, # 添加随机抖动避免任务雪崩
|
||
next_run_time=datetime.datetime.now() + datetime.timedelta(seconds=10) # 替代 date 触发器
|
||
)
|
||
|
||
# 添加任务错误监听器
|
||
scheduler.add_listener(job_error_listener, EVENT_JOB_ERROR)
|
||
|
||
try:
|
||
logger.info(f"started successfully.")
|
||
scheduler.start() # 阻塞运行
|
||
except (KeyboardInterrupt, SystemExit):
|
||
logger.info(f"Shutting down ...")
|