76 lines
1.8 KiB
Rust
76 lines
1.8 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!("pict-rs.queue.push", "completed" => (!self.armed).to_string(), "queue" => self.queue).increment(1);
|
|
}
|
|
}
|
|
|
|
impl Drop for PopMetricsGuard {
|
|
fn drop(&mut self) {
|
|
metrics::histogram!("pict-rs.queue.pop.duration", "completed" => (!self.armed).to_string(), "queue" => self.queue).record(self.start.elapsed().as_secs_f64());
|
|
metrics::counter!("pict-rs.queue.pop", "completed" => (!self.armed).to_string(), "queue" => self.queue).increment(1);
|
|
}
|
|
}
|
|
|
|
impl Drop for WaitMetricsGuard {
|
|
fn drop(&mut self) {
|
|
metrics::histogram!("pict-rs.upload.wait.duration", "completed" => (!self.armed).to_string()).record(self.start.elapsed().as_secs_f64());
|
|
metrics::counter!("pict-rs.upload.wait", "completed" => (!self.armed).to_string())
|
|
.increment(1);
|
|
}
|
|
}
|