58 lines
1.9 KiB
Python
58 lines
1.9 KiB
Python
from apscheduler.schedulers.background import BackgroundScheduler
|
|
from apscheduler.triggers.interval import IntervalTrigger
|
|
from utils.health_check import HealthChecker
|
|
from models import HealthCheckConfig, db
|
|
from app import create_app
|
|
import logging
|
|
|
|
# 配置日志
|
|
logging.basicConfig(level=logging.INFO)
|
|
logger = logging.getLogger(__name__)
|
|
|
|
def init_scheduler(app):
|
|
"""初始化定时任务调度器"""
|
|
scheduler = BackgroundScheduler()
|
|
|
|
with app.app_context():
|
|
# 获取所有启用的健康检查配置
|
|
configs = HealthCheckConfig.query.filter_by(is_enabled=True).all()
|
|
|
|
for config in configs:
|
|
# 为每个平台添加定时任务
|
|
scheduler.add_job(
|
|
func=lambda platform=config.platform: run_health_check(app, platform),
|
|
trigger=IntervalTrigger(seconds=config.check_interval),
|
|
id=f'health_check_{config.platform}',
|
|
name=f'{config.platform}健康检查',
|
|
replace_existing=True
|
|
)
|
|
logger.info(f"已添加{config.platform}健康检查任务，间隔{config.check_interval}秒")
|
|
|
|
scheduler.start()
|
|
logger.info("定时任务调度器已启动")
|
|
|
|
return scheduler
|
|
|
|
def run_health_check(app, platform):
|
|
"""执行健康检查"""
|
|
with app.app_context():
|
|
try:
|
|
logger.info(f"开始执行{platform}健康检查")
|
|
HealthChecker.check_platform(platform)
|
|
logger.info(f"{platform}健康检查完成")
|
|
except Exception as e:
|
|
logger.error(f"{platform}健康检查失败: {str(e)}")
|
|
|
|
if __name__ == '__main__':
|
|
app = create_app()
|
|
scheduler = init_scheduler(app)
|
|
|
|
try:
|
|
# 保持程序运行
|
|
import time
|
|
while True:
|
|
time.sleep(1)
|
|
except (KeyboardInterrupt, SystemExit):
|
|
scheduler.shutdown()
|
|
logger.info("定时任务调度器已停止")
|