TLA  Line data    Source code

       1                 : //! Helpers for observing duration on `HistogramVec` / `CounterVec` / `GaugeVec` / `MetricVec<T>`.
       2                 : 
       3                 : use std::{future::Future, time::Instant};
       4                 : 
       5                 : pub trait DurationResultObserver {
       6                 :     fn observe_result<T, E>(&self, res: &Result<T, E>, duration: std::time::Duration);
       7                 : }
       8                 : 
       9 CBC         550 : pub async fn observe_async_block_duration_by_result<
      10             550 :     T,
      11             550 :     E,
      12             550 :     F: Future<Output = Result<T, E>>,
      13             550 :     O: DurationResultObserver,
      14             550 : >(
      15             550 :     observer: &O,
      16             550 :     block: F,
      17             550 : ) -> Result<T, E> {
      18             550 :     let start = Instant::now();
      19           28234 :     let result = block.await;
      20             550 :     let duration = start.elapsed();
      21             550 :     observer.observe_result(&result, duration);
      22             550 :     result
      23             550 : }