1 files changed, 76 insertions, 0 deletions
diff --git a/src/file_sizes/file_sizes_default.rs b/src/file_sizes/file_sizes_default.rs
new file mode 100644
index 0000000..bed8584
--- /dev/null
+++ b/src/file_sizes/file_sizes_default.rs
@@ -0,0 +1,76 @@
+//! size computation for non linux
+
+use crate::task_sync::TaskLifetime;
+use crossbeam::channel::unbounded;
+use crossbeam::sync::WaitGroup;
+use std::fs;
+use std::path::{Path, PathBuf};
+use std::sync::atomic::{AtomicIsize, AtomicUsize, Ordering};
+use std::sync::Arc;
+use std::thread;
+use std::time::Duration;
+
+pub fn compute_dir_size(path: &Path, tl: &TaskLifetime) -> Option<u64> {
+    let size = Arc::new(AtomicUsize::new(0));
+
+    // this MPMC channel contains the directory paths which must be handled
+    let (dirs_sender, dirs_receiver) = unbounded();
+
+    // busy is the number of directories which are either being processed or queued
+    // We use this count to determine when threads can stop waiting for tasks
+    let busy = Arc::new(AtomicIsize::new(0));
+    busy.fetch_add(1, Ordering::Relaxed);
+    dirs_sender.send(Some(PathBuf::from(path))).unwrap();
+
+    let wg = WaitGroup::new();
+    let period = Duration::from_micros(50);
+    for _ in 0..8 {
+        let size = Arc::clone(&size);
+        let busy = Arc::clone(&busy);
+        let wg = wg.clone();
+        let (dirs_sender, dirs_receiver) = (dirs_sender.clone(), dirs_receiver.clone());
+        let tl = tl.clone();
+        thread::spawn(move || {
+            loop {
+                let o = dirs_receiver.recv_timeout(period);
+                if let Ok(Some(open_dir)) = o {
+                    if let Ok(entries) = fs::read_dir(&open_dir) {
+                        for e in entries.flatten() {
+                            if let Ok(md) = e.metadata() {
+                                if md.is_dir() {
+                                    busy.fetch_add(1, Ordering::Relaxed);
+                                    dirs_sender.send(Some(e.path())).unwrap();
+                                }
+                                size.fetch_add(md.len() as usize, Ordering::Relaxed);
+                            }
+                        }
+                    }
+                    busy.fetch_sub(1, Ordering::Relaxed);
+                    dirs_sender.send(None).unwrap();
+                } else if busy.load(Ordering::Relaxed) < 1 {
+                    break;
+                }
+                if tl.is_expired() {
+                    break;
+                }
+            }
+            drop(wg);
+        });
+    }
+    wg.wait();
+
+    if tl.is_expired() {
+        return None;
+    }
+    let size: usize = size.load(Ordering::Relaxed);
+    let size: u64 = size as u64;
+    Some(size)
+}
+
+
+pub fn compute_file_size(path: &Path) -> u64 {
+    match fs::metadata(path) {
+        Ok(m) => m.len(),
+        Err(_) => 0,
+    }
+}