summaryrefslogtreecommitdiffstats
path: root/src/app/data_harvester/processes/freebsd.rs
blob: 0d92d8736a7f55de59e73262d1518d9b3203b847 (plain)
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
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
//! Process data collection for FreeBSD.  Uses sysinfo.

use std::io;
use std::process::Command;

use hashbrown::HashMap;
use serde::{Deserialize, Deserializer};

use crate::data_harvester::{deserialize_xo, processes::UnixProcessExt};
use crate::Pid;

#[derive(Deserialize, Debug, Default)]
#[serde(rename_all = "kebab-case")]
struct ProcessInformation {
    process: Vec<ProcessRow>,
}

#[derive(Deserialize, Debug)]
#[serde(rename_all = "kebab-case")]
struct ProcessRow {
    #[serde(deserialize_with = "pid")]
    pid: i32,
    #[serde(deserialize_with = "percent_cpu")]
    percent_cpu: f64,
}

pub(crate) struct FreeBSDProcessExt;

impl UnixProcessExt for FreeBSDProcessExt {
    #[inline]
    fn has_backup_proc_cpu_fn() -> bool {
        true
    }

    fn backup_proc_cpu(pids: &[Pid]) -> io::Result<HashMap<Pid, f64>> {
        if pids.is_empty() {
            return Ok(HashMap::new());
        }

        let output = Command::new("ps")
            .args(["--libxo", "json", "-o", "pid,pcpu", "-p"])
            .args(pids.iter().map(i32::to_string))
            .output()?;

        deserialize_xo("process-information", &output.stdout).map(
            |process_info: ProcessInformation| {
                process_info
                    .process
                    .into_iter()
                    .map(|row| (row.pid, row.percent_cpu))
                    .collect()
            },
        )
    }
}

fn pid<'de, D>(deserializer: D) -> Result<i32, D::Error>
where
    D: Deserializer<'de>,
{
    let s = String::deserialize(deserializer)?;
    s.parse().map_err(serde::de::Error::custom)
}

fn percent_cpu<'de, D>(deserializer: D) -> Result<f64, D::Error>
where
    D: Deserializer<'de>,
{
    let s = String::deserialize(deserializer)?;
    s.parse().map_err(serde::de::Error::custom)
}