| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051 |
- """
- 定时任务 - 清理过期数据
- """
- import asyncio
- from datetime import timedelta
- from app.core.logger import logger
- from app.database import SessionLocal
- from app.models.api_nonce import APINonce
- from app.utils.timezone import beijing_now_naive
- async def cleanup_expired_nonces_task():
- """定时清理过期的 API Nonce 数据"""
- while True:
- try:
- db = SessionLocal()
- try:
- # 删除已过期的 nonce（比过期时间早5分钟的数据）
- expire_threshold = beijing_now_naive() - timedelta(minutes=5)
- result = db.query(APINonce).filter(
- APINonce.expires_at < expire_threshold
- ).delete()
- db.commit()
- if result > 0:
- logger.info(f"[Cleanup] 已清理 {result} 条过期的 api_nonces 记录")
- finally:
- db.close()
- except Exception as e:
- logger.exception(f"[Cleanup] 清理 api_nonces 失败: {e}")
- try:
- db.rollback()
- except Exception:
- pass
- # 每小时执行一次
- await asyncio.sleep(3600)
- def start_cleanup_tasks():
- """启动所有定时清理任务"""
- loop = asyncio.new_event_loop()
- asyncio.set_event_loop(loop)
- # 启动 Nonce 清理任务
- loop.create_task(cleanup_expired_nonces_task())
- # 在后台线程中运行事件循环
- import threading
- thread = threading.Thread(target=loop.run_forever, daemon=True)
- thread.start()
- logger.info("[Cleanup] 定时清理任务已启动")
|
