Jędrzej Stuczyński
60 lines
1.9 KiB
Rust
60 lines
1.9 KiB
Rust
// Copyright 2024 - Nym Technologies SA <contact@nymtech.net>
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
use std::sync::atomic::{AtomicUsize, Ordering};
|
|
use std::sync::Arc;
|
|
|
|
#[derive(Default)]
|
|
pub struct NetworkStats {
|
|
// for now just experiment with basic data, we could always extend it
|
|
active_ingress_mixnet_connections: AtomicUsize,
|
|
|
|
active_ingress_websocket_connections: AtomicUsize,
|
|
|
|
// the reason for additional `Arc` on this one is that the handler wasn't
|
|
// designed with metrics in mind and this single counter has been woven through
|
|
// the call stack
|
|
active_egress_mixnet_connections: Arc<AtomicUsize>,
|
|
}
|
|
|
|
impl NetworkStats {
|
|
pub fn new_active_ingress_mixnet_client(&self) {
|
|
self.active_ingress_mixnet_connections
|
|
.fetch_add(1, Ordering::Relaxed);
|
|
}
|
|
|
|
pub fn disconnected_ingress_mixnet_client(&self) {
|
|
self.active_ingress_mixnet_connections
|
|
.fetch_sub(1, Ordering::Relaxed);
|
|
}
|
|
|
|
pub fn new_ingress_websocket_client(&self) {
|
|
self.active_ingress_websocket_connections
|
|
.fetch_add(1, Ordering::Relaxed);
|
|
}
|
|
|
|
pub fn disconnected_ingress_websocket_client(&self) {
|
|
self.active_ingress_websocket_connections
|
|
.fetch_sub(1, Ordering::Relaxed);
|
|
}
|
|
|
|
pub fn active_ingress_mixnet_connections_count(&self) -> usize {
|
|
self.active_ingress_mixnet_connections
|
|
.load(Ordering::Relaxed)
|
|
}
|
|
|
|
pub fn active_ingress_websocket_connections_count(&self) -> usize {
|
|
self.active_ingress_websocket_connections
|
|
.load(Ordering::SeqCst)
|
|
}
|
|
|
|
pub fn active_egress_mixnet_connections_counter(&self) -> Arc<AtomicUsize> {
|
|
self.active_egress_mixnet_connections.clone()
|
|
}
|
|
|
|
pub fn active_egress_mixnet_connections_count(&self) -> usize {
|
|
self.active_egress_mixnet_connections
|
|
.load(Ordering::Relaxed)
|
|
}
|
|
}
|