架构优化

app/api/v1/routes.py

@@ -9,7 +9,7 @@ from app.api.service import TaskManagerService, GzhCookieManager
 server_blueprint = Blueprint("api", __name__, url_prefix="/api")
 
 
-def server_routes(pools, log_service):
+def server_routes(pools, log_service, config):
     @server_blueprint.route("/get_cover", methods=["POST"])
     async def get_cover():
         params = await request.get_json()
@@ -20,7 +20,7 @@ def server_routes(pools, log_service):
     async def run_task():
         trace_id = generate_task_trace_id()
         data = await request.get_json()
-        task_scheduler = TaskScheduler(data, log_service, pools, trace_id)
+        task_scheduler = TaskScheduler(data, log_service, pools, trace_id, config)
         response = await task_scheduler.deal()
         return jsonify(response)
 

app/jobs/task_scheduler.py

@@ -29,8 +29,8 @@ class TaskScheduler(TaskHandler):
         result = await scheduler.deal()
     """
 
-    def __init__(self, data: dict, log_service, db_client, trace_id: str):
-        super().__init__(data, log_service, db_client, trace_id)
+    def __init__(self, data: dict, log_service, db_client, trace_id: str, config):
+        super().__init__(data, log_service, db_client, trace_id, config)
         self.table = TaskUtils.validate_table_name(TaskConstants.TASK_TABLE)
 
     # ==================== 数据库操作 ====================

task_app.py

@@ -3,13 +3,16 @@ import logging
 from quart_cors import cors
 from quart import Quart
 
+from app.core.config import GlobalConfigSettings
 from app.core.dependency import log_service, mysql_manager
 from app.api.v1 import server_routes
 
 
 app = Quart(__name__)
 app = cors(app, allow_origin="*")
-routes = server_routes(mysql_manager, log_service)
+config = GlobalConfigSettings()
+
+routes = server_routes(mysql_manager, log_service, config)
 app.register_blueprint(routes)
 
 logging.basicConfig(level=logging.INFO)