use statsd prefix in metrics endpoint

Author: mcharytoniuk

src/balancer/management_service/app_data.rs

@@ -18,4 +18,5 @@ pub struct AppData {
     pub generate_tokens_sender_collection: Arc<GenerateTokensSenderCollection>,
     pub model_metadata_sender_collection: Arc<ModelMetadataSenderCollection>,
     pub state_database: Arc<dyn StateDatabase>,
+    pub statsd_prefix: String,
 }

src/balancer/management_service/http_route/get_metrics.rs

@@ -24,19 +24,20 @@ async fn respond(app_data: Data<AppData>) -> Result<impl Responder, Box<dyn Erro
         .buffered_request_manager
         .buffered_request_counter
         .get();
+    let statsd_prefix = app_data.statsd_prefix.clone();
 
     let metrics_response = formatdoc! {"
-        # HELP paddler_slots_processing Number of processing slots
-        # TYPE paddler_slots_processing gauge
-        paddler_slots_processing {slots_processing}
+        # HELP {statsd_prefix}slots_processing Number of processing slots
+        # TYPE {statsd_prefix}slots_processing gauge
+        {statsd_prefix}slots_processing {slots_processing}
 
-        # HELP paddler_slots_total Number of total slots
-        # TYPE paddler_slots_total gauge
-        paddler_slots_total {slots_total}
+        # HELP {statsd_prefix}slots_total Number of total slots
+        # TYPE {statsd_prefix}slots_total gauge
+        {statsd_prefix}slots_total {slots_total}
 
-        # HELP paddler_requests_buffered Number of buffered requests
-        # TYPE paddler_requests_buffered gauge
-        paddler_requests_buffered {buffered_requests_count}
+        # HELP {statsd_prefix}requests_buffered Number of buffered requests
+        # TYPE {statsd_prefix}requests_buffered gauge
+        {statsd_prefix}requests_buffered {buffered_requests_count}
     "};
 
     Ok(HttpResponse::Ok()

src/balancer/management_service/mod.rs

@@ -38,6 +38,7 @@ pub struct ManagementService {
     pub generate_tokens_sender_collection: Arc<GenerateTokensSenderCollection>,
     pub model_metadata_sender_collection: Arc<ModelMetadataSenderCollection>,
     pub state_database: Arc<dyn StateDatabase>,
+    pub statsd_prefix: String,
     #[cfg(feature = "web_admin_panel")]
     pub web_admin_panel_service_configuration: Option<WebAdminPanelServiceConfiguration>,
 }
@@ -70,6 +71,7 @@ impl Service for ManagementService {
             generate_tokens_sender_collection: self.generate_tokens_sender_collection.clone(),
             model_metadata_sender_collection: self.model_metadata_sender_collection.clone(),
             state_database: self.state_database.clone(),
+            statsd_prefix: self.statsd_prefix.clone(),
         });
 
         HttpServer::new(move || {

src/balancer/statsd_service/configuration.rs

@@ -1,6 +1,7 @@
 use std::net::SocketAddr;
 use std::time::Duration;
 
+#[derive(Clone)]
 pub struct Configuration {
     pub statsd_addr: SocketAddr,
     pub statsd_prefix: String,

src/cmd/balancer.rs

@@ -172,6 +172,7 @@ impl Handler for Balancer {
             generate_tokens_sender_collection,
             model_metadata_sender_collection,
             state_database: state_database.clone(),
+            statsd_prefix: self.statsd_prefix.clone(),
             #[cfg(feature = "web_admin_panel")]
             web_admin_panel_service_configuration: self.get_web_admin_panel_service_configuration(),
         });