apollo_network_benchmark: setup Prometheus metrics exporter

Author: sirandreww-starkware

Cargo.lock

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

crates/apollo_network_benchmark/Cargo.toml

@@ -1,6 +1,8 @@
 //! Runs a node that stress tests the p2p communication of the network.
+use std::net::{Ipv4Addr, SocketAddr, SocketAddrV4};
 
 use clap::Parser;
+use metrics_exporter_prometheus::PrometheusBuilder;
 use tracing::Level;
 
 #[cfg(test)]
@@ -29,5 +31,13 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
 
     println!("Starting network stress test with args:\n{args:?}");
 
+    // Set up metrics
+    let builder = PrometheusBuilder::new().with_http_listener(SocketAddr::V4(SocketAddrV4::new(
+        Ipv4Addr::UNSPECIFIED,
+        args.runner.metric_port,
+    )));
+
+    builder.install().expect("Failed to install prometheus recorder/exporter");
+
     Ok(())
 }

crates/apollo_network_benchmark/src/bin/broadcast_network_stress_test_node/main.rs

@@ -3,7 +3,11 @@ use clap::Parser;
 #[derive(Parser, Debug, Clone)]
 #[command(version, about, long_about = None)]
 /// Arguments from the runner, not meant to be set by the user.
-pub struct RunnerArgs {}
+pub struct RunnerArgs {
+    /// The port to run the Prometheus metrics server on
+    #[arg(long, env)]
+    pub metric_port: u16,
+}
 
 #[derive(Parser, Debug, Clone)]
 #[command(version, about, long_about = None)]