Pause rollups during snapshot streaming. (#6282)

hypermodeinc · Sep 30, 2020 · b777913 · b777913
1 parent a846471
commit b777913
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/worker/snapshot.go b/worker/snapshot.go
@@ -255,6 +255,13 @@ func doStreamSnapshot(snap *pb.Snapshot, out pb.Worker_StreamSnapshotServer) err
 }
 
 func (w *grpcWorker) StreamSnapshot(stream pb.Worker_StreamSnapshotServer) error {
+	// Pause rollups during snapshot streaming.
+	closer, err := groups().Node.startTask(opSnapshot)
+	if err != nil {
+		return err
+	}
+	defer closer.Done()
+
 	n := groups().Node
 	if n == nil || n.Raft() == nil {
 		return conn.ErrNoNode