改为遍历配置文件
This commit is contained in:
mark.tian
@@ -2,64 +2,6 @@
|
||||
|
||||
# API配置
|
||||
API_CONFIG = {
|
||||
# 主要API接口
|
||||
'main_api': {
|
||||
'url': 'https://jsonplaceholder.typicode.com/posts',
|
||||
'method': 'POST',
|
||||
'headers': {
|
||||
'Content-Type': 'application/json',
|
||||
'User-Agent': 'Celery-Task/1.0'
|
||||
},
|
||||
'timeout': 30,
|
||||
'body': {
|
||||
'title': '每日定时任务',
|
||||
'body': '这是通过配置文件设置的请求体数据',
|
||||
'userId': 1
|
||||
}
|
||||
},
|
||||
|
||||
# 备用API接口
|
||||
'backup_api': {
|
||||
'url': 'https://jsonplaceholder.typicode.com/comments',
|
||||
'method': 'POST',
|
||||
'headers': {
|
||||
'Content-Type': 'application/json',
|
||||
'User-Agent': 'Celery-Task/1.0'
|
||||
},
|
||||
'timeout': 30,
|
||||
'body': {
|
||||
'name': '备用任务',
|
||||
'email': 'backup@example.com',
|
||||
'body': '这是备用API的请求体'
|
||||
}
|
||||
},
|
||||
|
||||
# 健康检查API
|
||||
'health_check': {
|
||||
'url': 'https://jsonplaceholder.typicode.com/posts/1',
|
||||
'method': 'GET',
|
||||
'headers': {
|
||||
'User-Agent': 'Celery-Task/1.0'
|
||||
},
|
||||
'timeout': 10,
|
||||
'body': {} # GET请求通常不需要body，但可以设置查询参数
|
||||
},
|
||||
|
||||
# GET请求示例API
|
||||
'get_api': {
|
||||
'url': 'https://jsonplaceholder.typicode.com/posts',
|
||||
'method': 'GET',
|
||||
'headers': {
|
||||
'User-Agent': 'Celery-Task/1.0'
|
||||
},
|
||||
'timeout': 30,
|
||||
'body': {
|
||||
'userId': 1, # 这将作为查询参数
|
||||
'limit': 10
|
||||
}
|
||||
},
|
||||
|
||||
# PUT请求示例API
|
||||
'put_api': {
|
||||
'url': 'https://jsonplaceholder.typicode.com/posts/1',
|
||||
'method': 'PUT',
|
||||
|
||||
@@ -9,16 +9,16 @@ logger = logging.getLogger(__name__)
|
||||
|
||||
@app.task(bind=True, max_retries=None)
|
||||
def invoke_call_api_task(self):
|
||||
"""遍历API配置文件中的所有API配置并调用接口"""
|
||||
"""遍历API配置文件中的所有API信息并调用接口"""
|
||||
current_time = datetime.now().strftime('%Y-%m-%d %H:%M:%S')
|
||||
logger.info(f"开始遍历API配置文件执行呼叫任务 - 当前时间: {current_time}")
|
||||
logger.info(f"开始遍历API配置文件执行调用任务 - 当前时间: {current_time}")
|
||||
|
||||
# 获取重试配置
|
||||
max_retries = RETRY_CONFIG['max_retries']
|
||||
retry_delay = RETRY_CONFIG['retry_delay']
|
||||
retry_on_status = RETRY_CONFIG['retry_on_status']
|
||||
|
||||
# 遍历所有API配置
|
||||
# 遍历所有API信息
|
||||
results = []
|
||||
success_count = 0
|
||||
|
||||
@@ -32,7 +32,7 @@ def invoke_call_api_task(self):
|
||||
payload = call_config['body']
|
||||
|
||||
try:
|
||||
logger.info(f"正在调用呼叫任务API: {method} {call_url}")
|
||||
logger.info(f"正在调用任务API: {method} {call_url}")
|
||||
|
||||
# 根据method选择请求方式
|
||||
if method == 'POST':
|
||||
@@ -53,14 +53,14 @@ def invoke_call_api_task(self):
|
||||
except:
|
||||
result_data = {'response': response.text}
|
||||
|
||||
logger.info(f"呼叫任务API调用成功，响应数据: {result_data}")
|
||||
print(f"[{current_time}] ✅ 呼叫任务API成功 ({api_key}) - {method} {call_url}")
|
||||
logger.info(f"任务API调用成功，响应数据: {result_data}")
|
||||
print(f"[{current_time}] ✅ 任务API成功 ({api_key}) - {method} {call_url}")
|
||||
success_count += 1
|
||||
|
||||
results.append({
|
||||
'call_api_key': api_key,
|
||||
'status': 'success',
|
||||
'message': f'呼叫任务API调用成功 ({method})',
|
||||
'message': f'任务API调用成功 ({method})',
|
||||
'response': result_data,
|
||||
'timestamp': current_time,
|
||||
'retry_count': getattr(self.request, 'retries', 0)
|
||||
@@ -69,28 +69,28 @@ def invoke_call_api_task(self):
|
||||
# 需要重试的状态码
|
||||
current_retry = getattr(self.request, 'retries', 0)
|
||||
if current_retry < max_retries:
|
||||
logger.warning(f"呼叫任务API需要重试，当前重试次数: {current_retry + 1}/{max_retries}")
|
||||
print(f"[{current_time}] 🔄 呼叫任务API失败，正在重试 ({current_retry + 1}/{max_retries}) - {api_key} - 状态码: {response.status_code}")
|
||||
raise self.retry(countdown=retry_delay, exc=Exception(f"呼叫任务API失败，状态码: {response.status_code}"))
|
||||
logger.warning(f"任务API需要重试，当前重试次数: {current_retry + 1}/{max_retries}")
|
||||
print(f"[{current_time}] 🔄 任务API失败，正在重试 ({current_retry + 1}/{max_retries}) - {api_key} - 状态码: {response.status_code}")
|
||||
raise self.retry(countdown=retry_delay, exc=Exception(f"任务API失败，状态码: {response.status_code}"))
|
||||
else:
|
||||
logger.error(f"呼叫任务API失败，已达到最大重试次数: {max_retries}")
|
||||
print(f"[{current_time}] ❌ 呼叫任务API失败，已达到最大重试次数 - {api_key} - 状态码: {response.status_code}")
|
||||
logger.error(f"任务API失败，已达到最大重试次数: {max_retries}")
|
||||
print(f"[{current_time}] ❌ 任务API失败，已达到最大重试次数 - {api_key} - 状态码: {response.status_code}")
|
||||
results.append({
|
||||
'call_api_key': api_key,
|
||||
'status': 'failed',
|
||||
'message': f'呼叫任务API失败，已达到最大重试次数: {max_retries}',
|
||||
'message': f'任务API失败，已达到最大重试次数: {max_retries}',
|
||||
'response': response.text,
|
||||
'timestamp': current_time,
|
||||
'retry_count': current_retry + 1
|
||||
})
|
||||
else:
|
||||
# 其他失败状态码，不重试
|
||||
logger.error(f"呼叫任务API失败，状态码: {response.status_code}, 响应: {response.text}")
|
||||
print(f"[{current_time}] ❌ 呼叫任务API失败 - {api_key} - 状态码: {response.status_code}")
|
||||
logger.error(f"任务API失败，状态码: {response.status_code}, 响应: {response.text}")
|
||||
print(f"[{current_time}] ❌ 任务API失败 - {api_key} - 状态码: {response.status_code}")
|
||||
results.append({
|
||||
'call_api_key': api_key,
|
||||
'status': 'failed',
|
||||
'message': f'呼叫任务API失败，状态码: {response.status_code}',
|
||||
'message': f'任务API失败，状态码: {response.status_code}',
|
||||
'response': response.text,
|
||||
'timestamp': current_time,
|
||||
'retry_count': getattr(self.request, 'retries', 0)
|
||||
@@ -100,16 +100,16 @@ def invoke_call_api_task(self):
|
||||
# 网络异常，需要重试
|
||||
current_retry = getattr(self.request, 'retries', 0)
|
||||
if current_retry < max_retries:
|
||||
logger.warning(f"呼叫任务API异常，需要重试，当前重试次数: {current_retry + 1}/{max_retries}")
|
||||
print(f"[{current_time}] 🔄 呼叫任务API异常，正在重试 ({current_retry + 1}/{max_retries}) - {api_key} - 异常: {str(e)}")
|
||||
logger.warning(f"任务API异常，需要重试，当前重试次数: {current_retry + 1}/{max_retries}")
|
||||
print(f"[{current_time}] 🔄 任务API异常，正在重试 ({current_retry + 1}/{max_retries}) - {api_key} - 异常: {str(e)}")
|
||||
raise self.retry(countdown=retry_delay, exc=e)
|
||||
else:
|
||||
logger.error(f"呼叫任务API异常，已达到最大重试次数: {max_retries}")
|
||||
print(f"[{current_time}] ❌ 呼叫任务API异常，已达到最大重试次数 - {api_key} - 异常: {str(e)}")
|
||||
logger.error(f"任务API异常，已达到最大重试次数: {max_retries}")
|
||||
print(f"[{current_time}] ❌ 任务API异常，已达到最大重试次数 - {api_key} - 异常: {str(e)}")
|
||||
results.append({
|
||||
'call_api_key': api_key,
|
||||
'status': 'error',
|
||||
'message': f'呼叫任务API异常，已达到最大重试次数: {max_retries}',
|
||||
'message': f'任务API异常，已达到最大重试次数: {max_retries}',
|
||||
'error': str(e),
|
||||
'timestamp': current_time,
|
||||
'retry_count': current_retry + 1
|
||||
@@ -117,7 +117,7 @@ def invoke_call_api_task(self):
|
||||
|
||||
# 汇总结果
|
||||
total_count = len(results)
|
||||
logger.info(f"遍历API配置文件执行完成 - 成功: {success_count}/{total_count}")
|
||||
logger.info(f"遍历API配置文件完成 - 成功: {success_count}/{total_count}")
|
||||
print(f"[{current_time}] 📊 任务汇总 - 成功: {success_count}/{total_count}")
|
||||
|
||||
return {
|
||||
|
||||
Reference in New Issue
Block a user