Function worker_process

api_key_manager/unit_test.py:250–356

工作进程函数，模拟多个用户同时请求API密钥

(worker_id, num_requests, sources=None, req_queue=None, res_queue=None)

    logger.info("\nAPI密钥轮询测试完成")

def worker_process(worker_id, num_requests, sources=None, req_queue=None, res_queue=None):
    """工作进程函数，模拟多个用户同时请求API密钥"""
    if sources is None:
        sources = get_available_sources()  # 使用过滤后的可用源
    
    local_success = 0
    local_failure = 0
    api_keys_used = {}
    
    logger.info(f"工作进程 {worker_id} 启动，将执行 {num_requests} 个请求")
    
    # 如果使用请求队列模式
    if req_queue is not None and res_queue is not None:
        for _ in range(num_requests):
            try:
                # 从队列中获取任务
                task = req_queue.get(timeout=1)
                source_name = task['source_name']
                
                # 获取API密钥
                api_key = get_api_key(source_name)
                if api_key:
                    # 记录使用的API密钥
                    if api_key not in api_keys_used:
                        api_keys_used[api_key] = 0
                    api_keys_used[api_key] += 1
                    
                    # 随机决定是成功还是失败
                    status = random.random() < SUCCESS_RATE
                    
                    # 记录API密钥使用情况
                    if record_api_key_usage(api_key, source_name, status=status):
                        if status:
                            local_success += 1
                        else:
                            local_failure += 1
                        logger.info(f"进程 {worker_id} - 请求: 源 {source_name} - 状态 {'成功' if status else '失败'}")
                    else:
                        local_failure += 1
                        logger.error(f"进程 {worker_id} - 请求: 记录使用情况失败")
                else:
                    local_failure += 1
                    logger.error(f"进程 {worker_id} - 请求: 获取API密钥失败")
                
                # 标记任务完成
                req_queue.task_done()
                
            except queue.Empty:
                # 队列为空，可能所有任务已分发完
                break
            except Exception as e:
                logger.error(f"工作进程 {worker_id} 处理任务时出错: {str(e)}")
                local_failure += 1
                if req_queue is not None:
                    req_queue.task_done()
            
            # 随机短暂延迟，模拟真实请求间隔
            time.sleep(random.uniform(0.1, 0.3))