design design batch

2024-12-11 14:40:58 +08:00
parent 84fe2663f4
commit be16c95faa
4 changed files with 7 additions and 5 deletions
--- a/app/service/design_batch/utils/MQ.py
+++ b/app/service/design_batch/utils/MQ.py
@@ -2,9 +2,11 @@ import json

 import pika

+from app.core.config import RABBITMQ_PARAMS
+

 def publish_status(task_id, progress, result):
-    connection = pika.BlockingConnection(pika.ConnectionParameters('10.1.2.213'))
+    connection = pika.BlockingConnection(pika.ConnectionParameters(**RABBITMQ_PARAMS))
    channel = connection.channel()
    channel.queue_declare(queue='DesignBatch', durable=True)
    message = {'task_id': task_id, 'progress': progress, "result": result}