Skip to content

Commit

Permalink
stop using the peer.Set (#201)
Browse files Browse the repository at this point in the history
  • Loading branch information
marten-seemann authored Jun 27, 2022
1 parent 7108bf5 commit 29bf33f
Showing 1 changed file with 7 additions and 3 deletions.
10 changes: 7 additions & 3 deletions p2p/host/peerstore/pstoremem/addr_book.go
Original file line number Diff line number Diff line change
Expand Up @@ -147,17 +147,21 @@ func (mab *memoryAddrBook) gc() {

func (mab *memoryAddrBook) PeersWithAddrs() peer.IDSlice {
// deduplicate, since the same peer could have both signed & unsigned addrs
pidSet := peer.NewSet()
set := make(map[peer.ID]struct{})
for _, s := range mab.segments {
s.RLock()
for pid, amap := range s.addrs {
if len(amap) > 0 {
pidSet.Add(pid)
set[pid] = struct{}{}
}
}
s.RUnlock()
}
return pidSet.Peers()
peers := make(peer.IDSlice, 0, len(set))
for pid := range set {
peers = append(peers, pid)
}
return peers
}

// AddAddr calls AddAddrs(p, []ma.Multiaddr{addr}, ttl)
Expand Down

0 comments on commit 29bf33f

Please sign in to comment.