给每个定时任务增加一个开关

This commit is contained in:
mark.tian
2025-12-10 08:19:04 +08:00
parent 37c8503f40
commit 573aa29f36
3 changed files with 10 additions and 1 deletions

View File

@@ -41,6 +41,7 @@ engine = create_engine(
)
@celery_app.task(bind=True, name='push_data_to_dtc')
@conditional_task(enabled=settings.enabled_push_data_to_dtc_task)
def push_data_to_dtc_task(self):
"""
推送数据给DTC的Celery任务
@@ -221,7 +222,7 @@ def push_data_to_dtc_task(self):
@celery_app.task(bind=True, name='call_api', max_retries=None)
@conditional_task(enabled=False)
@conditional_task(enabled=settings.enabled_call_api_task)
def execute_call_api_task(self):
"""遍历API配置文件中的所有API信息并调用接口"""
current_time = datetime.now().strftime('%Y-%m-%d %H:%M:%S')

View File

@@ -18,6 +18,10 @@ class Settings(BaseSettings):
celery_timezone: str = "Asia/Shanghai"
celery_enable_utc: bool = False
# 任务配置
enabled_push_data_to_dtc_task: bool = False # 是否启用推送数据给DTC的任务
enabled_call_api_task: bool = False # 是否启用调用API的任务
# Flower监控配置
flower_enabled: bool = True # 是否启用Flower监控
flower_url: str = "http://localhost:5555" # Flower访问URL