Method get_api_key

LoadBalancing.py:59–122

从API密钥管理服务获取API密钥，失败时使用备用方案 Args: source_name (str): 源名称 Returns: str: API密钥 Raises: Exception: 如果主备用方案都无法获取API密钥

(source_name)

    @staticmethod
    def get_api_key(source_name):
        """从API密钥管理服务获取API密钥，失败时使用备用方案
        
        Args:
            source_name (str): 源名称
            
        Returns:
            str: API密钥
            
        Raises:
            Exception: 如果主备用方案都无法获取API密钥
        """
        logger = logging.getLogger(__name__)
        # 防止日志向上传播，避免重复打印
        logger.propagate = False
        
        try:
            # 优先尝试调用API密钥管理服务API
            response = requests.post(
                f"{API_KEY_MANAGER_URL}{API_KEY_MANAGER_GET_ENDPOINT}", 
                json={"source_name": source_name}, 
                timeout=INNER_TIMEOUT
            )
            
            if response.status_code == 200:
                result = response.json()
                logger.info(f"成功从API密钥管理服务获取 {source_name} 的API密钥")
                return result["api_key"]
            else:
                # 如果API调用失败，记录警告并尝试备用方案
                logger.warning(f"API密钥服务返回错误状态码: {response.status_code}，尝试使用备用方案")
                raise Exception(f"API service returned status code: {response.status_code}")
                
        except Exception as e:
            # 连接失败时，尝试使用备用方案：从pool_mapping中获取API密钥
            logger.warning(f"无法连接到API密钥管理服务 (已离线): {str(e)}，使用兜底离线配获取API密钥")
            
            try:
                if source_name not in pool_mapping:
                    raise ValueError(f"兜底离线配中未找到源 '{source_name}' 的配置")
                
                source_pool = pool_mapping[source_name]
                if not source_pool:
                    raise ValueError(f"源 '{source_name}' 的API密钥池为空")
                
                # 从所有账户的所有API密钥中随机选择一个
                all_api_keys = []
                for account, keys in source_pool.items():
                    for key_info in keys:
                        if "api_key" in key_info:
                            all_api_keys.append(key_info["api_key"])
                
                if not all_api_keys:
                    raise ValueError(f"源 '{source_name}' 的兜底离线配中没有有效的API密钥")
                
                selected_api_key = random.choice(all_api_keys)
                logger.warning(f"使用兜底离线配置为源 '{source_name}' 随机选择了一个API密钥")
                return selected_api_key