extract to lib:

Cargo.lock

@@ -752,6 +752,12 @@ dependencies = [
  "want",
 ]
 
+[[package]]
+name = "iai"
+version = "0.1.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "71a816c97c42258aa5834d07590b718b4c9a598944cd39a52dc25b351185d678"
+
 [[package]]
 name = "idna"
 version = "0.2.2"
@@ -1034,6 +1040,7 @@ dependencies = [
  "futures-lite",
  "futures-util",
  "heim",
+ "iai",
  "once_cell",
  "parse-display",
  "regex",

Cargo.toml

@@ -16,3 +16,6 @@ futures-lite = "1"
 parse-display = "0.4"
 regex = { version = "1", default-features = false }
 once_cell = "1"
+
+[dev-dependencies]
+iai = "0.1"

src/heim.rs

@@ -38,16 +38,11 @@ pub struct DiskUsage {
     pub free: u64,
 }
 
-#[derive(Default)]
-pub struct Heim {}
-
-impl Heim {
-    pub async fn temperatures(&self) -> Result<HashMap<TemperatureLabel, f32>> {
+pub async fn temperatures() -> Result<HashMap<TemperatureLabel, f32>> {
     // ugly workaround problems between async-fs and tokio
     let results = tokio::task::spawn_blocking(|| {
         futures_lite::future::block_on(
-                heim::sensors::temperatures()
-                    .collect::<Vec<Result<TemperatureSensor, heim::Error>>>(),
+            heim::sensors::temperatures().collect::<Vec<Result<TemperatureSensor, heim::Error>>>(),
         )
     })
     .await
@@ -64,23 +59,23 @@ impl Heim {
         _ => None,
     });
     Ok(results.collect())
-    }
+}
 
-    pub async fn memory(&self) -> Result<Memory> {
+pub async fn memory() -> Result<Memory> {
     let memory = heim::memory::memory().await?;
     Ok(Memory {
         total: memory.total().get::<information::byte>(),
         free: memory.free().get::<information::byte>(),
         available: memory.available().get::<information::byte>(),
     })
-    }
+}
 
-    pub async fn cpu_time(&self) -> Result<f64> {
+pub async fn cpu_time() -> Result<f64> {
     let time = time().await?;
     Ok(time.user().get::<time::second>() + time.system().get::<time::second>())
-    }
+}
 
-    pub async fn network_stats(&self) -> Result<impl Stream<Item = IOStats>> {
+pub async fn network_stats() -> Result<impl Stream<Item = IOStats>> {
     let networks = heim::net::io_counters().await?;
     Ok(networks
         .filter_map(|network| future::ready(network.ok()))
@@ -90,13 +85,13 @@ impl Heim {
             bytes_sent: network.bytes_sent().get::<information::byte>(),
             bytes_received: network.bytes_recv().get::<information::byte>(),
         }))
-    }
+}
 
-    pub async fn hostname(&self) -> Result<String> {
+pub async fn hostname() -> Result<String> {
     Ok(heim::host::platform().await?.hostname().to_string())
-    }
+}
 
-    pub async fn disk_stats(&self) -> Result<impl Stream<Item = IOStats>> {
+pub async fn disk_stats() -> Result<impl Stream<Item = IOStats>> {
     static DISK_REGEX: Lazy<Regex> =
         Lazy::new(|| Regex::new(r"^([sv]d[a-z]+|nvme\dn\d)$").unwrap());
     let disks = heim::disk::io_counters().await?;
@@ -116,9 +111,9 @@ impl Heim {
             bytes_sent: disk.write_bytes().get::<information::byte>(),
             bytes_received: disk.read_bytes().get::<information::byte>(),
         }))
-    }
+}
 
-    pub async fn disk_usage(&self) -> Result<impl Stream<Item = DiskUsage>> {
+pub async fn disk_usage() -> Result<impl Stream<Item = DiskUsage>> {
     Ok(heim::disk::partitions_physical()
         .await?
         .filter_map(|result| future::ready(result.ok()))
@@ -135,5 +130,4 @@ impl Heim {
             size: usage.total().get::<information::byte>(),
             free: usage.free().get::<information::byte>(),
         }))
-    }
 }

src/lib.rs

