feat(新功能): batch generate 增加接口说明,design batch队列修改

fix(修复bug):
docs(文档变更):
refactor(重构):
test(增加测试):
This commit is contained in:
zchengrong
2025-04-22 11:09:13 +08:00
parent 4e55275e6e
commit e1231d3d0e
4 changed files with 73 additions and 12 deletions

View File

@@ -2,16 +2,16 @@ import json
import pika
from app.core.config import RABBITMQ_PARAMS
from app.core.config import RABBITMQ_PARAMS, BATCH_DESIGN_RABBITMQ_QUEUES
def publish_status(task_id, progress, result):
connection = pika.BlockingConnection(pika.ConnectionParameters(**RABBITMQ_PARAMS))
channel = connection.channel()
channel.queue_declare(queue='DesignBatch', durable=True)
channel.queue_declare(queue=BATCH_DESIGN_RABBITMQ_QUEUES, durable=True)
message = {'task_id': task_id, 'progress': progress, "result": result}
channel.basic_publish(exchange='',
routing_key='DesignBatch',
routing_key=BATCH_DESIGN_RABBITMQ_QUEUES,
body=json.dumps(message),
properties=pika.BasicProperties(
delivery_mode=2,