infrarust 1.1.2

A Rust adaptation of the Infrared Minecraft proxy
Documentation 
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
40
41
42
43
44
45
46
47
48
49
50

pub mod exporter;
pub mod metrics;
pub mod tracing;

pub use metrics::{init_meter_provider, MeterProviderGuard};
use sysinfo::{ProcessRefreshKind, ProcessesToUpdate};
pub use tracing::TracerProviderGuard;

pub use opentelemetry::global;

use lazy_static::lazy_static;
use metrics::InfrarustMetrics;
use std::collections::HashSet;

lazy_static! {
    pub static ref TELEMETRY: InfrarustMetrics = InfrarustMetrics::new();
}

#[derive(Debug, Clone, Copy)]
pub enum Direction {
    Incoming,
    Outgoing,
    Internal,
}

pub fn start_system_metrics_collection() {
    tokio::spawn(async move {
        let mut sys = sysinfo::System::new_all();
        let pid = sysinfo::Pid::from(std::process::id() as usize);
        let cpu_count = sys.cpus().len() as f64;

        loop {
            tokio::time::sleep(sysinfo::MINIMUM_CPU_UPDATE_INTERVAL).await;
            sys.refresh_processes_specifics(
                ProcessesToUpdate::All,
                true,
                ProcessRefreshKind::nothing().with_cpu(),
            );
            if let Some(process) = sys.process(pid) {
                TELEMETRY.update_system_metrics(
                    f64::from(process.cpu_usage()) / cpu_count,
                    process.memory() as f64,
                    process.tasks().unwrap_or(&HashSet::new()).len() as i64,
                );
            } else {
                TELEMETRY.internal_errors.add(1, &[]);
            }
        }
    });
}