correctly buffer work messages

whyrusleeping · whyrusleeping · commit 006b1249fc40 · 2015-05-11T23:22:23.000-07:00
diff --git a/exchange/bitswap/peermanager.go b/exchange/bitswap/peermanager.go
@@ -75,7 +75,7 @@ func (pm *PeerManager) startPeerHandler(p peer.ID) {
 
 	mq := new(msgQueue)
 	mq.done = make(chan struct{})
-	mq.work = make(chan struct{})
+	mq.work = make(chan struct{}, 1)
 	mq.p = p
 
 	pm.peers[p] = mq
@@ -112,7 +112,7 @@ func (pm *PeerManager) runQueue(mq *msgQueue) {
 			mq.wlmsg = nil
 			mq.lk.Unlock()
 
-			if !wlm.Empty() {
+			if wlm != nil && !wlm.Empty() {
 				// send wantlist updates
 				err = pm.network.SendMessage(context.TODO(), mq.p, wlm)
 				if err != nil {
