1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39

use std::sync::{Arc, atomic};
/// `Stats` middleware
///
#[derive(Default)]
pub struct Stats {
    total_requests: Arc<atomic::AtomicUsize>,
    total_request_time: Arc<atomic::AtomicUsize>,
}

impl Stats {
    pub fn incr_total_requests(&self) {
        self.total_requests.fetch_add(1, atomic::Ordering::SeqCst);
    }

    pub fn add_request_time(&self, micros: usize) {
        self.total_request_time.fetch_add(
            micros,
            atomic::Ordering::SeqCst,
        );
    }

    pub fn get_stats(&self) -> String {
        let total_requests = self.total_requests.load(atomic::Ordering::SeqCst);
        let total_requests_time = self.total_request_time.load(atomic::Ordering::SeqCst);

        let avg_request_time = if total_requests > 0 {
            total_requests_time / total_requests
        } else {
            0
        };

        format!(
            "total_requests: {}, total_request_time: {} μs, avg_request_time: {} μs",
            total_requests,
            total_requests_time,
            avg_request_time
        )
    }
}