Merge pull request #68 from ydb-platform/ctx_canel_fix

Reader error is not fatal in case of ctx cancelation
ydb-platform · Dec 20, 2024 · cbae920 · cbae920
2 parents 6bebf62 + 1bcd724
commit cbae920
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/internal/reader/reader.go b/internal/reader/reader.go
@@ -216,7 +216,11 @@ func ReadTopic(ctx context.Context, streamInfo StreamInfo, reader *client.TopicR
 	for ctx.Err() == nil {
 		msg, err := reader.ReadMessage(ctx)
 		if err != nil {
-			xlog.Fatal(ctx, "Unable to read message", zap.Error(err))
+			if ctx.Err() != nil {
+				xlog.Error(ctx, "Unable to read message", zap.Error(err))
+			} else {
+				xlog.Fatal(ctx, "Unable to read message, fatal error ctx is not cancelled", zap.Error(err))
+			}
 			return
 		}