previa-runner 1.0.0-alpha.22

API for remote execution of integration and load tests via HTTP streaming (SSE).
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

use sysinfo::{Pid, ProcessRefreshKind, ProcessesToUpdate, System};

use crate::server::models::RunnerInfoResponse;

#[derive(Debug)]
pub struct RuntimeSampler {
    pid: u32,
    process_pid: Pid,
    system: System,
}

impl RuntimeSampler {
    pub fn new() -> Self {
        let pid = std::process::id();
        let process_pid = Pid::from_u32(pid);
        let mut sampler = Self {
            pid,
            process_pid,
            system: System::new(),
        };
        sampler.refresh();
        sampler
    }

    pub fn snapshot(&mut self) -> Option<RunnerInfoResponse> {
        self.refresh();
        let process = self.system.process(self.process_pid)?;
        Some(RunnerInfoResponse {
            pid: self.pid,
            memory_bytes: process.memory(),
            virtual_memory_bytes: process.virtual_memory(),
            cpu_usage_percent: process.cpu_usage(),
            network_tx_bytes: 0,
            network_rx_bytes: 0,
            network_total_bytes: 0,
        })
    }

    fn refresh(&mut self) {
        self.system.refresh_processes_specifics(
            ProcessesToUpdate::Some(&[self.process_pid]),
            true,
            ProcessRefreshKind::nothing().with_memory().with_cpu(),
        );
    }
}

pub fn snapshot_current_process_runtime() -> Option<RunnerInfoResponse> {
    RuntimeSampler::new().snapshot()
}