use std::sync::{Arc, atomic};
#[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
)
}
}