fix(relayer): reconnect after notify close (#16329)

taikoxyz · Mar 5, 2024 · 8987156 · 8987156
1 parent 7423ffa
commit 8987156
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/packages/relayer/pkg/queue/rabbitmq/queue.go b/packages/relayer/pkg/queue/rabbitmq/queue.go
@@ -212,6 +212,13 @@ func (r *RabbitMQ) Notify(ctx context.Context, wg *sync.WaitGroup) error {
 				slog.Error("rabbitmq notify close connection")
 			}
 
+			r.Close(ctx)
+
+			if err := r.connect(); err != nil {
+				slog.Error("error reconnecting to rabbitmq after notify closed", "err", err.Error())
+				return err
+			}
+
 			return queue.ErrClosed
 		case err := <-r.chErrCh:
 			if err != nil {
@@ -220,6 +227,13 @@ func (r *RabbitMQ) Notify(ctx context.Context, wg *sync.WaitGroup) error {
 				slog.Error("rabbitmq notify close channel")
 			}
 
+			r.Close(ctx)
+
+			if err := r.connect(); err != nil {
+				slog.Error("error reconnecting to rabbitmq after notify closed", "err", err.Error())
+				return err
+			}
+
 			return queue.ErrClosed
 		case returnMsg := <-r.notifyReturnCh:
 			slog.Error("rabbitmq notify return", "id", returnMsg.MessageId, "err", returnMsg.ReplyText)