readings-probe 0.1.8

Probe for vital metrics process logging
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
51
52
53
54
55
56
57
58
59
60

use libc::*;

use std::time::Duration;

use super::{OsReadings, ReadingsResult};

#[repr(C)]
struct BasicTaskInfo {
    pub virtual_size: u64,
    pub resident_size: u64,
    pub resident_size_max: u64,
    pub user_time: timeval,
    pub system_time: timeval,
    pub policy: c_int,
    pub suspend_count: c_uint,
}

extern "C" {
    fn mach_task_self() -> c_uint;
    fn task_info(
        task: c_uint,
        flavor: c_int,
        task_info: *mut BasicTaskInfo,
        count: *mut c_uint,
    ) -> c_uint;
}

fn basic_task_info() -> BasicTaskInfo {
    unsafe {
        let mut info = std::mem::zeroed();
        let mut count: c_uint =
            (::std::mem::size_of::<BasicTaskInfo>() / ::std::mem::size_of::<c_uint>()) as c_uint;
        let me = mach_task_self();
        task_info(me, 20, &mut info, &mut count);
        info
    }
}

fn get_rusage() -> rusage {
    unsafe {
        let mut usage = std::mem::zeroed();
        getrusage(RUSAGE_SELF, &mut usage);
        usage
    }
}

#[cfg(any(target_os = "macos", target_os = "ios"))]
pub(crate) fn get_os_readings() -> ReadingsResult<OsReadings> {
    let info = basic_task_info();
    let rusage = get_rusage();
    Ok(OsReadings {
        virtual_size: info.virtual_size,
        resident_size: info.resident_size,
        resident_size_max: info.resident_size_max,
        user_time: Duration::from_secs(rusage.ru_utime.tv_sec as _) + Duration::from_micros(rusage.ru_utime.tv_usec as _),
        system_time: Duration::from_secs(rusage.ru_stime.tv_sec as _) + Duration::from_micros(rusage.ru_stime.tv_usec as _),
        minor_fault: rusage.ru_minflt as u64,
        major_fault: rusage.ru_majflt as u64,
    })
}