use crate::global_registry;
use prometheus_client::{
encoding::text::encode,
metrics::counter::Counter,
};
use std::ops::Deref;
#[derive(Default, Debug)]
pub struct ServicesMetrics {
pub busy: Counter,
pub idle: Counter,
}
impl ServicesMetrics {
pub fn register_service(service_name: &str) -> ServicesMetrics {
let reg =
regex::Regex::new("^[a-zA-Z_:][a-zA-Z0-9_:]*$").expect("It is a valid Regex");
if !reg.is_match(service_name) {
panic!("The service {} has incorrect name.", service_name);
}
let lifecycle = ServicesMetrics::default();
let mut lock = global_registry().registry.lock();
let mut encoded_bytes = String::new();
encode(&mut encoded_bytes, lock.deref())
.expect("Unable to decode service metrics");
let reg = regex::Regex::new(format!("\\b{}\\b", service_name).as_str())
.expect("It is a valid Regex");
if reg.is_match(encoded_bytes.as_str()) {
tracing::warn!("Service with '{}' name is already registered", service_name);
}
lock.register(
format!("{}_idle_ns", service_name),
format!("The idle time of the {} service", service_name),
lifecycle.idle.clone(),
);
lock.register(
format!("{}_busy_ns", service_name),
format!("The busy time of the {} service", service_name),
lifecycle.busy.clone(),
);
lifecycle
}
}
#[test]
fn register_success() {
ServicesMetrics::register_service("Foo");
ServicesMetrics::register_service("Bar");
}