Always close pipe reader on WriteWALSegment return

The pipe writer will deadlock if writing out the WAL segment fails for any reason.
beeper · Dec 16, 2023 · 2b4ec80 · 2b4ec80
1 parent 9d79838
commit 2b4ec80
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/replica.go b/replica.go
@@ -237,6 +237,12 @@ func (r *Replica) syncWAL(ctx context.Context) (err error) {
 	var g errgroup.Group
 	g.Go(func() error {
 		_, err := r.Client.WriteWALSegment(ctx, pos, pr)
+
+		// Always close pipe reader to signal writers.
+		if e := pr.CloseWithError(err); err == nil {
+			return e
+		}
+
 		return err
 	})