edward/mangadex-home-rs
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

Use macro for metrics generation

Browse source
This commit is contained in:
Edward Shen 2021-05-23 19:05:57 -04:00
parent 579d316945
commit b8c12b463f
Signed by: edward
GPG key ID: 19182661E818369F
1 changed files with 45 additions and 36 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

87
src/metrics.rs
View file

@ -1,47 +1,56 @@
use once_cell::sync::Lazy; use once_cell::sync::Lazy;
use prometheus::{register_int_counter, IntCounter}; use prometheus::{register_int_counter, IntCounter};
pub static CACHE_HIT_COUNTER: Lazy<IntCounter> = macro_rules! init_counters {
Lazy::new(|| register_int_counter!("cache_hit", "The number of cache hits").unwrap()); ($(($counter:ident, $ty:ty, $name:literal, $desc:literal),)*) => {
$(
pub static CACHE_MISS_COUNTER: Lazy<IntCounter> = pub static $counter: Lazy<$ty> = Lazy::new(|| {
Lazy::new(|| register_int_counter!("cache_miss", "The number of cache misses").unwrap()); register_int_counter!($name, $desc).unwrap()
pub static REQUESTS_TOTAL_COUNTER: Lazy<IntCounter> = Lazy::new(|| {
register_int_counter!("requests_total", "The total number of requests served.").unwrap()
});
pub static REQUESTS_DATA_COUNTER: Lazy<IntCounter> = Lazy::new(|| {
register_int_counter!(
"requests_data",
"The number of requests served from the /data endpoint."
)
.unwrap()
});
pub static REQUESTS_DATA_SAVER_COUNTER: Lazy<IntCounter> = Lazy::new(|| {
register_int_counter!(
"requests_data_saver",
"The number of requests served from the /data-saver endpoint."
)
.unwrap()
});
pub static REQUESTS_OTHER_COUNTER: Lazy<IntCounter> = Lazy::new(|| {
register_int_counter!(
"requests_other",
"The total number of request not served by primary endpoints."
)
.unwrap()
}); });
)*
#[allow(clippy::shadow_unrelated)] #[allow(clippy::shadow_unrelated)]
pub fn init() { pub fn init() {
// we just need to initialize these, getting the values registers them. $(let _a = $counter.get();)*
let _a = CACHE_HIT_COUNTER.get();
let _a = CACHE_MISS_COUNTER.get();
let _a = REQUESTS_TOTAL_COUNTER.get();
let _a = REQUESTS_DATA_COUNTER.get();
let _a = REQUESTS_DATA_SAVER_COUNTER.get();
let _a = REQUESTS_OTHER_COUNTER.get();
} }
};
}
init_counters!(
(
CACHE_HIT_COUNTER,
IntCounter,
"cache_hit",
"The number of cache hits."
),
(
CACHE_MISS_COUNTER,
IntCounter,
"cache_miss",
"The number of cache misses."
),
(
REQUESTS_TOTAL_COUNTER,
IntCounter,
"requests_total",
"The total number of requests served."
),
(
REQUESTS_DATA_COUNTER,
IntCounter,
"requests_data",
"The number of requests served from the /data endpoint."
),
(
REQUESTS_DATA_SAVER_COUNTER,
IntCounter,
"requests_data_saver",
"The number of requests served from the /data-saver endpoint."
),
(
REQUESTS_OTHER_COUNTER,
IntCounter,
"requests_other",
"The total number of request not served by primary endpoints."
),
);