- initialize fetchers map with size of length pp.fetchers

- use ListPeers in test case.
vimeo · Dec 17, 2022 · 39d4a10 · 39d4a10
1 parent 1fdd87e
commit 39d4a10
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/peers.go b/peers.go
@@ -399,8 +399,8 @@ func (pp *PeerPicker) listPeers() map[string]RemoteFetcher {
 	pp.mu.RLock()
 	defer pp.mu.RUnlock()
 
-	// deep copy of pp.fetchers map.
-	fetchers := make(map[string]RemoteFetcher)
+	// deep copy of fetchers map with size equal to length of pp.fetchers.
+	fetchers := make(map[string]RemoteFetcher, len(pp.fetchers))
 	for p, f := range pp.fetchers {
 		fetchers[p] = f
 	}

diff --git a/peers_test.go b/peers_test.go
@@ -244,7 +244,7 @@ func TestPeersIncremental(t *testing.T) {
 					fetcherURIs[f.(*TestFetcher).uri] = struct{}{}
 				}
 
-				allFetchers := u.peerPicker.listPeers()
+				allFetchers := u.ListPeers()
 				allPeerIDs := make(map[string]struct{}, len(allFetchers))
 				allFetcherURIs := make(map[string]struct{}, len(allFetchers))
 				for peerID, fetcher := range allFetchers {