From 3d5531739961c64bbbc1f268d0d1a7fa37f6f77d Mon Sep 17 00:00:00 2001 From: Pierre Krieger Date: Fri, 19 Apr 2019 11:10:53 +0200 Subject: [PATCH] &mut self -> &mut Self --- core/src/swarm/swarm.rs | 12 +++++++----- 1 file changed, 7 insertions(+), 5 deletions(-) diff --git a/core/src/swarm/swarm.rs b/core/src/swarm/swarm.rs index 127daebfaf8..9cf9e0347d6 100644 --- a/core/src/swarm/swarm.rs +++ b/core/src/swarm/swarm.rs @@ -219,14 +219,16 @@ where TBehaviour: NetworkBehaviour, /// /// Any incoming connection and any dialing attempt will immediately be rejected. /// This function has no effect is the peer is already banned. - pub fn ban_peer_id(&mut self, peer_id: PeerId) { - self.banned_peers.insert(peer_id.clone()); - self.raw_swarm.peer(peer_id).into_connected().map(|c| c.close()); + pub fn ban_peer_id(me: &mut Self, peer_id: PeerId) { + me.banned_peers.insert(peer_id.clone()); + if let Some(c) = me.raw_swarm.peer(peer_id).into_connected() { + c.close(); + } } /// Unbans a peer. - pub fn unban_peer_id(&mut self, peer_id: PeerId) { - self.banned_peers.remove(&peer_id); + pub fn unban_peer_id(me: &mut Self, peer_id: PeerId) { + me.banned_peers.remove(&peer_id); } }