Proxy metrics prometheus server

proxy/Cargo.toml

@@ -18,3 +18,8 @@ dotenv = "0.15.0"
 tracing = "0.1.40"
 tracing-subscriber = "0.3.18"
 futures-util = "0.3.30"
+hyper = { version = "1.2.0", features = ["full"] }
+hyper-util = { version = "0.1.3", features = ["full"] }
+prometheus = "0.13.3"
+bytes = "1.5.0"
+http-body-util = "0.1.0"

proxy/src/config.rs

@@ -3,6 +3,7 @@ use std::{env, path::PathBuf};
 #[derive(Debug, Clone)]
 pub struct Config {
     pub proxy_addr: String,
+    pub prometheus_addr: String,
     pub ssl_crt_path: PathBuf,
     pub ssl_key_path: PathBuf,
     pub node_port: u16,
@@ -13,6 +14,7 @@ impl Config {
     pub fn new() -> Self {
         Self {
             proxy_addr: env::var("PROXY_ADDR").expect("PROXY_ADDR must be set"),
+            prometheus_addr: env::var("PROMETHEUS_ADDR").expect("PROMETHEUS_ADDR must be set"),
             ssl_crt_path: env::var("SSL_CRT_PATH")
                 .map(|e| e.into())
                 .expect("SSL_CRT_PATH must be set"),

proxy/src/main.rs

@@ -1,4 +1,6 @@
 use dotenv::dotenv;
+use metrics::Metrics;
+use prometheus::Registry;
 use regex::Regex;
 use std::{collections::HashMap, error::Error, fmt::Display, sync::Arc};
 use tokio::sync::RwLock;
@@ -8,7 +10,9 @@ use crate::config::Config;
 
 mod auth;
 mod config;
+mod metrics;
 mod proxy;
+mod utils;
 
 #[tokio::main]
 async fn main() -> Result<(), Box<dyn std::error::Error>> {
@@ -18,27 +22,31 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
     let state = Arc::new(RwLock::new(State::try_new()?));
 
     let auth = auth::start(state.clone());
+    let metrics = metrics::start(state.clone());
     let proxy_server = proxy::start(state.clone());
 
-    tokio::join!(auth, proxy_server);
+    tokio::join!(auth, metrics, proxy_server);
 
     Ok(())
 }
 
 #[derive(Debug, Clone)]
 pub struct State {
     config: Config,
+    metrics: Metrics,
     host_regex: Regex,
     consumers: HashMap<String, Consumer>,
 }
 impl State {
     pub fn try_new() -> Result<Self, Box<dyn Error>> {
         let config = Config::new();
+        let metrics = Metrics::try_new(Registry::default())?;
         let host_regex = Regex::new(r"(dmtr_[\w\d-]+)\.([\w]+)-([\w\d]+).+")?;
         let consumers = HashMap::new();
 
         Ok(Self {
             config,
+            metrics,
             host_regex,
             consumers,
         })