fix(worker): Flush the stream writer on error (DGRAPH-2499) (#6609)

hypermodeinc · Nov 10, 2020 · 7088c65 · 7088c65
1 parent ecace57
commit 7088c65
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/worker/snapshot.go b/worker/snapshot.go
@@ -68,6 +68,8 @@ func (n *node) populateSnapshot(snap pb.Snapshot, pl *conn.Pool) (int, error) {
 	var writer badgerWriter
 	if snap.SinceTs == 0 {
 		sw := pstore.NewStreamWriter()
+		defer sw.Cancel()
+
 		if err := sw.Prepare(); err != nil {
 			return 0, err
 		}