aidlab/AiDA_Python
6
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

测试RABBITMQ 配置

Browse Source
This commit is contained in:
zchen
2024-03-26 11:50:57 +08:00
parent 17ecd7bf0a
commit 6ab174b0cf
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
app/core/config.py
View File

@@ -70,4 +70,4 @@ RABBITMQ_PARAMS = {
"virtual_host": "/" "virtual_host": "/"
} }
RABBITMQ_QUEUES = os.getenv("RABBITMQ_QUEUES", "SuperResolution-local") RABBITMQ_QUEUES = os.getenv("RABBITMQ_QUEUES", "SuperResolution-dev")

4
app/service/super_resolution/service.py
View File

@@ -16,7 +16,7 @@ import tritonclient.grpc as grpcclient
from PIL import Image from PIL import Image
from minio import Minio from minio import Minio
from app.core.config import MINIO_IP, MINIO_ACCESS, MINIO_SECRET, MINIO_SECURE, MINIO_PORT, REDIS_HOST, REDIS_PORT, REDIS_DB, SR_MODEL_NAME, RABBITMQ_PARAMS from app.core.config import MINIO_IP, MINIO_ACCESS, MINIO_SECRET, MINIO_SECURE, MINIO_PORT, REDIS_HOST, REDIS_PORT, REDIS_DB, SR_MODEL_NAME, RABBITMQ_PARAMS, RABBITMQ_QUEUES
from app.schemas.super_resolution import SuperResolutionModel from app.schemas.super_resolution import SuperResolutionModel
from app.service.utils.decorator import RunTime from app.service.utils.decorator import RunTime
@@ -136,7 +136,7 @@ def publish_message(sr_data):
connection = pika.BlockingConnection(pika.ConnectionParameters(**RABBITMQ_PARAMS)) connection = pika.BlockingConnection(pika.ConnectionParameters(**RABBITMQ_PARAMS))
channel = connection.channel() channel = connection.channel()
# 发布消息，并设置回调函数 # 发布消息，并设置回调函数
channel.basic_publish(exchange='', routing_key='SuperResolution-local', body=sr_data) channel.basic_publish(exchange='', routing_key=RABBITMQ_QUEUES, body=sr_data)
logger.info(f" [x] Sent {sr_data}") logger.info(f" [x] Sent {sr_data}")
connection.close() connection.close()