1 files changed, 45 insertions, 24 deletions

diff --git a/src/app/data_harvester/memory/heim.rs b/src/app/data_harvester/memory/heim.rs
index e0d9c2c9..29f16910 100644
--- a/src/app/data_harvester/memory/heim.rs
+++ b/src/app/data_harvester/memory/heim.rs
@@ -1,18 +1,10 @@
 //! Data collection for memory via heim.
 
-#[derive(Debug, Clone)]
+#[derive(Debug, Clone, Default)]
 pub struct MemHarvest {
     pub mem_total_in_kib: u64,
     pub mem_used_in_kib: u64,
-}
-
-impl Default for MemHarvest {
-    fn default() -> Self {
-        MemHarvest {
-            mem_total_in_kib: 0,
-            mem_used_in_kib: 0,
-        }
-    }
+    pub use_percent: Option<f64>,
 }
 
 pub async fn get_mem_data(
@@ -36,37 +28,36 @@ pub async fn get_ram_data() -> crate::utils::error::Result<Option<MemHarvest>> {
     let (mem_total_in_kib, mem_used_in_kib) = {
         #[cfg(target_os = "linux")]
         {
+            use heim::memory::os::linux::MemoryExt;
+            use heim::units::information::kilobyte;
+
             // For Linux, the "kilobyte" value in the .total call is actually kibibytes - see
             // https://access.redhat.com/documentation/en-us/red_hat_enterprise_linux/6/html/deployment_guide/s2-proc-meminfo
             //
             // Heim parses this as kilobytes (https://github.com/heim-rs/heim/blob/master/heim-memory/src/sys/linux/memory.rs#L82)
             // even though it probably shouldn't...
 
-            use heim::memory::os::linux::MemoryExt;
-
             (
-                memory.total().get::<heim::units::information::kilobyte>(),
-                memory.used().get::<heim::units::information::kilobyte>(),
+                memory.total().get::<kilobyte>(),
+                memory.used().get::<kilobyte>(),
             )
         }
         #[cfg(target_os = "macos")]
         {
             use heim::memory::os::macos::MemoryExt;
+            use heim::units::information::kibibyte;
             (
-                memory.total().get::<heim::units::information::kibibyte>(),
-                memory.active().get::<heim::units::information::kibibyte>()
-                    + memory.wire().get::<heim::units::information::kibibyte>(),
+                memory.total().get::<kibibyte>(),
+                memory.active().get::<kibibyte>() + memory.wire().get::<kibibyte>(),
             )
         }
         #[cfg(target_os = "windows")]
         {
-            let mem_total_in_kib = memory.total().get::<heim::units::information::kibibyte>();
+            use heim::units::information::kibibyte;
+            let mem_total_in_kib = memory.total().get::<kibibyte>();
             (
                 mem_total_in_kib,
-                mem_total_in_kib
-                    - memory
-                        .available()
-                        .get::<heim::units::information::kibibyte>(),
+                mem_total_in_kib - memory.available().get::<kibibyte>(),
             )
         }
     };
@@ -74,14 +65,44 @@ pub async fn get_ram_data() -> crate::utils::error::Result<Option<MemHarvest>> {
     Ok(Some(MemHarvest {
         mem_total_in_kib,
         mem_used_in_kib,
+        use_percent: if mem_total_in_kib == 0 {
+            None
+        } else {
+            Some(mem_used_in_kib as f64 / mem_total_in_kib as f64 * 100.0)
+        },
     }))
 }
 
 pub async fn get_swap_data() -> crate::utils::error::Result<Option<MemHarvest>> {
     let memory = heim::memory::swap().await?;
 
+    let (mem_total_in_kib, mem_used_in_kib) = {
+        #[cfg(target_os = "linux")]
+        {
+            // Similar story to above - heim parses this information incorrectly as far as I can tell, so kilobytes = kibibytes here.
+            use heim::units::information::kilobyte;
+            (
+                memory.total().get::<kilobyte>(),
+                memory.used().get::<kilobyte>(),
+            )
+        }
+        #[cfg(any(target_os = "windows", target_os = "macos"))]
+        {
+            use heim::units::information::kibibyte;
+            (
+                memory.total().get::<kibibyte>(),
+                memory.used().get::<kibibyte>(),
+            )
+        }
+    };
+
     Ok(Some(MemHarvest {
-        mem_total_in_kib: memory.total().get::<heim::units::information::kibibyte>(),
-        mem_used_in_kib: memory.used().get::<heim::units::information::kibibyte>(),
+        mem_total_in_kib,
+        mem_used_in_kib,
+        use_percent: if mem_total_in_kib == 0 {
+            None
+        } else {
+            Some(mem_used_in_kib as f64 / mem_total_in_kib as f64 * 100.0)
+        },
     }))
 }