kitedb 0.2.18

High-performance embedded graph database
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
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69

//! Lightweight mmap abstraction for native + wasm builds.
//!
//! On native targets, this is a thin wrapper over memmap2::Mmap.
//! On wasm32-wasi, we fall back to reading the full file into memory.

use std::fs::File;

#[cfg(target_arch = "wasm32")]
use std::io::{Read, Seek, SeekFrom};
#[cfg(target_arch = "wasm32")]
use std::sync::Arc;

#[cfg(not(target_arch = "wasm32"))]
pub type Mmap = memmap2::Mmap;

#[cfg(target_arch = "wasm32")]
#[derive(Clone)]
pub struct Mmap {
  data: Arc<Vec<u8>>,
}

#[cfg(target_arch = "wasm32")]
impl Mmap {
  /// Read the entire file into memory.
  pub fn map(file: &File) -> std::io::Result<Self> {
    let mut handle = file.try_clone()?;
    handle.seek(SeekFrom::Start(0))?;
    let mut buffer = Vec::new();
    handle.read_to_end(&mut buffer)?;
    Ok(Self {
      data: Arc::new(buffer),
    })
  }

  pub fn len(&self) -> usize {
    self.data.len()
  }
}

#[cfg(target_arch = "wasm32")]
impl std::ops::Deref for Mmap {
  type Target = [u8];

  fn deref(&self) -> &Self::Target {
    &self.data
  }
}

/// Map a file into memory (native uses unsafe mmap, wasm reads to memory).
///
/// # Safety
/// Callers must ensure the file is not mutated while the mapping is live.
pub fn map_file(file: &File) -> std::io::Result<Mmap> {
  #[cfg(not(target_arch = "wasm32"))]
  unsafe {
    let len = file.metadata()?.len();
    if len == 0 {
      return Err(std::io::Error::new(
        std::io::ErrorKind::InvalidInput,
        "cannot mmap an empty file",
      ));
    }
    Mmap::map(file)
  }
  #[cfg(target_arch = "wasm32")]
  {
    Mmap::map(file)
  }
}