handle notifyClose

sx-network · Nov 3, 2023 · d9cd613 · d9cd613
1 parent 0edec90
commit d9cd613
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/datafeed/mq.go b/datafeed/mq.go
@@ -175,6 +175,14 @@ func (mq *MQService) startConsumer(
 		}()
 	}
 
+	go func() {
+		notifyCloseError := <-mq.connection.Channel.NotifyClose(make(chan *amqp.Error))
+		if notifyCloseError != nil {
+			mq.logger.Debug("Got notifyCloseError error")
+			errors <- fmt.Errorf("Connection closed: %v", notifyCloseError)
+		}
+	}()
+
 	// stop the consumer upon sigterm
 	go func() {
 		<-ctx.Done()