Skip to content

Commit 6ddf9b4

Browse files
refactor(libp2p): rename underlays to overlayToPeerID in peer registry (#5337)
1 parent 9f2f7aa commit 6ddf9b4

File tree

1 file changed

+8
-8
lines changed

1 file changed

+8
-8
lines changed

pkg/p2p/libp2p/peer.go

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@ import (
1818
)
1919

2020
type peerRegistry struct {
21-
underlays map[string]libp2ppeer.ID // map overlay address to underlay peer id
21+
overlayToPeerID map[string]libp2ppeer.ID // map overlay address to underlay peer id
2222
overlays map[libp2ppeer.ID]swarm.Address // map underlay peer id to overlay address
2323
full map[libp2ppeer.ID]bool // map to track whether a node is full or light node (true=full)
2424
bee260Compatibility map[libp2ppeer.ID]bool // map to track bee260 backward compatibility
@@ -37,7 +37,7 @@ type disconnecter interface {
3737

3838
func newPeerRegistry() *peerRegistry {
3939
return &peerRegistry{
40-
underlays: make(map[string]libp2ppeer.ID),
40+
overlayToPeerID: make(map[string]libp2ppeer.ID),
4141
overlays: make(map[libp2ppeer.ID]swarm.Address),
4242
full: make(map[libp2ppeer.ID]bool),
4343
bee260Compatibility: make(map[libp2ppeer.ID]bool),
@@ -77,7 +77,7 @@ func (r *peerRegistry) Disconnected(_ network.Network, c network.Conn) {
7777
delete(r.connections, peerID)
7878
overlay := r.overlays[peerID]
7979
delete(r.overlays, peerID)
80-
delete(r.underlays, overlay.ByteString())
80+
delete(r.overlayToPeerID, overlay.ByteString())
8181
for _, cancel := range r.streams[peerID] {
8282
cancel()
8383
}
@@ -146,12 +146,12 @@ func (r *peerRegistry) addIfNotExists(c network.Conn, overlay swarm.Address, ful
146146
// this is solving a case of multiple underlying libp2p connections for the same peer
147147
r.connections[peerID][c] = struct{}{}
148148

149-
if _, exists := r.underlays[overlay.ByteString()]; exists {
149+
if _, exists := r.overlayToPeerID[overlay.ByteString()]; exists {
150150
return true
151151
}
152152

153153
r.streams[peerID] = make(map[network.Stream]context.CancelFunc)
154-
r.underlays[overlay.ByteString()] = peerID
154+
r.overlayToPeerID[overlay.ByteString()] = peerID
155155
r.overlays[peerID] = overlay
156156
r.full[peerID] = full
157157
return false
@@ -160,7 +160,7 @@ func (r *peerRegistry) addIfNotExists(c network.Conn, overlay swarm.Address, ful
160160

161161
func (r *peerRegistry) peerID(overlay swarm.Address) (peerID libp2ppeer.ID, found bool) {
162162
r.mu.RLock()
163-
peerID, found = r.underlays[overlay.ByteString()]
163+
peerID, found = r.overlayToPeerID[overlay.ByteString()]
164164
r.mu.RUnlock()
165165
return peerID, found
166166
}
@@ -223,9 +223,9 @@ func (r *peerRegistry) isConnected(peerID libp2ppeer.ID, remoteAddr ma.Multiaddr
223223

224224
func (r *peerRegistry) remove(overlay swarm.Address) (found, full bool, peerID libp2ppeer.ID) {
225225
r.mu.Lock()
226-
peerID, found = r.underlays[overlay.ByteString()]
226+
peerID, found = r.overlayToPeerID[overlay.ByteString()]
227227
delete(r.overlays, peerID)
228-
delete(r.underlays, overlay.ByteString())
228+
delete(r.overlayToPeerID, overlay.ByteString())
229229
delete(r.connections, peerID)
230230
for _, cancel := range r.streams[peerID] {
231231
cancel()

0 commit comments

Comments
 (0)