From 55a286f521e809d2aadd8a2f2625956fb5678835 Mon Sep 17 00:00:00 2001
From: windyboy <windyboy@gmail.com>
Date: Wed, 7 Aug 2024 11:10:08 +0800
Subject: [PATCH] refactor: Improve error handling in NATS message handling

---
 internal/nats/sub.go | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/internal/nats/sub.go b/internal/nats/sub.go
index c020c30..2d42e3c 100644
--- a/internal/nats/sub.go
+++ b/internal/nats/sub.go
@@ -45,7 +45,11 @@ func Subscribe(config *config.Config, marshaler *handlers.PlainTextMarshaler, ha
 	}
 
 	for msg := range messages {
-		handler.HandleMessage(msg)
-		msg.Ack()
+		if err := handler.HandleMessage(msg); err == nil {
+			msg.Ack()
+		} else {
+			logger.Error("Failed to handle message", err, map[string]interface{}{"message": msg})
+			msg.Nack()
+		}
 	}
 }