flash-kv 0.2.1

A simple k/v store API inspired by bitcask
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44

use std::{
  fs, io,
  path::{Path, PathBuf},
};

// calculate available disk space
pub fn available_disk_space() -> u64 {
  fs2::available_space(PathBuf::from("/")).unwrap_or_default()
}

// calculate the total size of directory in disk
pub fn dir_disk_size<P: AsRef<Path>>(dir_path: P) -> u64 {
  fs_extra::dir::get_size(dir_path).unwrap_or_default()
}

pub fn copy_dir<P: AsRef<Path>>(src: P, dst: P, exclude: &[&str]) -> io::Result<()> {
  //
  if !dst.as_ref().exists() {
    fs::create_dir_all(&dst)?;
  }

  for dir_entry in fs::read_dir(&src)? {
    let entry = dir_entry?;
    let src_path = entry.path();

    if exclude.iter().any(|&x| src_path.ends_with(x)) {
      continue;
    }

    let dst_path = dst.as_ref().join(src_path.file_name().unwrap());
    if entry.file_type()?.is_dir() {
      copy_dir(src_path, dst_path, exclude)?;
    } else {
      fs::copy(&src_path, &dst_path)?;
    }
  }
  Ok(())
}

#[test]
fn test_available_disk_space() {
  let size = available_disk_space();
  assert!(size > 0);
}