feature: setup basic actix metrics server

Author: vid277

crates/grpc/Cargo.toml

@@ -18,6 +18,7 @@ tokio = { workspace = true, features = ["full"] }
 tonic = { workspace = true }
 types = { path = "../types" }
 tracing = { workspace = true }
+metrics = "0.24"
 
 [lints]
 workspace = true

crates/grpc/src/grpc_handler.rs

@@ -91,9 +91,15 @@ impl NodeControl for NodeControlService {
         &self,
         request: Request<CheckBalanceRequest>,
     ) -> Result<Response<CheckBalanceResponse>, Status> {
-        let request = request.into_inner();
-        let response = grpc_operator::check_balance(&self.network, request).await?;
-        Ok(Response::new(response))
+        let result = {
+            let request = request.into_inner();
+            grpc_operator::check_balance(&self.network, request).await
+        };
+
+        match result {
+            Ok(resp) => Ok(Response::new(resp)),
+            Err(status) => Err(status),
+        }
     }
 
     async fn get_chain_info(

crates/node/Cargo.toml

@@ -61,6 +61,9 @@ num-traits = "0.2.19"
 abci = { path = "../abci" }
 messenger = { path = "../../messenger" }
 grpc = { path = "../grpc" }
+metrics = "0.24"
+metrics-exporter-prometheus = { version = "0.17", features = ["http-listener"] }
+actix-web = "4"
 
 [dev-dependencies]
 tempfile = "3.8"

crates/node/src/lib.rs

@@ -36,7 +36,6 @@ pub struct PeerData {
 
 pub struct NodeState<N: Network, W: Wallet> {
     pub handlers: Vec<Box<dyn Handler<N, W>>>,
-
     pub peer_id: PeerId,
     pub peers: HashSet<PeerId>,
 

crates/node/src/start_node.rs

@@ -6,15 +6,18 @@ use crate::{
     NodeConfig, NodeState, key_manager::load_and_decrypt_keypair, swarm_manager::build_swarm,
     wallet::TaprootWallet,
 };
+use actix_web::{App, HttpResponse, HttpServer, web};
 use bitcoin::Network;
 use grpc::grpc_handler::NodeControlService;
+use metrics_exporter_prometheus::{PrometheusBuilder, PrometheusHandle};
 use std::path::{Path, PathBuf};
+use std::sync::Arc;
 use tokio::sync::broadcast;
 use tonic::transport::Server;
 use tracing_appender::rolling::{RollingFileAppender, Rotation};
 use tracing_subscriber::{EnvFilter, fmt, prelude::*};
 
-use std::sync::Arc;
+type PrometheusHandler = Arc<PrometheusHandle>;
 
 pub async fn start_node(
     config: NodeConfig,
@@ -76,6 +79,43 @@ pub async fn start_node(
         tracing::info!("Logging initialized with console output only");
     }
 
+    let prometheus_handle: Arc<PrometheusHandle> = {
+        let builder = PrometheusBuilder::new();
+        Arc::new(
+            builder
+                .install_recorder()
+                .expect("failed to install Prometheus recorder"),
+        )
+    };
+
+    let metrics_server_handle = tokio::spawn(async move {
+        async fn metrics_endpoint(handler: web::Data<PrometheusHandler>) -> HttpResponse {
+            metrics::counter!("metrics_scrape_requests_total").increment(1);
+            HttpResponse::Ok()
+                .content_type("text/plain")
+                .body(handler.render())
+        }
+
+        async fn health_endpoint() -> HttpResponse {
+            HttpResponse::Ok().content_type("text/plain").body("OK")
+        }
+
+        tracing::info!("Starting metrics server on 0.0.0.0:8080");
+
+        let server = HttpServer::new(move || {
+            App::new()
+                .app_data(web::Data::new(prometheus_handle.clone()))
+                .route("/metrics", web::get().to(metrics_endpoint))
+                .route("/health", web::get().to(health_endpoint))
+        })
+        .bind(("0.0.0.0", 8080))
+        .expect("Failed to bind metrics endpoint");
+
+        tracing::info!("Metrics server bound successfully, starting to serve");
+
+        server.run().await.expect("Metrics server failed");
+    });
+
     let keypair = match load_and_decrypt_keypair(&config) {
         Ok(kp) => kp,
         Err(e) => {
@@ -218,6 +258,12 @@ pub async fn start_node(
                 Err(e) => tracing::error!("Chain interface error: {}", e),
             }
         }
+        result = metrics_server_handle => {
+            match result {
+                Ok(()) => tracing::info!("Metrics server stopped"),
+                Err(e) => tracing::error!("Metrics server error: {}", e),
+            }
+        }
     }
 
     Ok(())

docker-compose.yaml

@@ -11,6 +11,7 @@ services:
     entrypoint: "/app/cli run --key-file-path /app/configs/node_1.json --config-file-path /app/configs/node_1.yaml --use-mock-oracle"
     ports:
       - "50051:50051"
+      - "8081:8080"
     volumes:
       - ./keys/node_1/node_1.json:/app/configs/node_1.json
       - ./keys/node_1/node_1.yaml:/app/configs/node_1.yaml
@@ -29,6 +30,7 @@ services:
     entrypoint: "/app/cli run --key-file-path /app/configs/node_2.json --config-file-path /app/configs/node_2.yaml --use-mock-oracle"
     ports:
       - "50052:50051"
+      - "8082:8080"
     volumes:
       - ./keys/node_2/node_2.json:/app/configs/node_2.json
       - ./keys/node_2/node_2.yaml:/app/configs/node_2.yaml
@@ -47,6 +49,7 @@ services:
     entrypoint: "/app/cli run --key-file-path /app/configs/node_3.json --config-file-path /app/configs/node_3.yaml --use-mock-oracle"
     ports:
       - "50053:50051"
+      - "8083:8080"
     volumes:
       - ./keys/node_3/node_3.json:/app/configs/node_3.json
       - ./keys/node_3/node_3.yaml:/app/configs/node_3.yaml
@@ -65,6 +68,7 @@ services:
     entrypoint: "/app/cli run --key-file-path /app/configs/node_4.json --config-file-path /app/configs/node_4.yaml --use-mock-oracle"
     ports:
       - "50054:50051"
+      - "8084:8080"
     volumes:
       - ./keys/node_4/node_4.json:/app/configs/node_4.json
       - ./keys/node_4/node_4.yaml:/app/configs/node_4.yaml
@@ -83,6 +87,7 @@ services:
     entrypoint: "/app/cli run --key-file-path /app/configs/node_5.json --config-file-path /app/configs/node_5.yaml --use-mock-oracle"
     ports:
       - "50055:50051"
+      - "8085:8080"
     volumes:
       - ./keys/node_5/node_5.json:/app/configs/node_5.json
       - ./keys/node_5/node_5.yaml:/app/configs/node_5.yaml