Prevent sending data to closed dirtyResult channel

Azure · Sep 5, 2019 · f5592ff · f5592ff
1 parent 7dd2941
commit f5592ff
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/storage/storage.go b/storage/storage.go
@@ -517,9 +517,10 @@ func (sl *LeaserCheckpointer) persistDirtyPartitions(ctx context.Context) error
 	for _, pid := range pids {
 		go func(id string) {
 			err := sl.persistLease(ctx, id)
-			resCh <- dirtyResult{
-				Err:         err,
-				PartitionID: id,
+			select {
+			case <-ctx.Done():
+				return
+			case resCh <- dirtyResult{PartitionID: id, Err: err}:
 			}
 		}(pid)
 	}