update error check to check context is done before panicing for eror

streemtech · Oct 20, 2024 · 0ac59f1 · 0ac59f1
1 parent b119432
commit 0ac59f1
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/internal/redisstreams/streams.go b/internal/redisstreams/streams.go
@@ -34,9 +34,18 @@ func (s *StreamListener) Listen() {
 				Count:   1,
 			}).Result()
 			if err != nil {
+
 				if errors.Is(err, context.Canceled) {
 					return
 				}
+
+				//select to check for ending satement
+				select {
+				case <-s.Ctx.Done():
+					return
+				default:
+				}
+
 				s.Logger(s.Ctx).Panic("failed to read data from stream", err, slog.String("streamlistener.stream", s.Key))
 			}