Function update_failed_keys_cache

api_key_manager/api.py:81–175

定时任务：更新失败的API密钥缓存

()

# ==================== 定时任务函数 ====================

def update_failed_keys_cache():
    """定时任务：更新失败的API密钥缓存"""
    global last_update_time
    start_time = datetime.now()
    logger.info("🔄 开始更新失败API密钥缓存...")
    
    try:
        # 获取数据库会话
        db = next(get_db())
        
        # 计算时间窗口
        cutoff_time = datetime.now() - timedelta(minutes=TOLERANCE_TIMER_SPAN)
        
        # 重置缓存
        failed_keys_cache.clear()
        key_failure_count_cache.clear()
        stats_cache.clear()
        
        # 批量查询所有失败的API密钥 - 使用优化的索引查询
        failed_keys_query = db.query(
            ApiKeyUsage.source_name,
            ApiKeyUsage.api_key,
            func.count(ApiKeyUsage.request_id).label('failure_count'),
            func.max(ApiKeyUsage.finish_time).label('last_failure')
        ).filter(
            and_(
                ApiKeyUsage.status == False,
                ApiKeyUsage.finish_time >= cutoff_time
            )
        ).group_by(
            ApiKeyUsage.source_name,
            ApiKeyUsage.api_key
        ).all()
        
        # 统计信息
        total_failed_keys = 0
        source_stats = defaultdict(lambda: {
            'failed_keys': 0,
            'total_failures': 0,
            'models_affected': set(),
            'last_failure': None
        })
        
        # 处理查询结果
        for row in failed_keys_query:
            source_name = row.source_name
            api_key = row.api_key
            failure_count = row.failure_count
            last_failure = row.last_failure
            
            # 添加到失败缓存
            failed_keys_cache[source_name].add(api_key)
            # 记录失败次数
            key_failure_count_cache[source_name][api_key] = failure_count
            total_failed_keys += 1
            
            # 更新统计信息
            source_stats[source_name]['failed_keys'] += 1