custom-utils 0.12.0

Provide customized tools based on features
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

use tokio::task::JoinHandle;

#[cfg(not(all(target_os = "linux", feature = "prod")))]
pub fn daemon() -> JoinHandle<()> {
    tokio::spawn(async move {})
}
#[cfg(all(target_os = "linux", feature = "prod"))]
pub fn daemon() -> JoinHandle<()> {
    tokio::spawn(async move {
        use libsystemd::daemon::{self, NotifyState};
        use log::{debug, error, info, warn};
        if !daemon::booted() {
            info!("Not running systemd, early exit.");
            return;
        };
        let timeout = match daemon::watchdog_enabled(true) {
            Some(time) => time,
            None => {
                info!("watchdog_enabled false");
                return;
            }
        };
        let mut sleep_sec = timeout.as_secs();
        debug!("daemon timeout = {}s", sleep_sec);
        if sleep_sec <= 15 {
            warn!("watchdog timeout <= 10s");
            return;
        }
        sleep_sec -= 10;
        loop {
            match daemon::notify(false, &[NotifyState::Watchdog]) {
                Ok(_res) => {
                    debug!("deamon::notify true");
                }
                Err(err) => {
                    error!("daemon error: {}", err);
                    break;
                }
            }
            tokio::time::sleep(tokio::time::Duration::from_secs(sleep_sec)).await;
        }
    })
}