apollo_network_benchmark: setup tokio metrics exporter

Author: sirandreww-starkware

Cargo.lock

@@ -13,6 +13,7 @@ clap = { workspace = true, features = ["derive", "env"] }
 lazy_static.workspace = true
 metrics-exporter-prometheus.workspace = true
 tokio = { workspace = true, features = ["full", "sync"] }
+tokio-metrics = { workspace = true, features = ["metrics-rs-integration", "rt"] }
 tracing.workspace = true
 tracing-subscriber.workspace = true
 

crates/apollo_network_benchmark/Cargo.toml

@@ -1,8 +1,10 @@
 //! Runs a node that stress tests the p2p communication of the network.
 use std::net::{Ipv4Addr, SocketAddr, SocketAddrV4};
+use std::time::Duration;
 
 use clap::Parser;
 use metrics_exporter_prometheus::PrometheusBuilder;
+use tokio_metrics::RuntimeMetricsReporterBuilder;
 use tracing::Level;
 
 #[cfg(test)]
@@ -39,5 +41,12 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
 
     builder.install().expect("Failed to install prometheus recorder/exporter");
 
+    // Start the tokio runtime metrics reporter to automatically collect and export runtime metrics
+    tokio::spawn(
+        RuntimeMetricsReporterBuilder::default()
+            .with_interval(Duration::from_secs(1))
+            .describe_and_run(),
+    );
+
     Ok(())
 }