infomgr 0.2.2

Show system information.
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
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117

#[allow(unused)]
use {
    clap::Parser,
    infomgr_con::util::{
        cpuload::{CpuReport, CpuTime, CpuTimeDiff},
        meminfo::{MemInfo, MemoryReport},
        report::{InfomgrRawReport, InfomgrReport},
        InfoCon,
    },
    jlogger_tracing::{jdebug, jerror, jinfo, jwarn, JloggerBuilder, LevelFilter},
    std::io::{Error, ErrorKind, Result},
    std::sync::{mpsc, Arc},
};

#[cfg(feature = "enable_ipcon")]
use infomgr_con::util::ipcon::IpconInfomgr;

#[derive(Parser, Debug)]
#[clap(author, version, about, long_about=None)]
struct Cli {
    /// Name of the Sender.
    /// It should be "<group name>@<peer name)" in case of using IPCON.
    #[clap(short, long, default_value_t=String::from("fake"))]
    name: String,

    /// Sending interval.
    #[clap(short, long, default_value_t = 1)]
    interval: u64,

    /// Sending raw infomgr report.
    #[clap(short, long)]
    raw: bool,
}

#[allow(unused)]
fn report(name: String, interval: u64) {
    let name = Arc::new(name);
    let (sender, receiver) = mpsc::channel();

    cfg_if::cfg_if! {
        if #[cfg(feature = "enable_ipcon")] {
            let mut ipcon: IpconInfomgr = Default::default();
            let _ = ipcon.sender(name, receiver);
        } else {
            let _ = std::thread::spawn(move || -> Result<()> {
                loop {
                    let ir = receiver.recv()
                        .map_err(|e| Error::new(ErrorKind::ConnectionReset, format!("{}",e)))?;

                    println!("{}", ir);

                }
            });
        }
    }

    let mut cpu_start = CpuTime::new().unwrap();
    loop {
        std::thread::sleep(std::time::Duration::from_secs(interval));
        let cpu_end = CpuTime::new().unwrap();
        let mut report: InfomgrReport = InfomgrReport {
            cpu: (cpu_end - cpu_start).cpu_report(),
            memory: MemInfo::load(),
        };
        report.memory = MemInfo::load();
        if sender.send(report).is_err() {
            break;
        }
        cpu_start = cpu_end;
    }
}

#[allow(unused)]
fn raw_report(name: String, interval: u64) {
    let name = Arc::new(name);
    let (sender, receiver) = mpsc::channel();

    cfg_if::cfg_if! {
        if #[cfg(feature = "enable_ipcon")] {
            let mut ipcon: IpconInfomgr = Default::default();
            let _ = ipcon.sender(name, receiver);
        } else {
            let _ = std::thread::spawn(move || -> Result<()> {
                loop {
                    let ir = receiver.recv()
                        .map_err(|e| Error::new(ErrorKind::ConnectionReset, format!("{}",e)))?;

                    println!("{}", ir);

                }
            });
        }
    }

    loop {
        let mut report: InfomgrRawReport = Default::default();
        let cpu = CpuTime::new().unwrap();
        report.cpu = cpu;
        report.memory = MemInfo::load();
        if sender.send(report).is_err() {
            break;
        }
        std::thread::sleep(std::time::Duration::from_secs(interval));
    }
}

fn main() {
    let cli = Cli::parse();

    JloggerBuilder::new().build();

    if cli.raw {
        raw_report(cli.name, cli.interval);
    } else {
        report(cli.name, cli.interval);
    }
}