@@ -0,0 +1,133 @@
+pub mod heim;
+mod zfs;
+
+use crate::heim::*;
+use crate::zfs::pools;
+use color_eyre::Result;
+use futures_util::stream::StreamExt;
+use futures_util::{pin_mut, try_join};
+use std::collections::{HashMap, HashSet};
+use std::fmt::Write;
+
+pub async fn get_metrics() -> Result<String> {
+    let (hostname, pools, cpu, memory, network, temperatures, disks, disk_usage): (
+        String,
+        Vec<DiskUsage>,
+        f64,
+        Memory,
+        _,
+        HashMap<TemperatureLabel, f32>,
+        _,
+        _,
+    ) = try_join! {
+        hostname(),
+        pools(),
+        cpu_time(),
+        memory(),
+        network_stats(),
+        temperatures(),
+        disk_stats(),
+        disk_usage(),
+    }?;
+    pin_mut!(network);
+    pin_mut!(disks);
+    pin_mut!(disk_usage);
+    let mut result = String::with_capacity(256);
+    writeln!(&mut result, "cpu_time{{host=\"{}\"}} {:.1}", hostname, cpu).ok();
+    writeln!(
+        &mut result,
+        "memory_total{{host=\"{}\"}} {}",
+        hostname, memory.total
+    )
+    .ok();
+    writeln!(
+        &mut result,
+        "memory_available{{host=\"{}\"}} {}",
+        hostname, memory.available
+    )
+    .ok();
+    writeln!(
+        &mut result,
+        "memory_free{{host=\"{}\"}} {}",
+        hostname, memory.free
+    )
+    .ok();
+    for pool in pools {
+        writeln!(
+            &mut result,
+            "zfs_pool_size{{host=\"{}\", pool=\"{}\"}} {}",
+            hostname, pool.name, pool.size
+        )
+        .ok();
+        writeln!(
+            &mut result,
+            "zfs_pool_free{{host=\"{}\", pool=\"{}\"}} {}",
+            hostname, pool.name, pool.free
+        )
+        .ok();
+    }
+    while let Some(network) = network.next().await {
+        let network: IOStats = network;
+        if network.bytes_received > 0 || network.bytes_sent > 0 {
+            writeln!(
+                &mut result,
+                "net_sent{{host=\"{}\", network=\"{}\"}} {}",
+                hostname, network.interface, network.bytes_sent
+            )
+            .ok();
+            writeln!(
+                &mut result,
+                "net_received{{host=\"{}\", network=\"{}\"}} {}",
+                hostname, network.interface, network.bytes_received
+            )
+            .ok();
+        }
+    }
+    while let Some(disk) = disks.next().await {
+        let disk: IOStats = disk;
+        if disk.bytes_received > 0 && disk.bytes_sent > 0 {
+            writeln!(
+                &mut result,
+                "disk_sent{{host=\"{}\", disk=\"{}\"}} {}",
+                hostname, disk.interface, disk.bytes_sent
+            )
+            .ok();
+            writeln!(
+                &mut result,
+                "disk_received{{host=\"{}\", disk=\"{}\"}} {}",
+                hostname, disk.interface, disk.bytes_received
+            )
+            .ok();
+        }
+    }
+
+    let mut found_sizes = HashSet::new();
+    while let Some(disk) = disk_usage.next().await {
+        let disk: DiskUsage = disk;
+        if disk.size > 0 {
+            if found_sizes.insert((disk.size, disk.free)) {
+                writeln!(
+                    &mut result,
+                    "disk_size{{host=\"{}\", disk=\"{}\"}} {}",
+                    hostname, disk.name, disk.size
+                )
+                .ok();
+                writeln!(
+                    &mut result,
+                    "disk_free{{host=\"{}\", disk=\"{}\"}} {}",
+                    hostname, disk.name, disk.free
+                )
+                .ok();
+            }
+        }
+    }
+    for (label, temp) in temperatures {
+        writeln!(
+            &mut result,
+            "temperature{{host=\"{}\", sensor=\"{}\"}} {:.1}",
+            hostname, label, temp
+        )
+        .ok();
+    }
+    Ok(result)
+}

src/main.rs

@@ -1,13 +1,5 @@
-pub mod heim;
-mod zfs;
-
-use crate::heim::{DiskUsage, Heim, IOStats, Memory, TemperatureLabel};
-use crate::zfs::ZFS;
 use color_eyre::{Report, Result};
-use futures_util::stream::StreamExt;
-use futures_util::{pin_mut, try_join};
-use std::collections::{HashMap, HashSet};
-use std::fmt::Write;
+use palantir::get_metrics;
 use warp::reject::Reject;
 use warp::{Filter, Rejection};
 
