feat(net/peer): set rpc added peer as static (#9201)

Signed-off-by: jsvisa <delweng@gmail.com>
This commit is contained in:
Delweng
2024-07-01 18:23:28 +08:00
committed by GitHub
parent 9cd377881c
commit d4fa9defbd
3 changed files with 13 additions and 5 deletions

View File

@ -562,7 +562,7 @@ where
}
}
NetworkHandleMessage::RemovePeer(peer_id, kind) => {
self.swarm.state_mut().remove_peer(peer_id, kind);
self.swarm.state_mut().remove_peer_kind(peer_id, kind);
}
NetworkHandleMessage::DisconnectPeer(peer_id, reason) => {
self.swarm.sessions_mut().disconnect(peer_id, reason);

View File

@ -278,9 +278,10 @@ where
self.peers_manager.add_peer_kind(peer_id, kind, addr, None)
}
pub(crate) fn remove_peer(&mut self, peer_id: PeerId, kind: PeerKind) {
/// Removes a peer and its address with the given kind from the peerset.
pub(crate) fn remove_peer_kind(&mut self, peer_id: PeerId, kind: PeerKind) {
match kind {
PeerKind::Basic => self.peers_manager.remove_peer(peer_id),
PeerKind::Basic | PeerKind::Static => self.peers_manager.remove_peer(peer_id),
PeerKind::Trusted => self.peers_manager.remove_peer_from_trusted_set(peer_id),
}
}