fix: peers request metric

harshavamsi · May 9, 2022 · 264e35b · 264e35b
1 parent a2ff73e
commit 264e35b
Showing 1 changed file with 8 additions and 5 deletions.
diff --git a/network/src/peer.rs b/network/src/peer.rs
@@ -434,6 +434,13 @@ impl<N: Network, E: Environment> Peer<N, E> {
 
                                     is_ready_to_send
                                 }
+                                Message::PeerResponse(_, _rtt_start) => {
+                                    // Stop the clock on internal RTT.
+                                    #[cfg(any(feature = "test", feature = "prometheus"))]
+                                    metrics::histogram!(metrics::internal_rtt::PEERS_REQUEST, _rtt_start.expect("rtt should be present with metrics enabled").elapsed());
+
+                                    true
+                                }
                                 _ => true,
                             };
                             // Send the message if it is ready.
@@ -549,15 +556,11 @@ impl<N: Network, E: Environment> Peer<N, E> {
                                         warn!("[PeerRequest] {}", error);
                                     }
                                 }
-                                Message::PeerResponse(peer_ips, _rtt_start) => {
+                                Message::PeerResponse(peer_ips, _) => {
                                     // Adds the given peer IPs to the list of candidate peers.
                                     if let Err(error) = peers_router.send(PeersRequest::ReceivePeerResponse(peer_ips)).await {
                                         warn!("[PeerResponse] {}", error);
                                     }
-
-                                    // Stop the clock on internal RTT.
-                                    #[cfg(any(feature = "test", feature = "prometheus"))]
-                                    metrics::histogram!(metrics::internal_rtt::PEERS_REQUEST, _rtt_start.expect("rtt should be present with metrics enabled").elapsed());
                                 }
                                 Message::Ping(version, fork_depth, node_type, status, block_hash, block_header) => {
                                     // Ensure the message protocol version is not outdated.