fix: use maxWaitingSecondsForNewMessage from argument instead of hard…

…code
huantt · Dec 4, 2023 · e21ca47 · e21ca47
1 parent f5ae472
commit e21ca47
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/cmd/streamer.go b/cmd/streamer.go
@@ -71,14 +71,17 @@ func CreateStreamCmd() (*cobra.Command, error) {
 					}
 				}
 			}()
-			maxWaitingSecondsForNewMessage := time.Duration(10) * time.Second
+			maxWaitingDurationForNewMessage := time.Duration(-1)
+			if maxWaitingSecondsForNewMessage != -1 {
+				maxWaitingDurationForNewMessage = time.Duration(maxWaitingSecondsForNewMessage) * time.Second
+			}
 			streamer := impl.NewStreamer(
 				consumer,
 				producer,
 				fromTopic,
 				toTopic,
 				deliveryChan,
-				impl.StreamerOptions{MaxWaitingTimeForNewMessage: &maxWaitingSecondsForNewMessage},
+				impl.StreamerOptions{MaxWaitingTimeForNewMessage: &maxWaitingDurationForNewMessage},
 			)
 			transferredCount, err := streamer.Run()
 			if err != nil {