Added disconnect for all connections

Author: 0xMimir

p2p/src/disconnection/p2p_disconnection_reducer.rs

@@ -24,23 +24,30 @@ impl P2pDisconnectedState {
             P2pDisconnectionAction::Init { peer_id, reason } => {
                 #[cfg(feature = "p2p-libp2p")]
                 if p2p_state.is_libp2p_peer(&peer_id) {
-                    if let Some((&addr, _)) = p2p_state
+                    let connections = p2p_state
                         .network
                         .scheduler
                         .connections
                         .iter()
-                        .find(|(_, conn_state)| conn_state.peer_id() == Some(&peer_id))
-                    {
-                        let Some(peer) = p2p_state.peers.get_mut(&peer_id) else {
-                            bug_condition!("Invalid state for: `P2pDisconnectionAction::Finish`");
-                            return Ok(());
-                        };
-                        peer.status = P2pPeerStatus::Disconnecting { time: meta.time() };
+                        .filter(|(_, conn_state)| conn_state.peer_id() == Some(&peer_id))
+                        .map(|(addr, _)| *addr)
+                        .collect::<Vec<_>>();
 
-                        let dispatcher = state_context.into_dispatcher();
-                        dispatcher.push(P2pNetworkSchedulerAction::Disconnect { addr, reason });
-                        dispatcher.push(P2pDisconnectionAction::Finish { peer_id });
+                    let Some(peer) = p2p_state.peers.get_mut(&peer_id) else {
+                        bug_condition!("Invalid state for: `P2pDisconnectionAction::Finish`");
+                        return Ok(());
+                    };
+                    peer.status = P2pPeerStatus::Disconnecting { time: meta.time() };
+
+                    let dispatcher = state_context.into_dispatcher();
+                    for addr in connections {
+                        dispatcher.push(P2pNetworkSchedulerAction::Disconnect {
+                            addr,
+                            reason: reason.clone(),
+                        });
                     }
+
+                    dispatcher.push(P2pDisconnectionAction::Finish { peer_id });
                     return Ok(());
                 }