error-check the resume token persistence

FGasper · FGasper · commit 57a155a6b647 · 2024-11-05T13:28:43.000-05:00
diff --git a/internal/verifier/change_stream.go b/internal/verifier/change_stream.go
@@ -122,6 +122,10 @@ func (verifier *Verifier) iterateChangeStream(ctx context.Context, cs *mongo.Cha
 			)
 		}
 
+		if err == nil {
+			err = persistResumeTokenIfNeeded()
+		}
+
 		if err != nil {
 			if !errors.Is(err, context.Canceled) {
 				verifier.changeStreamErrChan <- err
@@ -130,8 +134,6 @@ func (verifier *Verifier) iterateChangeStream(ctx context.Context, cs *mongo.Cha
 			return
 		}
 
-		persistResumeTokenIfNeeded()
-
 		select {
 		// If the changeStreamEnderChan has a message, the user has indicated that
 		// source writes are ended. This means we should exit rather than continue

Original file line number	Diff line number	Diff line change
`@@ -122,6 +122,10 @@ func (verifier Verifier) iterateChangeStream(ctx context.Context, cs mongo.Cha`
`122`	`122`	`)`
`123`	`123`	`}`
`124`	`124`
	`125`	`+ if err == nil {`
	`126`	`+ err = persistResumeTokenIfNeeded()`
	`127`	`+ }`
	`128`	`+`
`125`	`129`	`if err != nil {`
`126`	`130`	`if !errors.Is(err, context.Canceled) {`
`127`	`131`	`verifier.changeStreamErrChan <- err`
`@@ -130,8 +134,6 @@ func (verifier Verifier) iterateChangeStream(ctx context.Context, cs mongo.Cha`
`130`	`134`	`return`
`131`	`135`	`}`
`132`	`136`
`133`		`- persistResumeTokenIfNeeded()`
`134`		`-`
`135`	`137`	`select {`
`136`	`138`	`// If the changeStreamEnderChan has a message, the user has indicated that`
`137`	`139`	`// source writes are ended. This means we should exit rather than continue`