Prechádzať zdrojové kódy

Fix: 修复硬编码

iaun 2 mesiacov pred
rodič
commit
1efe46c672
3 zmenil súbory, kde vykonal 7 pridanie a 4 odobranie
  1. 1 1
      DOCKER_DEPLOY.md
  2. 5 2
      backend/app/api/tasks.py
  3. 1 1
      docker-compose.yml

+ 1 - 1
DOCKER_DEPLOY.md

@@ -28,7 +28,7 @@ docker-compose down
 docker run -p 5000:5000 your-image-name
 
 # 只运行 Worker
-docker run your-image-name celery -A celery_worker.celery worker --loglevel=info
+docker run your-image-name celery -A celery_worker.celery_app worker --loglevel=info
 
 # 初始化数据库
 docker-compose exec web python init_db.py

+ 5 - 2
backend/app/api/tasks.py

@@ -14,6 +14,7 @@ Requirements: 3.1, 3.4, 4.1, 4.2, 4.3, 4.4, 4.5, 4.6
 import os
 import json
 import uuid
+import redis
 from datetime import datetime
 from flask import jsonify, request, current_app
 from werkzeug.utils import secure_filename
@@ -257,7 +258,8 @@ def create_task():
         # 先测试Redis连接
         import redis
         print(f"🔍 测试Redis连接...")
-        r = redis.Redis(host='localhost', port=6379, db=0)
+        broker_url = current_app.config.get('CELERY_BROKER_URL', 'redis://localhost:6379/0')
+        r = redis.from_url(broker_url)
         r.ping()
         print(f"✅ Redis连接成功")
         
@@ -511,7 +513,8 @@ def upload_scan():
     
     try:
         # Test Redis connection
-        r = redis.Redis(host='localhost', port=6379, db=0)
+        broker_url = current_app.config.get('CELERY_BROKER_URL', 'redis://localhost:6379/0')
+        r = redis.from_url(broker_url)
         r.ping()
         
         # Initialize Celery

+ 1 - 1
docker-compose.yml

@@ -38,7 +38,7 @@ services:
   # Celery worker
   worker:
     build: .
-    command: celery -A celery_worker.celery worker --loglevel=info
+    command: celery -A celery_worker.celery_app worker --loglevel=info
     environment:
       - FLASK_ENV=production
       - SECRET_KEY=${SECRET_KEY:-change-me-in-production}