feat(metrics): count the number of relayed circuit connections

maidsafe · Dec 23, 2024 · 4768674 · 4768674
1 parent 4e671d7
commit 4768674
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 0 deletions.
diff --git a/ant-networking/src/event/swarm.rs b/ant-networking/src/event/swarm.rs
@@ -105,6 +105,18 @@ impl SwarmDriver {
                     libp2p::relay::Event::ReservationTimedOut { src_peer_id } => {
                         self.connected_relay_clients.remove(&src_peer_id);
                     }
+                    libp2p::relay::Event::CircuitReqAccepted { .. } => {
+                        #[cfg(feature = "open-metrics")]
+                        if let Some(metrics_recorder) = &self.metrics_recorder {
+                            metrics_recorder.open_relayed_circuit_connections.inc();
+                        }
+                    }
+                    libp2p::relay::Event::CircuitClosed { .. } => {
+                        #[cfg(feature = "open-metrics")]
+                        if let Some(metrics_recorder) = &self.metrics_recorder {
+                            metrics_recorder.open_relayed_circuit_connections.dec();
+                        }
+                    }
                     _ => {}
                 }
             }

diff --git a/ant-networking/src/metrics/mod.rs b/ant-networking/src/metrics/mod.rs
@@ -42,6 +42,7 @@ pub(crate) struct NetworkMetricsRecorder {
     pub(crate) connected_peers: Gauge,
     pub(crate) estimated_network_size: Gauge,
     pub(crate) open_connections: Gauge,
+    pub(crate) open_relayed_circuit_connections: Gauge,
     pub(crate) peers_in_routing_table: Gauge,
     pub(crate) records_stored: Gauge,
 
@@ -98,6 +99,12 @@ impl NetworkMetricsRecorder {
             "The estimated number of nodes in the network calculated by the peers in our RT",
             estimated_network_size.clone(),
         );
+        let open_relayed_circuit_connections = Gauge::default();
+        sub_registry.register(
+            "open_relayed_circuit_connections",
+            "The number of active circuit connections that we are relaying between peers",
+            open_relayed_circuit_connections.clone(),
+        );
         let open_connections = Gauge::default();
         sub_registry.register(
             "open_connections",
@@ -214,6 +221,7 @@ impl NetworkMetricsRecorder {
             estimated_network_size,
             connected_peers,
             open_connections,
+            open_relayed_circuit_connections,
             peers_in_routing_table,
             relevant_records,
             max_records,