From b13a5a940b40e6b24c1031275905dc763e667c2e Mon Sep 17 00:00:00 2001
From: Brian Tiger Chow <brian.holderchow@gmail.com>
Date: Wed, 19 Nov 2014 23:11:58 -0800
Subject: [PATCH] refactor(bitswap) move wantlist to loop receive

License: MIT
Signed-off-by: Brian Tiger Chow <brian@perfmode.com>
---
 exchange/bitswap/bitswap.go | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/exchange/bitswap/bitswap.go b/exchange/bitswap/bitswap.go
index a14d68cc0..608656e53 100644
--- a/exchange/bitswap/bitswap.go
+++ b/exchange/bitswap/bitswap.go
@@ -178,6 +178,9 @@ func (bs *bitswap) run(ctx context.Context) {
 			}
 		case ks := <-bs.batchRequests:
 			// TODO: implement batching on len(ks) > X for some X
+			for _, k := range ks {
+				bs.wantlist.Add(k)
+			}
 			if len(ks) == 0 {
 				log.Warning("Received batch request for zero blocks")
 				continue