prevent blocking

FGasper · FGasper · commit e740f006ba86 · 2024-11-07T09:18:06.000-05:00
diff --git a/internal/verifier/change_stream.go b/internal/verifier/change_stream.go
@@ -160,7 +160,14 @@ func (verifier *Verifier) iterateChangeStream(ctx context.Context, cs *mongo.Cha
 		}
 
 		if err != nil && !errors.Is(err, context.Canceled) {
-			verifier.changeStreamErrChan <- err
+			if changeStreamEnded {
+				// Avoid potentially blocking if the change stream is done.
+				verifier.logger.Fatal().
+					Err(err).
+					Msg("Failed to finish reading change stream.")
+			} else {
+				verifier.changeStreamErrChan <- err
+			}
 		}
 
 		if changeStreamEnded {

Original file line number	Diff line number	Diff line change
`@@ -160,7 +160,14 @@ func (verifier Verifier) iterateChangeStream(ctx context.Context, cs mongo.Cha`
`160`	`160`	`}`
`161`	`161`
`162`	`162`	`if err != nil && !errors.Is(err, context.Canceled) {`
`163`		`- verifier.changeStreamErrChan <- err`
	`163`	`+ if changeStreamEnded {`
	`164`	`+ // Avoid potentially blocking if the change stream is done.`
	`165`	`+ verifier.logger.Fatal().`
	`166`	`+ Err(err).`
	`167`	`+ Msg("Failed to finish reading change stream.")`
	`168`	`+ } else {`
	`169`	`+ verifier.changeStreamErrChan <- err`
	`170`	`+ }`
`164`	`171`	`}`
`165`	`172`
`166`	`173`	`if changeStreamEnded {`