metrics.rs

//! Example demonstrating the pluto-p2p metrics functionality.
//!
//! To run this example, run the local Prometheus and Grafana containers:
//! ```bash
//! docker compose -f test-infra/docker-compose.yml up -d
//! ```
//!
//! Then run the example:
//! ```bash
//! cargo run --example metrics -p pluto-p2p
//! ```
//!
//! Metrics will be available in Grafana at http://localhost:3000.

use std::net::SocketAddr;

use pluto_p2p::metrics::{
    ConnectionType, Direction, P2P_METRICS, PeerConnectionLabels, PeerNetworkLabels,
    PeerStreamLabels, Protocol, RelayConnectionLabels,
};
use vise_exporter::MetricsExporter;

#[tokio::main]
async fn main() {
    let bind_address = SocketAddr::from(([0, 0, 0, 0], 9464));

    let exporter = MetricsExporter::default()
        .bind(bind_address)
        .await
        .expect("Failed to bind metrics exporter");
    tokio::spawn(async move {
        exporter
            .start()
            .await
            .expect("Failed to start metrics exporter");
    });

    P2P_METRICS.ping_latency_secs["rust"].observe(1.0);
    P2P_METRICS.ping_error_total["rust"].inc();
    P2P_METRICS.ping_success["rust"].set(1);
    P2P_METRICS.reachability_status.set(1);
    P2P_METRICS.relay_connections["rust"].set(1);
    P2P_METRICS.peer_connection_types
        [&PeerConnectionLabels::new("rust", ConnectionType::Direct, Protocol::Tcp)]
        .set(1);
    P2P_METRICS.relay_connection_types
        [&RelayConnectionLabels::new("rust", ConnectionType::Direct, Protocol::Tcp)]
        .set(1);
    P2P_METRICS.peer_streams[&PeerStreamLabels::new("rust", Direction::Inbound, Protocol::Tcp)]
        .set(1);
    P2P_METRICS.peer_connection_total["rust"].inc();
    P2P_METRICS.peer_network_receive_bytes_total[&PeerNetworkLabels::new("rust", Protocol::Tcp)]
        .inc();
    P2P_METRICS.peer_network_sent_bytes_total[&PeerNetworkLabels::new("rust", Protocol::Tcp)].inc();

    // Wait for 20 seconds to see the logs in Loki
    std::thread::sleep(std::time::Duration::from_secs(20));
}