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

feat（新功能）:

Browse Source 
fix（修复bug）:   手动断开mq连接
docs（文档变更）:
refactor（重构）:
test(增加测试):
This commit is contained in:
zchengrong
2025-04-22 15:58:07 +08:00
parent ac9c9b016d
commit 4dbb259b8b
3 changed files with 7 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
app/service/generate_image/service_generate_product_image.py
View File

@@ -320,6 +320,7 @@ class GenerateProductImage:
dict_gen_product_data, str_gen_product_data = self.read_tasks_status() dict_gen_product_data, str_gen_product_data = self.read_tasks_status()
if DEBUG is False: if DEBUG is False:
self.channel.basic_publish(exchange='', routing_key=GPI_RABBITMQ_QUEUES, body=str_gen_product_data) self.channel.basic_publish(exchange='', routing_key=GPI_RABBITMQ_QUEUES, body=str_gen_product_data)
self.connection.close()
logger.info(f" [x] Sent to： {GPI_RABBITMQ_QUEUES} data：@@@@ {json.dumps(dict_gen_product_data, indent=4)}") logger.info(f" [x] Sent to： {GPI_RABBITMQ_QUEUES} data：@@@@ {json.dumps(dict_gen_product_data, indent=4)}")

1
app/service/generate_image/service_generate_relight_image.py
View File

@@ -139,6 +139,7 @@ class GenerateRelightImage:
dict_gen_product_data, str_gen_product_data = self.read_tasks_status() dict_gen_product_data, str_gen_product_data = self.read_tasks_status()
if DEBUG is False: if DEBUG is False:
self.channel.basic_publish(exchange='', routing_key=GRI_RABBITMQ_QUEUES, body=str_gen_product_data) self.channel.basic_publish(exchange='', routing_key=GRI_RABBITMQ_QUEUES, body=str_gen_product_data)
self.connection.close()
logger.info(f" [x] Sent to： {GRI_RABBITMQ_QUEUES} data：@@@@ {json.dumps(dict_gen_product_data, indent=4)}") logger.info(f" [x] Sent to： {GRI_RABBITMQ_QUEUES} data：@@@@ {json.dumps(dict_gen_product_data, indent=4)}")

8
app/service/generate_image/service_pose_transform.py
View File

@@ -29,6 +29,9 @@ logger = logging.getLogger()
class PoseTransformService: class PoseTransformService:
def __init__(self, request_data): def __init__(self, request_data):
if DEBUG is False:
self.connection = pika.BlockingConnection(pika.ConnectionParameters(**RABBITMQ_PARAMS))
self.channel = self.connection.channel()
self.grpc_client = grpcclient.InferenceServerClient(url=PT_MODEL_URL) self.grpc_client = grpcclient.InferenceServerClient(url=PT_MODEL_URL)
self.redis_client = redis.StrictRedis(host=REDIS_HOST, port=REDIS_PORT, db=REDIS_DB, decode_responses=True) self.redis_client = redis.StrictRedis(host=REDIS_HOST, port=REDIS_PORT, db=REDIS_DB, decode_responses=True)
self.category = "pose_transform" self.category = "pose_transform"
@@ -108,9 +111,8 @@ class PoseTransformService:
finally: finally:
dict_pose_transform_data, str_pose_transform_data = self.read_tasks_status() dict_pose_transform_data, str_pose_transform_data = self.read_tasks_status()
if DEBUG is False: if DEBUG is False:
connection = pika.BlockingConnection(pika.ConnectionParameters(**RABBITMQ_PARAMS)) self.channel.basic_publish(exchange='', routing_key=PS_RABBITMQ_QUEUES, body=str_pose_transform_data)
channel = connection.channel() self.connection.close()
channel.basic_publish(exchange='', routing_key=PS_RABBITMQ_QUEUES, body=str_pose_transform_data)
logger.info(f" [x] Sent to： {PS_RABBITMQ_QUEUES} data：@@@@ {json.dumps(dict_pose_transform_data, indent=4)}") logger.info(f" [x] Sent to： {PS_RABBITMQ_QUEUES} data：@@@@ {json.dumps(dict_pose_transform_data, indent=4)}")