Dreaded_X
f75726b93a
All checks were successful
Build and deploy / Build container and manifests (push) Successful in 7m25s
37 lines
791 B
Rust
37 lines
791 B
Rust
use std::sync::atomic::{AtomicUsize, Ordering};
|
|
|
|
use crate::units::Unit;
|
|
|
|
#[derive(Debug, Default)]
|
|
pub struct Stats {
|
|
connections: AtomicUsize,
|
|
rx: AtomicUsize,
|
|
tx: AtomicUsize,
|
|
}
|
|
|
|
impl Stats {
|
|
pub fn add_connection(&self) {
|
|
self.connections.fetch_add(1, Ordering::Relaxed);
|
|
}
|
|
|
|
pub fn add_rx_bytes(&self, n: usize) {
|
|
self.rx.fetch_add(n, Ordering::Relaxed);
|
|
}
|
|
|
|
pub fn add_tx_bytes(&self, n: usize) {
|
|
self.tx.fetch_add(n, Ordering::Relaxed);
|
|
}
|
|
|
|
pub fn connections(&self) -> usize {
|
|
self.connections.load(Ordering::Relaxed)
|
|
}
|
|
|
|
pub fn rx(&self) -> Unit {
|
|
Unit::new(self.rx.load(Ordering::Relaxed), "B")
|
|
}
|
|
|
|
pub fn tx(&self) -> Unit {
|
|
Unit::new(self.tx.load(Ordering::Relaxed), "B")
|
|
}
|
|
}
|