From 53d80c839ad2aad0a0482ca43fca1fc3a95b9d1c Mon Sep 17 00:00:00 2001
From: zhangqian <zhangqian@123.com>
Date: 星期五, 03 十一月 2023 17:30:54 +0800
Subject: [PATCH] 更改路由
---
nsq/producer.go | 17 ++++++++++++++++-
1 files changed, 16 insertions(+), 1 deletions(-)
diff --git a/nsq/producer.go b/nsq/producer.go
index 516ed17..ade2e2f 100644
--- a/nsq/producer.go
+++ b/nsq/producer.go
@@ -4,15 +4,30 @@
"apsClient/conf"
"apsClient/pkg/logx"
"apsClient/pkg/nsqclient"
+ "sync/atomic"
)
-var producer nsqclient.Producer
+var (
+ producer nsqclient.Producer
+ initFlag int32
+)
func GetProducer() nsqclient.Producer {
return producer
}
+func StopProducer() {
+ if !atomic.CompareAndSwapInt32(&initFlag, 1, 0) {
+ return
+ }
+ nsqclient.DestroyProducerPool()
+}
+
func initProducer() (err error) {
+ if !atomic.CompareAndSwapInt32(&initFlag, 0, 1) {
+ return nil
+ }
+
producer, err = nsqclient.NewProducer(conf.Conf.NsqConf.NsqdAddr)
if err != nil {
logx.Errorf("NewProducer err:%v", err)
--
Gitblit v1.8.0