78 lines
2.3 KiB
Rust
78 lines
2.3 KiB
Rust
use anyhow::Result;
|
|
use kagiyama::prometheus::metrics::gauge::Gauge;
|
|
use kagiyama::{AlwaysReady, Watcher};
|
|
use mtemp::Mtemp;
|
|
use std::env;
|
|
use std::sync::atomic::AtomicU64;
|
|
use tokio::signal::ctrl_c;
|
|
use tokio::spawn;
|
|
use tokio::time::sleep;
|
|
use tokio::time::Duration;
|
|
|
|
#[tokio::main]
|
|
async fn main() -> Result<()> {
|
|
let interval_secs: u64 = env::var("MTEMP_INTERVAL")
|
|
.unwrap_or_else(|_| "1".to_string())
|
|
.parse()?;
|
|
let port = env::var("MTEMP_PORT").unwrap_or_else(|_| "9090".to_string());
|
|
|
|
let addr = format!("127.0.0.1:{}", port);
|
|
let mut watcher = Watcher::<AlwaysReady>::default();
|
|
watcher.start_server(addr.parse()?).await;
|
|
|
|
{
|
|
spawn(async move {
|
|
let interval = Duration::from_secs(interval_secs);
|
|
|
|
let cpu_gauge = Gauge::<f64, AtomicU64>::default();
|
|
{
|
|
let mut registry = watcher.metrics_registry();
|
|
registry.register(
|
|
"cpu_temp",
|
|
"Shows the temperature of the CPU in degrees C",
|
|
cpu_gauge.clone(),
|
|
);
|
|
}
|
|
|
|
let gpu_gauge = Gauge::<f64, AtomicU64>::default();
|
|
{
|
|
let mut registry = watcher.metrics_registry();
|
|
registry.register(
|
|
"gpu_temp",
|
|
"Shows the temperature of the GPU in degrees C",
|
|
gpu_gauge.clone(),
|
|
);
|
|
}
|
|
|
|
let disk_gauge = Gauge::<f64, AtomicU64>::default();
|
|
{
|
|
let mut registry = watcher.metrics_registry();
|
|
registry.register(
|
|
"disk_temp",
|
|
"Shows the temperature of the disk in degrees C",
|
|
disk_gauge.clone(),
|
|
);
|
|
}
|
|
|
|
loop {
|
|
sleep(interval).await;
|
|
if let Some(cpu) = Mtemp::cpu() {
|
|
cpu_gauge.set(cpu);
|
|
}
|
|
if let Some(gpu) = Mtemp::gpu() {
|
|
gpu_gauge.set(gpu);
|
|
}
|
|
if let Some(disk) = Mtemp::disk() {
|
|
disk_gauge.set(disk);
|
|
}
|
|
}
|
|
});
|
|
}
|
|
|
|
println!("interval set every {} seconds", interval_secs);
|
|
println!("listening on {}", addr);
|
|
ctrl_c().await?;
|
|
|
|
Ok(())
|
|
}
|