From f68b5a9f04ea091a132731cc3c10357a2b2c0b30 Mon Sep 17 00:00:00 2001
From: zchengrong <124802516+zchengrong@users.noreply.github.com>
Date: Tue, 22 Apr 2025 16:01:54 +0800
Subject: [PATCH] =?UTF-8?q?feat=EF=BC=88=E6=96=B0=E5=8A=9F=E8=83=BD?=
 =?UTF-8?q?=EF=BC=89:=20fix=EF=BC=88=E4=BF=AE=E5=A4=8Dbug=EF=BC=89:=20=20?=
 =?UTF-8?q?=20=E6=89=8B=E5=8A=A8=E6=96=AD=E5=BC=80mq=E8=BF=9E=E6=8E=A5=20d?=
 =?UTF-8?q?ocs=EF=BC=88=E6=96=87=E6=A1=A3=E5=8F=98=E6=9B=B4=EF=BC=89:=20re?=
 =?UTF-8?q?factor=EF=BC=88=E9=87=8D=E6=9E=84=EF=BC=89:=20test(=E5=A2=9E?=
 =?UTF-8?q?=E5=8A=A0=E6=B5=8B=E8=AF=95):?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 app/service/generate_image/service_generate_image.py       | 1 +
 app/service/generate_image/service_generate_multi_view.py  | 1 +
 app/service/generate_image/service_generate_single_logo.py | 1 +
 3 files changed, 3 insertions(+)

diff --git a/app/service/generate_image/service_generate_image.py b/app/service/generate_image/service_generate_image.py
index 4ed8fd4..cfd5fd5 100644
--- a/app/service/generate_image/service_generate_image.py
+++ b/app/service/generate_image/service_generate_image.py
@@ -180,6 +180,7 @@ class GenerateImage:
             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()
             # self.channel.basic_publish(exchange='', routing_key=GI_RABBITMQ_QUEUES, body=str_generate_data)
             logger.info(f" [x] Sent {json.dumps(dict_generate_data, indent=4)}")
 
diff --git a/app/service/generate_image/service_generate_multi_view.py b/app/service/generate_image/service_generate_multi_view.py
index c930ab2..248e604 100644
--- a/app/service/generate_image/service_generate_multi_view.py
+++ b/app/service/generate_image/service_generate_multi_view.py
@@ -105,6 +105,7 @@ class GenerateMultiView:
             dict_generate_data, str_generate_data = self.read_tasks_status()
             if DEBUG is False:
                 self.channel.basic_publish(exchange='', routing_key=GMV_RABBITMQ_QUEUES, body=str_generate_data)
+                self.connection.close()
             # self.channel.basic_publish(exchange='', routing_key=GI_RABBITMQ_QUEUES, body=str_generate_data)
             logger.info(f" [x] Sent {json.dumps(dict_generate_data, indent=4)}")
 
diff --git a/app/service/generate_image/service_generate_single_logo.py b/app/service/generate_image/service_generate_single_logo.py
index e3def3e..af182b2 100644
--- a/app/service/generate_image/service_generate_single_logo.py
+++ b/app/service/generate_image/service_generate_single_logo.py
@@ -98,6 +98,7 @@ class GenerateSingleLogoImage:
             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)}")