76 lines
1.9 KiB
Rust
76 lines
1.9 KiB
Rust
use std::time::Instant;
|
|
|
|
pub(super) struct PushMetricsGuard {
|
|
queue: &'static str,
|
|
armed: bool,
|
|
}
|
|
|
|
pub(super) struct PopMetricsGuard {
|
|
queue: &'static str,
|
|
start: Instant,
|
|
armed: bool,
|
|
}
|
|
|
|
pub(super) struct WaitMetricsGuard {
|
|
start: Instant,
|
|
armed: bool,
|
|
}
|
|
|
|
impl PushMetricsGuard {
|
|
pub(super) fn guard(queue: &'static str) -> Self {
|
|
Self { queue, armed: true }
|
|
}
|
|
|
|
pub(super) fn disarm(mut self) {
|
|
self.armed = false;
|
|
}
|
|
}
|
|
|
|
impl PopMetricsGuard {
|
|
pub(super) fn guard(queue: &'static str) -> Self {
|
|
Self {
|
|
queue,
|
|
start: Instant::now(),
|
|
armed: true,
|
|
}
|
|
}
|
|
|
|
pub(super) fn disarm(mut self) {
|
|
self.armed = false;
|
|
}
|
|
}
|
|
|
|
impl WaitMetricsGuard {
|
|
pub(super) fn guard() -> Self {
|
|
Self {
|
|
start: Instant::now(),
|
|
armed: true,
|
|
}
|
|
}
|
|
|
|
pub(super) fn disarm(mut self) {
|
|
self.armed = false;
|
|
}
|
|
}
|
|
|
|
impl Drop for PushMetricsGuard {
|
|
fn drop(&mut self) {
|
|
metrics::counter!(crate::init_metrics::QUEUE_PUSH, "completed" => (!self.armed).to_string(), "queue" => self.queue).increment(1);
|
|
}
|
|
}
|
|
|
|
impl Drop for PopMetricsGuard {
|
|
fn drop(&mut self) {
|
|
metrics::histogram!(crate::init_metrics::QUEUE_POP_DURATION, "completed" => (!self.armed).to_string(), "queue" => self.queue).record(self.start.elapsed().as_secs_f64());
|
|
metrics::counter!(crate::init_metrics::QUEUE_POP, "completed" => (!self.armed).to_string(), "queue" => self.queue).increment(1);
|
|
}
|
|
}
|
|
|
|
impl Drop for WaitMetricsGuard {
|
|
fn drop(&mut self) {
|
|
metrics::histogram!(crate::init_metrics::UPLOAD_WAIT_DURATION, "completed" => (!self.armed).to_string()).record(self.start.elapsed().as_secs_f64());
|
|
metrics::counter!(crate::init_metrics::UPLOAD_WAIT, "completed" => (!self.armed).to_string())
|
|
.increment(1);
|
|
}
|
|
}
|