增加多个消费者

notify/consumer.go

@@ -6,7 +6,6 @@ import (
 	"time"
 )
 
-
 type NsqConfig struct {
 	Topic   string
 	Channel string
@@ -28,3 +27,25 @@ func InitConsumer(consumer *NsqConfig) {
 		return
 	}
 }
+
+func InitBatchConsumer(consumer *NsqConfig, clientNum int) {
+	cfg := nsq.NewConfig()
+	cfg.LookupdPollInterval = time.Second * 30                       //设置重连时间
+	c, err := nsq.NewConsumer(consumer.Topic, consumer.Channel, cfg) // 新建一个消费者
+
+	if logs.CheckErr(err, "InitConsumer") {
+		return
+	}
+
+	if clientNum == 0 {
+		clientNum = 1
+	}
+	for i := 0; i < clientNum; i++ {
+		c.AddHandler(consumer.Handler) // 批量添加消费者接口
+	}
+	//建立NSQLookupd连接
+	err = c.ConnectToNSQD(consumer.Addr)
+	if logs.CheckErr(err, "ConnectToNSQD") {
+		return
+	}
+}