From 86fb07aed7d32ccd1a33f8bab0ea47aa3adbb359 Mon Sep 17 00:00:00 2001
From: Jeromy <jeromyj@gmail.com>
Date: Wed, 29 Apr 2015 01:36:47 -0700
Subject: [PATCH] try harder to not send duplicate blocks

---
 .../bitswap/decision/peer_request_queue.go    | 28 +++++++++++++++----
 exchange/bitswap/workers.go                   |  2 +-
 2 files changed, 24 insertions(+), 6 deletions(-)

diff --git a/exchange/bitswap/decision/peer_request_queue.go b/exchange/bitswap/decision/peer_request_queue.go
index e771ece0b..42928487d 100644
--- a/exchange/bitswap/decision/peer_request_queue.go
+++ b/exchange/bitswap/decision/peer_request_queue.go
@@ -46,7 +46,7 @@ func (tl *prq) Push(entry wantlist.Entry, to peer.ID) {
 	defer tl.lock.Unlock()
 	partner, ok := tl.partners[to]
 	if !ok {
-		partner = &activePartner{taskQueue: pq.New(wrapCmp(V1))}
+		partner = newActivePartner()
 		tl.pQueue.Push(partner)
 		tl.partners[to] = partner
 	}
@@ -57,12 +57,19 @@ func (tl *prq) Push(entry wantlist.Entry, to peer.ID) {
 		return
 	}
 
+	partner.activelk.Lock()
+	defer partner.activelk.Unlock()
+	_, ok = partner.activeBlocks[entry.Key]
+	if ok {
+		return
+	}
+
 	task := &peerRequestTask{
 		Entry:   entry,
 		Target:  to,
 		created: time.Now(),
 		Done: func() {
-			partner.TaskDone()
+			partner.TaskDone(entry.Key)
 			tl.lock.Lock()
 			tl.pQueue.Update(partner.Index())
 			tl.lock.Unlock()
@@ -93,7 +100,7 @@ func (tl *prq) Pop() *peerRequestTask {
 			continue // discarding tasks that have been removed
 		}
 
-		partner.StartTask()
+		partner.StartTask(out.Entry.Key)
 		partner.requests--
 		break // and return |out|
 	}
@@ -179,6 +186,8 @@ type activePartner struct {
 	activelk sync.Mutex
 	active   int
 
+	activeBlocks map[u.Key]struct{}
+
 	// requests is the number of blocks this peer is currently requesting
 	// request need not be locked around as it will only be modified under
 	// the peerRequestQueue's locks
@@ -191,6 +200,13 @@ type activePartner struct {
 	taskQueue pq.PQ
 }
 
+func newActivePartner() *activePartner {
+	return &activePartner{
+		taskQueue:    pq.New(wrapCmp(V1)),
+		activeBlocks: make(map[u.Key]struct{}),
+	}
+}
+
 // partnerCompare implements pq.ElemComparator
 func partnerCompare(a, b pq.Elem) bool {
 	pa := a.(*activePartner)
@@ -208,15 +224,17 @@ func partnerCompare(a, b pq.Elem) bool {
 }
 
 // StartTask signals that a task was started for this partner
-func (p *activePartner) StartTask() {
+func (p *activePartner) StartTask(k u.Key) {
 	p.activelk.Lock()
+	p.activeBlocks[k] = struct{}{}
 	p.active++
 	p.activelk.Unlock()
 }
 
 // TaskDone signals that a task was completed for this partner
-func (p *activePartner) TaskDone() {
+func (p *activePartner) TaskDone(k u.Key) {
 	p.activelk.Lock()
+	delete(p.activeBlocks, k)
 	p.active--
 	if p.active < 0 {
 		panic("more tasks finished than started!")
diff --git a/exchange/bitswap/workers.go b/exchange/bitswap/workers.go
index 4e2bf43b8..1fc59a214 100644
--- a/exchange/bitswap/workers.go
+++ b/exchange/bitswap/workers.go
@@ -11,7 +11,7 @@ import (
 	u "github.com/ipfs/go-ipfs/util"
 )
 
-var TaskWorkerCount = 16
+var TaskWorkerCount = 8
 
 func init() {
 	twc := os.Getenv("IPFS_BITSWAP_TASK_WORKERS")