@@ -23,131 +15,11 @@ impl From<Report> for ReportRejection {
 
 impl Reject for ReportRejection {}
 
-async fn get_metrics(heim: Heim, zfs: ZFS) -> Result<String, ReportRejection> {
-    let (hostname, pools, cpu, memory, network, temperatures, disks, disk_usage): (
-        String,
-        Vec<DiskUsage>,
-        f64,
-        Memory,
-        _,
-        HashMap<TemperatureLabel, f32>,
-        _,
-        _,
-    ) = try_join! {
-        heim.hostname(),
-        zfs.pools(),
-        heim.cpu_time(),
-        heim.memory(),
-        heim.network_stats(),
-        heim.temperatures(),
-        heim.disk_stats(),
-        heim.disk_usage(),
-    }?;
-    pin_mut!(network);
-    pin_mut!(disks);
-    pin_mut!(disk_usage);
-    let mut result = String::with_capacity(256);
-    writeln!(&mut result, "cpu_time{{host=\"{}\"}} {:.1}", hostname, cpu).ok();
-    writeln!(
-        &mut result,
-        "memory_total{{host=\"{}\"}} {}",
-        hostname, memory.total
-    )
-    .ok();
-    writeln!(
-        &mut result,
-        "memory_available{{host=\"{}\"}} {}",
-        hostname, memory.available
-    )
-    .ok();
-    writeln!(
-        &mut result,
-        "memory_free{{host=\"{}\"}} {}",
-        hostname, memory.free
-    )
-    .ok();
-    for pool in pools {
-        writeln!(
-            &mut result,
-            "zfs_pool_size{{host=\"{}\", pool=\"{}\"}} {}",
-            hostname, pool.name, pool.size
-        )
-        .ok();
-        writeln!(
-            &mut result,
-            "zfs_pool_free{{host=\"{}\", pool=\"{}\"}} {}",
-            hostname, pool.name, pool.free
-        )
-        .ok();
-    }
-    while let Some(network) = network.next().await {
-        let network: IOStats = network;
-        if network.bytes_received > 0 || network.bytes_sent > 0 {
-            writeln!(
-                &mut result,
-                "net_sent{{host=\"{}\", network=\"{}\"}} {}",
-                hostname, network.interface, network.bytes_sent
-            )
-            .ok();
-            writeln!(
-                &mut result,
-                "net_received{{host=\"{}\", network=\"{}\"}} {}",
-                hostname, network.interface, network.bytes_received
-            )
-            .ok();
-        }
-    }
-    while let Some(disk) = disks.next().await {
-        let disk: IOStats = disk;
-        if disk.bytes_received > 0 && disk.bytes_sent > 0 {
-            writeln!(
-                &mut result,
-                "disk_sent{{host=\"{}\", disk=\"{}\"}} {}",
-                hostname, disk.interface, disk.bytes_sent
-            )
-            .ok();
-            writeln!(
-                &mut result,
-                "disk_received{{host=\"{}\", disk=\"{}\"}} {}",
-                hostname, disk.interface, disk.bytes_received
-            )
-            .ok();
-        }
-    }
-
-    let mut found_sizes = HashSet::new();
-    while let Some(disk) = disk_usage.next().await {
-        let disk: DiskUsage = disk;
-        if disk.size > 0 {
-            if found_sizes.insert((disk.size, disk.free)) {
-                writeln!(
-                    &mut result,
-                    "disk_size{{host=\"{}\", disk=\"{}\"}} {}",
-                    hostname, disk.name, disk.size
-                )
-                .ok();
-                writeln!(
-                    &mut result,
-                    "disk_free{{host=\"{}\", disk=\"{}\"}} {}",
-                    hostname, disk.name, disk.free
-                )
-                .ok();
-            }
-        }
-    }
-    for (label, temp) in temperatures {
-        writeln!(
-            &mut result,
-            "temperature{{host=\"{}\", sensor=\"{}\"}} {:.1}",
-            hostname, label, temp
-        )
-        .ok();
-    }
-    Result::<_, ReportRejection>::Ok(result)
-}
-
-async fn serve_metrics(heim: Heim, zfs: ZFS) -> Result<String, Rejection> {
-    get_metrics(heim, zfs).await.map_err(warp::reject::custom)
+async fn serve_metrics() -> Result<String, Rejection> {
+    get_metrics()
+        .await
+        .map_err(ReportRejection::from)
+        .map_err(warp::reject::custom)
 }
 
 #[tokio::main]
@@ -163,13 +35,7 @@ async fn main() -> Result<()> {
     })
     .expect("Error setting Ctrl-C handler");
 
-    let heim = warp::any().map(|| Heim::default());
-    let zfs = warp::any().map(|| ZFS::default());
-
-    let metrics = warp::path!("metrics")
-        .and(heim)
-        .and(zfs)
-        .and_then(serve_metrics);
+    let metrics = warp::path!("metrics").and_then(serve_metrics);
 
     warp::serve(metrics).run(([0, 0, 0, 0], host_port)).await;
     Ok(())

src/zfs.rs

@@ -3,11 +3,7 @@ use color_eyre::Result;
 use std::process::Command;
 use tokio::task::spawn_blocking;
 
-#[derive(Default)]
-pub struct ZFS;
-
-impl ZFS {
-    pub async fn pools(self) -> Result<Vec<DiskUsage>> {
+pub async fn pools() -> Result<Vec<DiskUsage>> {
     spawn_blocking(move || {
         let mut z = Command::new("zpool");
         z.args(&["list", "-p", "-H", "-o", "name,size,free"]);
@@ -32,5 +28,4 @@ impl ZFS {
         }
     })
     .await?
-    }
 }