nsq增加重连机制

notify/producer.go

@@ -11,9 +11,13 @@ type NsqProducer struct {
 	producer *nsq.Producer
 }
 
-var NsqProducers *NsqProducer
+var (
+	NsqProducers *NsqProducer
+	nsqConfig  *NsqConfig //重试
+)
 
 func New(config *NsqConfig) {
+	nsqConfig = config
 	NsqProducers = InitProducer(config)
 }
 
@@ -38,9 +42,16 @@ func InitProducer(config *NsqConfig) *NsqProducer {
 //发布消息
 func (p *NsqProducer) Publish(topic string, message string) error {
 
-	logs.Info("topic","topic:%v,message:%v",topic,message)
-
+	//logs.Info("topic","topic:%v,message:%v",topic,message)
 	var err error
+	defer func() {
+		if err != nil {
+			//重试连接
+			InitProducer(nsqConfig)
+		}
+	}()
+
+
 	if p.producer != nil {
 		if message == "" { //不能发布空串，否则会导致error
 			return nil