diff --git a/router/local_peer.go b/router/local_peer.go index 3648038c..93f13ca9 100644 --- a/router/local_peer.go +++ b/router/local_peer.go @@ -261,28 +261,6 @@ func (peer *LocalPeer) handleSendProtocolMsg(m ProtocolMsg) { }) } -func (peer *LocalPeer) addConnection(conn Connection) { - peer.Lock() - peer.connections[conn.Remote().Name] = conn - peer.Unlock() -} - -func (peer *LocalPeer) deleteConnection(conn Connection) { - established := conn.Established() - peer.Lock() - delete(peer.connections, conn.Remote().Name) - if established { - peer.version += 1 - } - peer.Unlock() -} - -func (peer *LocalPeer) connectionEstablished(conn Connection) { - peer.Lock() - peer.version += 1 - peer.Unlock() -} - func (peer *LocalPeer) broadcastPeerUpdate(peers ...*Peer) { peer.Router.Routes.Recalculate() peer.handleSendProtocolMsg(ProtocolMsg{ProtocolUpdate, EncodePeers(append(peers, peer.Peer)...)}) diff --git a/router/peer.go b/router/peer.go index bffaa069..45e20dfd 100644 --- a/router/peer.go +++ b/router/peer.go @@ -86,6 +86,28 @@ func (peer *Peer) SetVersionAndConnections(version uint64, connections map[PeerN peer.connections = connections } +func (peer *LocalPeer) addConnection(conn Connection) { + peer.Lock() + defer peer.Unlock() + peer.connections[conn.Remote().Name] = conn +} + +func (peer *LocalPeer) deleteConnection(conn Connection) { + established := conn.Established() + peer.Lock() + defer peer.Unlock() + delete(peer.connections, conn.Remote().Name) + if established { + peer.version += 1 + } +} + +func (peer *LocalPeer) connectionEstablished(conn Connection) { + peer.Lock() + defer peer.Unlock() + peer.version += 1 +} + // Calculate the routing table from this peer to all peers reachable // from it, returning a "next hop" map of PeerNameX -> PeerNameY, // which says "in order to send a message to X, the peer should send