Request-Response: Add method connected_peers

protocols/request-response/src/lib.rs

@@ -450,6 +450,11 @@ where
         }
     }
 
+    /// List of currently connected peers.
+    pub fn connected_peers(&self) -> Vec<&PeerId> {
+        self.connected.keys().collect()
+    }
+
     /// Checks whether an outbound request to the peer with the provided
     /// [`PeerId`] initiated by [`RequestResponse::send_request`] is still
     /// pending, i.e. waiting for a response.
@@ -593,6 +598,29 @@ where
         addresses
     }
 
+    fn inject_address_change(
+        &mut self,
+        peer: &PeerId,
+        conn: &ConnectionId,
+        _old: &ConnectedPoint,
+        new: &ConnectedPoint,
+    ) {
+        let new_address = match new {
+            ConnectedPoint::Dialer { address } => Some(address.clone()),
+            ConnectedPoint::Listener { .. } => None,
+        };
+        let connections = self
+            .connected
+            .get_mut(peer)
+            .expect("Address change can only happen on an established connection.");
+
+        let connection = connections
+            .iter_mut()
+            .find(|c| &c.id == conn)
+            .expect("Address change can only happen on an established connection.");
+        connection.address = new_address;
+    }
+
     fn inject_connected(&mut self, peer: &PeerId) {
         if let Some(pending) = self.pending_outbound_requests.remove(peer) {
             for request in pending {
@@ -686,7 +714,7 @@ where
                     self.pending_events
                         .push_back(NetworkBehaviourAction::GenerateEvent(
                             RequestResponseEvent::OutboundFailure {
-                                peer: peer,
+                                peer,
                                 request_id: request.request_id,
                                 error: OutboundFailure::DialFailure,
                             },

protocols/request-response/tests/ping.rs

@@ -122,6 +122,11 @@ fn ping_protocol() {
                     assert_eq!(&peer, &peer2_id);
                 }
                 SwarmEvent::Behaviour(e) => panic!("Peer1: Unexpected event: {:?}", e),
+                SwarmEvent::ConnectionEstablished { peer_id, .. } => {
+                    assert_eq!(peer_id, peer2_id);
+                    assert!(swarm1.behaviour().is_connected(&peer_id));
+                    assert_eq!(swarm1.behaviour().connected_peers(), vec![&peer_id]);
+                }
                 _ => {}
             }
         }
@@ -157,6 +162,11 @@ fn ping_protocol() {
                     }
                 }
                 SwarmEvent::Behaviour(e) => panic!("Peer2: Unexpected event: {:?}", e),
+                SwarmEvent::ConnectionEstablished { peer_id, .. } => {
+                    assert_eq!(peer_id, peer1_id);
+                    assert!(swarm2.behaviour().is_connected(&peer_id));
+                    assert_eq!(swarm2.behaviour().connected_peers(), vec![&peer_id]);
+                }
                 _ => {}
             }
         }