Bitswap Getblocks: CancelWant on ctx cancel

License: MIT Signed-off-by: rht <[email protected]>
ipfs · Nov 22, 2015 · 5717463 · 5717463
1 parent b4ab684
commit 5717463
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/exchange/bitswap/bitswap.go b/exchange/bitswap/bitswap.go
@@ -208,6 +208,12 @@ func (bs *Bitswap) GetBlocks(ctx context.Context, keys []key.Key) (<-chan *block
 		log.Event(ctx, "Bitswap.GetBlockRequest.Start", &k)
 	}
 
+	go func() {
+		select {
+		case <-ctx.Done():
+			bs.CancelWants(keys)
+		}
+	}()
 	bs.wm.WantBlocks(keys)
 
 	req := &blockRequest{