From cab6eea89a0d82710369604ecb51f1cdb122e433 Mon Sep 17 00:00:00 2001
From: zhangqian <zhangqian@123.com>
Date: 星期四, 19 十月 2023 17:26:34 +0800
Subject: [PATCH] fix

---
 nsq/consumer.go |   25 ++++++++++---------------
 1 files changed, 10 insertions(+), 15 deletions(-)

diff --git a/nsq/consumer.go b/nsq/consumer.go
index feb2814..d75a999 100644
--- a/nsq/consumer.go
+++ b/nsq/consumer.go
@@ -5,12 +5,11 @@
 	"apsClient/constvar"
 	"apsClient/pkg/logx"
 	"apsClient/pkg/nsqclient"
-	"context"
 	"fmt"
 )
 
-func Consume(topic, channel string) (err error) {
-	c, err := nsqclient.NewNsqConsumer(context.Background(), topic, channel)
+func NewConsumer(topic, channel string) (c *nsqclient.NsqConsumer, err error) {
+	c, err = nsqclient.NewNsqConsumer(topic, channel)
 	if err != nil {
 		logx.Errorf("NewNsqConsumer err:%v", err)
 		return
@@ -22,19 +21,15 @@
 		handler = new(ScheduleTask)
 	case fmt.Sprintf(constvar.NsqTopicSendPlcAddress, conf.Conf.NsqConf.NodeId):
 		handler = &PlcAddress{Topic: topic}
+	case fmt.Sprintf(constvar.NsqTopicProcessParamsResponse, conf.Conf.NsqConf.NodeId):
+		handler = &ProcessParams{Topic: topic}
+	case fmt.Sprintf(constvar.NsqTopicApsProcessParams, conf.Conf.NsqConf.NodeId):
+		handler = &ProcessParamsSync{Topic: topic}
+	case fmt.Sprintf(constvar.NsqTopicDeviceUpdate, conf.Conf.NsqConf.NodeId):
+		handler = &DeviceUpdate{Topic: topic}
+	case fmt.Sprintf(constvar.NsqTopicPullDataResponse, conf.Conf.NsqConf.NodeId):
+		handler = &PullDataResponse{Topic: topic}
 	}
 	c.AddHandler(handler.HandleMessage)
-
-	if len(conf.Conf.NsqConf.NsqlookupdAddr) > 0 {
-		if err = c.RunLookupd(conf.Conf.NsqConf.NsqlookupdAddr, 1); err != nil {
-			logx.Errorf("RunLookupd err:%v", err)
-			return
-		}
-	} else {
-		if err = c.Run(conf.Conf.NsqConf.NsqdAddr, 1); err != nil {
-			logx.Errorf("Run err:%v", err)
-			return
-		}
-	}
 	return
 }

--
Gitblit v1.8.0