Formatting

Signed-off-by: Federico Torres <[email protected]>

Author: fedetorres93

examples/axum-utf-8.rs

@@ -5,15 +5,15 @@ use axum::http::{HeaderMap, StatusCode};
 use axum::response::{IntoResponse, Response};
 use axum::routing::get;
 use axum::Router;
+use prometheus_client::encoding::negotiate_escaping_scheme;
 use prometheus_client::encoding::text::encode;
+use prometheus_client::encoding::EscapingScheme::UnderscoreEscaping;
+use prometheus_client::encoding::ValidationScheme::UTF8Validation;
 use prometheus_client::metrics::counter::Counter;
 use prometheus_client::metrics::family::Family;
 use prometheus_client::registry::{Registry, RegistryBuilder};
 use std::sync::Arc;
 use tokio::sync::Mutex;
-use prometheus_client::encoding::EscapingScheme::UnderscoreEscaping;
-use prometheus_client::encoding::negotiate_escaping_scheme;
-use prometheus_client::encoding::ValidationScheme::UTF8Validation;
 
 #[derive(Debug)]
 pub struct Metrics {
@@ -22,7 +22,9 @@ pub struct Metrics {
 
 impl Metrics {
     pub fn inc_requests(&self, method: String) {
-        self.requests.get_or_create(&vec![("method.label".to_owned(), method)]).inc();
+        self.requests
+            .get_or_create(&vec![("method.label".to_owned(), method)])
+            .inc();
     }
 }
 
@@ -31,14 +33,15 @@ pub struct AppState {
     pub registry: Registry,
 }
 
-pub async fn metrics_handler(State(state): State<Arc<Mutex<AppState>>>, headers: HeaderMap) -> impl IntoResponse {
+pub async fn metrics_handler(
+    State(state): State<Arc<Mutex<AppState>>>,
+    headers: HeaderMap,
+) -> impl IntoResponse {
     let mut state = state.lock().await;
     let mut buffer = String::new();
     if let Some(accept) = headers.get("Accept") {
-        let escaping_scheme = negotiate_escaping_scheme(
-            accept.to_str().unwrap(),
-            state.registry.escaping_scheme()
-        );
+        let escaping_scheme =
+            negotiate_escaping_scheme(accept.to_str().unwrap(), state.registry.escaping_scheme());
         state.registry.set_escaping_scheme(escaping_scheme);
     }
     encode(&mut buffer, &state.registry).unwrap();
@@ -47,7 +50,8 @@ pub async fn metrics_handler(State(state): State<Arc<Mutex<AppState>>>, headers:
         .status(StatusCode::OK)
         .header(
             CONTENT_TYPE,
-            "application/openmetrics-text; version=1.0.0; charset=utf-8; escaping=".to_owned() + state.registry.escaping_scheme().as_str(),
+            "application/openmetrics-text; version=1.0.0; charset=utf-8; escaping=".to_owned()
+                + state.registry.escaping_scheme().as_str(),
         )
         .body(Body::from(buffer))
         .unwrap()
@@ -69,9 +73,11 @@ async fn main() {
             .with_escaping_scheme(UnderscoreEscaping)
             .build(),
     };
-    state
-        .registry
-        .register("requests.count", "Count of requests", metrics.requests.clone());
+    state.registry.register(
+        "requests.count",
+        "Count of requests",
+        metrics.requests.clone(),
+    );
     let metrics = Arc::new(Mutex::new(metrics));
     let state = Arc::new(Mutex::new(state));
 

src/encoding.rs

@@ -797,16 +797,25 @@ fn is_valid_legacy_prefix(prefix: Option<&Prefix>) -> bool {
     }
 }
 
-fn is_quoted_metric_name(name: &str, prefix: Option<&Prefix>, validation_scheme: &ValidationScheme) -> bool {
-    *validation_scheme == ValidationScheme::UTF8Validation && (!is_valid_legacy_metric_name(name) || !is_valid_legacy_prefix(prefix))
+fn is_quoted_metric_name(
+    name: &str,
+    prefix: Option<&Prefix>,
+    validation_scheme: &ValidationScheme,
+) -> bool {
+    *validation_scheme == ValidationScheme::UTF8Validation
+        && (!is_valid_legacy_metric_name(name) || !is_valid_legacy_prefix(prefix))
 }
 
 fn is_valid_legacy_label_name(label_name: &str) -> bool {
     if label_name.is_empty() {
         return false;
     }
     for (i, b) in label_name.chars().enumerate() {
-        if !((b >= 'a' && b <= 'z') || (b >= 'A' && b <= 'Z') || b == '_' || (b >= '0' && b <= '9' && i > 0)) {
+        if !((b >= 'a' && b <= 'z')
+            || (b >= 'A' && b <= 'Z')
+            || b == '_'
+            || (b >= '0' && b <= '9' && i > 0))
+        {
             return false;
         }
     }
@@ -901,7 +910,10 @@ fn escape_name(name: &str, scheme: &EscapingScheme) -> String {
 }
 
 /// Returns the escaping scheme to use based on the given header.
-pub fn negotiate_escaping_scheme(header: &str, default_escaping_scheme: EscapingScheme) -> EscapingScheme {
+pub fn negotiate_escaping_scheme(
+    header: &str,
+    default_escaping_scheme: EscapingScheme,
+) -> EscapingScheme {
     if header.contains("underscores") {
         return EscapingScheme::UnderscoreEscaping;
     }