From b3d87ae859f20f3bfb4d2e495d7a0f316900b670 Mon Sep 17 00:00:00 2001
From: zhangqian <zhangqian@123.com>
Date: 星期二, 28 十一月 2023 17:12:21 +0800
Subject: [PATCH] 获取生产进度接口有工序id参数从数据库查询实际开始时间和实际结束时间
---
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