Files
AiDA_Python/app/service/design_batch/utils/MQ.py
2024-12-11 11:06:20 +08:00

25 lines
826 B
Python

import json
import pika
from app.core.config import RABBITMQ_PARAMS
def publish_status(task_id, progress, result):
connection = pika.BlockingConnection(pika.ConnectionParameters(**RABBITMQ_PARAMS))
# connection = pika.BlockingConnection(pika.ConnectionParameters('10.1.2.190'))
channel = connection.channel()
channel.queue_declare(queue='DesignBatch', durable=True)
message = {'task_id': task_id, 'progress': progress, "result": result}
channel.basic_publish(exchange='',
routing_key='DesignBatch',
body=json.dumps(message),
properties=pika.BasicProperties(
delivery_mode=2,
))
connection.close()
if __name__ == '__main__':
publish_status("1", "1", "1")