go/worker/storage: Fix potential memory leak through context

oasisprotocol · Sep 30, 2021 · 3c71017 · 3c71017
1 parent 6df3e80
commit 3c71017
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/go/worker/storage/committee/node.go b/go/worker/storage/committee/node.go
@@ -529,9 +529,12 @@ func (n *Node) fetchDiff(round uint64, prevRoot, thisRoot storageApi.Root) {
 				"new_root", thisRoot,
 			)
 
+			ctx, cancel := context.WithCancel(n.ctx)
+			defer cancel()
+
 			// Prioritize committee nodes.
 			var selectedNode *node.Node
-			ctx := storageApi.WithNodeSelectionCallback(n.ctx, func(n *node.Node) {
+			ctx = storageApi.WithNodeSelectionCallback(ctx, func(n *node.Node) {
 				selectedNode = n
 			})
 			if committee := n.commonNode.Group.GetEpochSnapshot().GetStorageCommittee(); committee != nil {