- add: metrics exporting

packet-service/src/main.rs

@@ -6,6 +6,8 @@ use crate::metrics::{ACTIVE_CONNECTIONS, PACKETS_RECEIVED};
 use crate::packet::Packet;
 use crate::router::PacketRouter;
 use dotenv::dotenv;
+use prometheus::{self, Encoder, TextEncoder};
+use prometheus_exporter;
 use std::collections::HashMap;
 use std::env;
 use std::error::Error;
@@ -66,6 +68,7 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
     // Set the gRPC server address
     let addr = env::var("LISTEN_ADDR").unwrap_or_else(|_| "0.0.0.0".to_string());
     let port = env::var("PACKET_SERVICE_PORT").unwrap_or_else(|_| "4000".to_string());
+    let metrics_port = env::var("PACKET_METRICS_PORT").unwrap_or_else(|_| "4001".to_string());
     let health_port = env::var("HEALTH_CHECK_PORT").unwrap_or_else(|_| "8082".to_string());
 
     let consul_url = env::var("CONSUL_URL").unwrap_or_else(|_| "http://127.0.0.1:8500".to_string());
@@ -93,7 +96,7 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
         meta,
         &health_check_url,
     )
-    .await?;
+        .await?;
 
     // Start health-check endpoint
     consul_registration::start_health_check(addr.as_str()).await?;
@@ -153,6 +156,9 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
         }
     });
 
+    let binding = format!("{}:{}", &addr, metrics_port);
+    prometheus_exporter::start(binding.parse().unwrap()).unwrap();
+
     utils::signal_handler::wait_for_signal().await;
 
     consul_registration::deregister_service(&consul_url, service_id.as_str())