From 9d8329d44385ce9acf0efa0eb079fe8d1047e3fb Mon Sep 17 00:00:00 2001
From: howell <2827207845@qq.com>
Date: Tue, 11 Jan 2022 17:14:30 +0800
Subject: [PATCH] fix: shama

---
 internal/mq/mq.go | 4 +++-
 main.go           | 6 ++++++
 2 files changed, 9 insertions(+), 1 deletion(-)

diff --git a/internal/mq/mq.go b/internal/mq/mq.go
index b88dc87..46c72cb 100644
--- a/internal/mq/mq.go
+++ b/internal/mq/mq.go
@@ -30,7 +30,9 @@ func SetUpMq() {
 		log.Println(err)
 	}
 	Conn = &rabbitMq{conn}
+}
 
+func SetUpShaMa(){
 	uri1 := fmt.Sprintf("amqp://%s:%s@%s:%s",
 		configs.ConfigShaMaMqUser,
 		configs.ConfigShaMaMqPwd,
@@ -46,7 +48,7 @@ func SetUpMq() {
 
 func (o rabbitMq) ListenRetry(queue string, count int, handler func([]byte) error) {
 	log.Printf("start listening ... %s\n", queue)
-	ch, err := Conn.Channel()
+	ch, err := o.Channel()
 	if err != nil {
 		log.Printf("%s:%s\n", queue, err.Error())
 		return
diff --git a/main.go b/main.go
index 8924ef0..2e32414 100644
--- a/main.go
+++ b/main.go
@@ -45,6 +45,10 @@ func main() {
 		mq.SetUpMq()
 	}
 
+	if configs.IsProductionEnv() {
+		mq.SetUpShaMa()
+	}
+
 	if configs.IsProductionEnv() {
 		gin.SetMode("release")
 	} else {
@@ -125,6 +129,8 @@ func main() {
 		if err := h5Server.Shutdown(context.Background()); err != nil {
 			log.Printf("HTTP server Shutdown: %v", err)
 		}
+		_ = mq.Conn.Close()
+		_ = mq.Conn1.Close()
 		c.Stop()
 		close(idleConnClosed)
 	}()