feat（新功能）:

fix（修复bug）:  图片生成服务优化，避免mq连接超时
docs（文档变更）:
refactor（重构）:
test(增加测试):

app/service/generate_image/service_generate_single_logo.py

@@ -21,6 +21,7 @@ from tritonclient.utils import np_to_triton_dtype
 from app.core.config import *
 import tritonclient.grpc as grpcclient
 from app.schemas.generate_image import GenerateSingleLogoImageModel
+from app.service.generate_image.utils.mq import publish_status
 from app.service.generate_image.utils.upload_sd_image import upload_png_sd, upload_SDXL_image
 
 logger = logging.getLogger()
@@ -28,10 +29,6 @@ logger = logging.getLogger()
 
 class GenerateSingleLogoImage:
     def __init__(self, request_data):
-        if DEBUG is False:
-            self.connection = pika.BlockingConnection(pika.ConnectionParameters(**RABBITMQ_PARAMS))
-            self.channel = self.connection.channel()
-        self.minio_client = Minio(MINIO_URL, access_key=MINIO_ACCESS, secret_key=MINIO_SECRET, secure=MINIO_SECURE)
         self.grpc_client = grpcclient.InferenceServerClient(url=GSL_MODEL_URL)
         self.redis_client = redis.StrictRedis(host=REDIS_HOST, port=REDIS_PORT, db=REDIS_DB, decode_responses=True)
         self.batch_size = 1
@@ -96,10 +93,8 @@ class GenerateSingleLogoImage:
             raise Exception(str(e))
         finally:
             dict_generate_data, str_generate_data = self.read_tasks_status()
-            if DEBUG is False:
-                self.channel.basic_publish(exchange='', routing_key=GI_RABBITMQ_QUEUES, body=str_generate_data)
-                self.connection.close()
-            logger.info(f" [x] Sent {json.dumps(dict_generate_data, indent=4)}")
+            if not DEBUG:
+                publish_status(str_generate_data, GI_RABBITMQ_QUEUES)
 
 
 def infer_cancel(tasks_id):