Merge pull request #3 from arybhatt4533/add-disk-module

Add Disk Usage Metric Collector

Author: dev-dami

src/cli/mod.rs

@@ -1,3 +1,4 @@
+
 use clap::Parser;
 
 #[derive(Parser)]

src/lib.rs

@@ -5,28 +5,31 @@ pub mod output;
 pub mod tui;
 
 use crate::core::MetricCollector;
+use crate::modules::disk::DiskCollector;
 use crate::modules::{cpu::CpuCollector, memory::MemoryCollector};
 use crate::output::{OutputFormat, format_output};
 
 pub fn run() {
     let args = cli::parse_args();
 
     // get list of collectors based on args
-    let collectors: Vec<Box<dyn MetricCollector>> = match args.module.as_deref() {
-        Some("cpu") => vec![Box::new(CpuCollector::new())],
-        Some("memory") => vec![Box::new(MemoryCollector::new())],
-        Some(_) => {
-            eprintln!("Unknown module: {}", args.module.unwrap());
-            std::process::exit(1);
-        }
-        None => {
-            // default to all collectors
-            vec![
-                Box::new(CpuCollector::new()),
-                Box::new(MemoryCollector::new()),
-            ]
-        }
-    };
+   let collectors: Vec<Box<dyn MetricCollector>> = match args.module.as_deref() {
+    Some("cpu") => vec![Box::new(CpuCollector::new())],
+    Some("memory") => vec![Box::new(MemoryCollector::new())],
+    Some("disk") => vec![Box::new(DiskCollector::new())],
+    Some(_) => {
+        eprintln!("Unknown module: {}", args.module.unwrap());
+        std::process::exit(1);
+    }
+    None => {
+        vec![
+            Box::new(CpuCollector::new()),
+            Box::new(MemoryCollector::new()),
+            Box::new(DiskCollector::new()),    // default me disk bhi add
+        ]
+    }
+};
+
 
     // determine output format
     let format = match args.output.as_deref() {

src/modules/disk.rs

@@ -0,0 +1,43 @@
+use sysinfo::{DiskExt, System, SystemExt};
+use std::collections::HashMap;
+use crate::core::{MetricCollector, MetricData, MetricValue};
+
+pub struct DiskCollector;
+
+impl DiskCollector {
+    pub fn new() -> Self {
+        DiskCollector
+    }
+}
+
+impl MetricCollector for DiskCollector {
+    fn collect(&self) -> Result<MetricData, Box<dyn std::error::Error>> {
+        let mut sys = System::new_all();
+        sys.refresh_disks_list();
+
+        let mut total: u64 = 0;
+        let mut used: u64 = 0;
+        let mut free: u64 = 0;
+
+        for disk in sys.disks() {
+            total += disk.total_space();
+            free += disk.available_space();
+        }
+
+        used = total - free;
+
+        let mut map = HashMap::new();
+        map.insert("total".to_string(), MetricValue::Integer(total as i64));
+        map.insert("used".to_string(), MetricValue::Integer(used as i64));
+        map.insert("free".to_string(), MetricValue::Integer(free as i64));
+
+        Ok(MetricData {
+            timestamp: std::time::SystemTime::now(),
+            metrics: map,
+        })
+    }
+
+    fn name(&self) -> &'static str {
+        "disk"
+    }
+}

src/modules/mod.rs

@@ -1,2 +1,3 @@
 pub mod cpu;
-pub mod memory;
+pub mod memory;
+pub mod disk;