using daemonize

This commit is contained in:
Jorge Bolois 2024-02-03 14:50:04 +01:00
parent b382c5121d
commit 094ba3631d
4 changed files with 64 additions and 53 deletions

1
.gitignore vendored
View File

@ -4,4 +4,3 @@ Cargo.lock
**/*.rs.bk **/*.rs.bk
*.pdb *.pdb
/target /target
nohup.out

View File

@ -6,5 +6,6 @@ edition = "2021"
[dependencies] [dependencies]
tokio = { version = "1.35.1", features = ["full"]} tokio = { version = "1.35.1", features = ["full"]}
kagiyama = "0.3.0" kagiyama = "0.3.0"
mtemp = { git = "https://git.midefos.com/midefos/mtemp.git", branch = "main" }
anyhow = "1.0.79" anyhow = "1.0.79"
daemonize = "0.5.0"
mtemp = "0.1.21"

View File

@ -1,15 +1,34 @@
use anyhow::Result; use anyhow::{anyhow, Context, Result};
use daemonize::Daemonize;
use kagiyama::prometheus::metrics::gauge::Gauge; use kagiyama::prometheus::metrics::gauge::Gauge;
use kagiyama::{AlwaysReady, Watcher}; use kagiyama::{AlwaysReady, Watcher};
use mtemp::Mtemp; use mtemp::Mtemp;
use std::env; use std::env;
use std::fs::File;
use std::sync::atomic::AtomicU64; use std::sync::atomic::AtomicU64;
use tokio::spawn;
use tokio::time::sleep; use tokio::time::sleep;
use tokio::time::Duration; use tokio::time::Duration;
fn main() -> Result<()> {
let name = "mtemp-monitor";
let stdout = File::create(format!("/tmp/{}.log", name)).context("Error creating log file")?;
let daemonize = Daemonize::new()
.pid_file(format!("/tmp/{}.pid", name))
.stdout(stdout)
.chown_pid_file(true)
.working_directory("/");
match daemonize.start() {
Ok(_) => {
println!("daemon started");
tokio_main()
}
Err(err) => Err(anyhow!("{:?}", err)),
}
}
#[tokio::main] #[tokio::main]
async fn main() -> Result<()> { async fn tokio_main() -> Result<()> {
let interval_secs: u64 = env::var("MTEMP_INTERVAL") let interval_secs: u64 = env::var("MTEMP_INTERVAL")
.unwrap_or_else(|_| "1".to_string()) .unwrap_or_else(|_| "1".to_string())
.parse()?; .parse()?;
@ -19,9 +38,6 @@ async fn main() -> Result<()> {
let mut watcher = Watcher::<AlwaysReady>::default(); let mut watcher = Watcher::<AlwaysReady>::default();
watcher.start_server(addr.parse()?).await; watcher.start_server(addr.parse()?).await;
// CPU
{
spawn(async move {
let interval = Duration::from_secs(interval_secs); let interval = Duration::from_secs(interval_secs);
let cpu_gauge = Gauge::<f64, AtomicU64>::default(); let cpu_gauge = Gauge::<f64, AtomicU64>::default();
@ -54,6 +70,9 @@ async fn main() -> Result<()> {
); );
} }
println!("interval set every {} seconds", interval_secs);
println!("listening on {}", addr);
loop { loop {
sleep(interval).await; sleep(interval).await;
if let Some(cpu) = Mtemp::cpu() { if let Some(cpu) = Mtemp::cpu() {
@ -66,12 +85,4 @@ async fn main() -> Result<()> {
disk_gauge.set(disk); disk_gauge.set(disk);
} }
} }
});
}
println!("interval set every {} seconds", interval_secs);
println!("listening on {}", addr);
tokio::signal::ctrl_c().await.unwrap();
Ok(())
} }

View File

@ -1,5 +1,5 @@
#!/bin/bash #!/bin/bash
nohup cargo r -r & cargo run --release
sleep 1 sleep 1
echo "Monitor started" echo "Monitor started"