hyperi_rustlib/metrics/
process.rs

1// Project:   hyperi-rustlib
2// File:      src/metrics/process.rs
3// Purpose:   Process-level metrics collection
4// Language:  Rust
5//
6// License:   BUSL-1.1
7// Copyright: (c) 2026 HYPERI PTY LIMITED
8
9//! Process-level metrics collection.
10
11use std::sync::Arc;
12use std::time::{SystemTime, UNIX_EPOCH};
13
14use sysinfo::{ProcessRefreshKind, ProcessesToUpdate, RefreshKind, System};
15
16/// Process metrics collector.
17#[derive(Debug, Clone)]
18pub struct ProcessMetrics {
19    namespace: String,
20    system: Arc<std::sync::Mutex<System>>,
21    pid: sysinfo::Pid,
22    start_time: f64,
23}
24
25impl ProcessMetrics {
26    /// Create a new process metrics collector.
27    #[must_use]
28    pub fn new(namespace: &str) -> Self {
29        let pid = sysinfo::Pid::from_u32(std::process::id());
30        let system = System::new_with_specifics(
31            RefreshKind::nothing().with_processes(ProcessRefreshKind::everything()),
32        );
33
34        let start_time = SystemTime::now()
35            .duration_since(UNIX_EPOCH)
36            .map_or(0.0, |d| d.as_secs_f64());
37
38        let this = Self {
39            namespace: namespace.to_string(),
40            system: Arc::new(std::sync::Mutex::new(system)),
41            pid,
42            start_time,
43        };
44
45        // Register metric descriptions
46        this.register_metrics();
47        this
48    }
49
50    /// Register metric descriptions.
51    fn register_metrics(&self) {
52        let ns = &self.namespace;
53
54        metrics::describe_gauge!(
55            format!("{ns}_process_cpu_seconds_total"),
56            "Total user and system CPU time spent in seconds".to_string()
57        );
58        metrics::describe_gauge!(
59            format!("{ns}_process_resident_memory_bytes"),
60            "Resident memory size in bytes".to_string()
61        );
62        metrics::describe_gauge!(
63            format!("{ns}_process_virtual_memory_bytes"),
64            "Virtual memory size in bytes".to_string()
65        );
66        metrics::describe_gauge!(
67            format!("{ns}_process_open_fds"),
68            "Number of open file descriptors".to_string()
69        );
70        metrics::describe_gauge!(
71            format!("{ns}_process_start_time_seconds"),
72            "Start time of the process since unix epoch in seconds".to_string()
73        );
74    }
75
76    /// Update process metrics.
77    pub fn update(&self) {
78        let mut system = self.system.lock().expect("lock poisoned");
79        system.refresh_processes_specifics(
80            ProcessesToUpdate::Some(&[self.pid]),
81            true,
82            ProcessRefreshKind::everything(),
83        );
84
85        if let Some(process) = system.process(self.pid) {
86            let ns = &self.namespace;
87
88            // CPU time (approximate - sysinfo gives percentage, not total time)
89            let cpu_usage = f64::from(process.cpu_usage());
90            metrics::gauge!(format!("{ns}_process_cpu_seconds_total")).set(cpu_usage);
91
92            // Memory
93            let rss = process.memory();
94            let virtual_mem = process.virtual_memory();
95            metrics::gauge!(format!("{ns}_process_resident_memory_bytes")).set(rss as f64);
96            metrics::gauge!(format!("{ns}_process_virtual_memory_bytes")).set(virtual_mem as f64);
97
98            // File descriptors (Linux-specific)
99            #[cfg(target_os = "linux")]
100            {
101                if let Ok(fds) = count_open_fds() {
102                    metrics::gauge!(format!("{ns}_process_open_fds")).set(fds as f64);
103                }
104            }
105
106            // Start time
107            metrics::gauge!(format!("{ns}_process_start_time_seconds")).set(self.start_time);
108        }
109    }
110}
111
112/// Count open file descriptors (Linux only).
113#[cfg(target_os = "linux")]
114fn count_open_fds() -> std::io::Result<usize> {
115    let fd_dir = format!("/proc/{}/fd", std::process::id());
116    std::fs::read_dir(fd_dir).map(|entries| entries.count())
117}
118
119#[cfg(test)]
120mod tests {
121    use super::*;
122
123    #[test]
124    fn test_process_metrics_new() {
125        let pm = ProcessMetrics::new("test");
126        assert_eq!(pm.namespace, "test");
127        assert!(pm.start_time > 0.0);
128    }
129
130    #[test]
131    fn test_process_metrics_update() {
132        let pm = ProcessMetrics::new("test");
133        // Should not panic
134        pm.update();
135    }
136}
hyperi_rustlib/metrics/process.rs

hyperi_rustlib/metrics/
process.rs
