Spaces:
Paused
Paused
| import sys,asyncio | |
| #from apscheduler.schedulers.background import BackgroundScheduler | |
| from apscheduler.schedulers.asyncio import AsyncIOScheduler # 替换为异步调度器 | |
| from app.utils.logging import log | |
| from app.utils.stats import clean_expired_stats | |
| from app.config import api_call_stats | |
| from app.utils import check_version | |
| from zoneinfo import ZoneInfo | |
| from app.config import settings | |
| def handle_exception(exc_type, exc_value, exc_traceback): | |
| """ | |
| 全局异常处理函数 | |
| 处理未捕获的异常,并记录到日志中 | |
| """ | |
| if issubclass(exc_type, KeyboardInterrupt): | |
| sys.excepthook(exc_type, exc_value, exc_traceback) | |
| return | |
| from app.utils.error_handling import translate_error | |
| error_message = translate_error(str(exc_value)) | |
| log('error', f"未捕获的异常: {error_message}", status_code=500, error_message=error_message) | |
| ''' | |
| def schedule_cache_cleanup(response_cache_manager, active_requests_manager): | |
| """ | |
| 设置定期清理缓存和活跃请求的定时任务 | |
| 顺便定时检查更新 | |
| Args: | |
| response_cache_manager: 响应缓存管理器实例 | |
| active_requests_manager: 活跃请求管理器实例 | |
| """ | |
| scheduler = BackgroundScheduler() | |
| scheduler.add_job(response_cache_manager.clean_expired, 'interval', minutes=1) # 每分钟清理过期缓存 | |
| scheduler.add_job(active_requests_manager.clean_completed, 'interval', seconds=30) # 每30秒清理已完成的活跃请求 | |
| scheduler.add_job(active_requests_manager.clean_long_running, 'interval', minutes=5, args=[300]) # 每5分钟清理运行超过5分钟的任务 | |
| scheduler.add_job(clean_expired_stats, 'interval', minutes=5,args=[api_call_stats]) # 每5分钟清理过期的统计数据 | |
| scheduler.add_job(check_version, 'interval', minutes=1) # 每4小时检查更新 | |
| scheduler.start() | |
| return scheduler | |
| ''' | |
| def schedule_cache_cleanup(response_cache_manager, active_requests_manager): | |
| """ | |
| 设置定期清理缓存和活跃请求的定时任务 | |
| 顺便定时检查更新 | |
| Args: | |
| response_cache_manager: 响应缓存管理器实例 | |
| active_requests_manager: 活跃请求管理器实例 | |
| """ | |
| beijing_tz = ZoneInfo("Asia/Shanghai") | |
| scheduler = AsyncIOScheduler(timezone=beijing_tz) # 使用 AsyncIOScheduler 替代 BackgroundScheduler | |
| # 添加任务时直接传递异步函数(无需额外包装) | |
| scheduler.add_job(response_cache_manager.clean_expired, 'interval', minutes=1) | |
| scheduler.add_job(active_requests_manager.clean_completed, 'interval', seconds=30) | |
| scheduler.add_job(active_requests_manager.clean_long_running, 'interval', minutes=5, args=[300]) | |
| scheduler.add_job(clean_expired_stats, 'interval', minutes=5, args=[api_call_stats]) | |
| scheduler.add_job(check_version, 'interval', hours=4) | |
| scheduler.add_job(api_call_stats_clean, 'cron', hour=16,minute=0) | |
| scheduler.start() | |
| return scheduler | |
| def api_call_stats_clean(): | |
| settings.api_call_stats = { | |
| 'last_24h': { | |
| 'total': {}, | |
| 'by_endpoint': {} | |
| }, | |
| 'hourly': { | |
| 'total': {}, | |
| 'by_endpoint': {} | |
| }, | |
| 'minute': { | |
| 'total': {}, | |
| 'by_endpoint': {} | |
| } | |
| } |