Fix again

tomaka · Apr 17, 2020 · 80c0d3d · 80c0d3d
1 parent c10ef27
commit 80c0d3d
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/swarm/src/lib.rs b/swarm/src/lib.rs
@@ -730,9 +730,15 @@ where TBehaviour: NetworkBehaviour<ProtocolsHandler = THandler>,
                         } else {
                             log::trace!("Condition for new dialing attempt to {:?} not met: {:?}",
                                 peer_id, condition);
-                            if let Some(mut peer) = this.network.peer(peer_id.clone()).into_dialing() {
-                                let addrs = this.behaviour.addresses_of_peer(peer.id());
-                                peer.connection().add_addresses(addrs);
+                            match this.network.peer(peer_id.clone()) {
+                                Peer::Dialing(mut peer) => {
+                                    let addrs = this.behaviour.addresses_of_peer(peer.id());
+                                    peer.connection().add_addresses(addrs);
+                                },
+                                Peer::Connected(_) => {},
+                                Peer::Disconnected(..) | Peer::Local => {
+                                    this.behaviour.inject_dial_failure(&peer_id);
+                                }
                             }
                         };
                     }