view src/env.rs @ 166:ef5426303667

Simplify in-memory implementation of RandomAccess file.
author Lewin Bormann <lbo@spheniscida.de>
date Sun, 16 Jul 2017 10:56:40 +0200
parents 54f8c4f68b99
children 9e7e1a943718
line wrap: on
line source

//! An `env` is an abstraction layer that allows the database to run both on different platforms as
//! well as persisting data on disk or in memory.

use error::{self, Result};

use std::io::prelude::*;
use std::fs::File;
use std::os::unix::fs::FileExt;
use std::path::Path;
use std::sync::Mutex;

pub trait RandomAccess {
    fn read_at(&self, off: usize, len: usize) -> Result<Vec<u8>>;
}

impl RandomAccess for File {
    fn read_at(&self, off: usize, len: usize) -> Result<Vec<u8>> {
        let mut buf = vec![0 as u8; len];
        error::from_io_result((self as &FileExt).read_at(buf.as_mut(), off as u64)).map(|_| buf)
    }
}

/// BufferBackedFile implements RandomAccess on a cursor. It wraps the cursor in a mutex
/// to enable using an immutable receiver, like the File implementation.
pub type BufferBackedFile = Mutex<Vec<u8>>;

impl RandomAccess for BufferBackedFile {
    fn read_at(&self, off: usize, len: usize) -> Result<Vec<u8>> {
        let c = self.lock().unwrap();
        if off + len > c.len() {
            return Err(error::Status::new(error::StatusCode::InvalidArgument, "off-limits read"));
        }
        Ok(c[off..off + len].to_vec())
    }
}

pub struct FileLock {
    pub id: String,
}

pub trait Env {
    fn open_sequential_file(&self, &Path) -> Result<Box<Read>>;
    fn open_random_access_file(&self, &Path) -> Result<Box<RandomAccess>>;
    fn open_writable_file(&self, &Path) -> Result<Box<Write>>;
    fn open_appendable_file(&self, &Path) -> Result<Box<Write>>;

    fn exists(&self, &Path) -> Result<bool>;
    fn children(&self, &Path) -> Result<Vec<String>>;
    fn size_of(&self, &Path) -> Result<usize>;

    fn delete(&self, &Path) -> Result<()>;
    fn mkdir(&self, &Path) -> Result<()>;
    fn rmdir(&self, &Path) -> Result<()>;
    fn rename(&self, &Path, &Path) -> Result<()>;

    fn lock(&self, &Path) -> Result<FileLock>;
    fn unlock(&self, l: FileLock);

    fn new_logger(&self, &Path) -> Result<Logger>;

    fn micros(&self) -> u64;
    fn sleep_for(&self, micros: u32);
}

pub struct Logger {
    dst: Box<Write>,
}

impl Logger {
    pub fn new(w: Box<Write>) -> Logger {
        Logger { dst: w }
    }

    pub fn log(&mut self, message: &String) {
        let _ = self.dst.write(message.as_bytes());
        let _ = self.dst.write("\n".as_bytes());
    }
}