luojunhui 1 месяц назад
Родитель
Сommit
29ee817e24
1 измененных файлов с 11 добавлено и 1 удалено
  1. 11 1
      app/api/service/task_scheduler.py

+ 11 - 1
app/api/service/task_scheduler.py

@@ -18,6 +18,9 @@ from app.jobs.task_utils import (
     TaskConcurrencyError,
     TaskUtils,
 )
+from app.core.config import GlobalConfigSettings
+from app.core.database import DatabaseManager
+from app.core.observability import LogService
 
 
 class TaskScheduler(TaskHandler):
@@ -29,7 +32,14 @@ class TaskScheduler(TaskHandler):
         result = await scheduler.deal()
     """
 
-    def __init__(self, data: dict, log_service, db_client, trace_id: str, config):
+    def __init__(
+            self,
+            data: dict,
+            log_service: LogService,
+            db_client: DatabaseManager,
+            trace_id: str,
+            config: GlobalConfigSettings
+    ):
         super().__init__(data, log_service, db_client, trace_id, config)
         self.table = TaskUtils.validate_table_name(TaskConstants.TASK_TABLE)