Apply suggestions from code review

Co-Authored-By: dirkmc <[email protected]>
ipfs · Mar 16, 2020 · b63cf22 · b63cf22
1 parent 6d57007
commit b63cf22
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/internal/peermanager/peermanager.go b/internal/peermanager/peermanager.go
@@ -170,12 +170,12 @@ func (pm *PeerManager) SendCancels(ctx context.Context, cancelKs []cid.Cid) {
 	}
 }
 
-// CurrentWants returns the list of pending wants, want-haves and want-blocks.
+// CurrentWants returns the list of pending wants (both want-haves and want-blocks).
 func (pm *PeerManager) CurrentWants() []cid.Cid {
 	pm.pqLk.RLock()
 	defer pm.pqLk.RUnlock()
 
-	return pm.pwm.GetWantBlocks()
+	return pm.pwm.GetWants()
 }
 
 // CurrentWantBlocks returns the list of pending want-blocks

diff --git a/internal/peermanager/peerwantmanager.go b/internal/peermanager/peerwantmanager.go
@@ -191,7 +191,7 @@ func (pwm *peerWantManager) GetWantHaves() []cid.Cid {
 	return res.Keys()
 }
 
-// GetWants returns the set of all wants, want-blocks and want-haves.
+// GetWants returns the set of all wants (both want-blocks and want-haves).
 func (pwm *peerWantManager) GetWants() []cid.Cid {
 	res := cid.NewSet()