From 15f1cdb2c358d1c120b5ab4bc28ef596e3d93b80 Mon Sep 17 00:00:00 2001
From: zhangqian <zhangqian@123.com>
Date: 星期六, 21 十月 2023 14:22:58 +0800
Subject: [PATCH] 关闭nsq时也关闭生产者

---
 nsq/producer.go |    4 ++++
 nsq/nsq.go      |    1 +
 2 files changed, 5 insertions(+), 0 deletions(-)

diff --git a/nsq/nsq.go b/nsq/nsq.go
index 3561a25..18fa6ce 100644
--- a/nsq/nsq.go
+++ b/nsq/nsq.go
@@ -89,4 +89,5 @@
 
 func Stop() {
 	defaultConsumerManager.stop()
+	StopProducer()
 }
diff --git a/nsq/producer.go b/nsq/producer.go
index 516ed17..2f1e947 100644
--- a/nsq/producer.go
+++ b/nsq/producer.go
@@ -12,6 +12,10 @@
 	return producer
 }
 
+func StopProducer() {
+	nsqclient.DestroyProducerPool()
+}
+
 func initProducer() (err error) {
 	producer, err = nsqclient.NewProducer(conf.Conf.NsqConf.NsqdAddr)
 	if err != nil {

--
Gitblit v1.8.0