piperocksdb 0.1.5

A pipebase plugin using rocksdb rust wrapper
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

use pipebase::common::LeftRight;
use pipebytes::{AsBytes, FromBytes, IntoBytes};
use rocksdb::{DBWithThreadMode, SingleThreaded, WriteBatch, DB};

pub struct RocksDBClient {
    db: DBWithThreadMode<SingleThreaded>,
}

impl RocksDBClient {
    pub fn new(path: &str) -> anyhow::Result<Self> {
        let db = DB::open_default(path)?;
        Ok(RocksDBClient { db })
    }

    pub fn get<K, V>(&self, key: &K) -> anyhow::Result<Option<V>>
    where
        K: AsBytes,
        V: FromBytes,
    {
        match self.db.get(key.as_bytes()?)? {
            Some(bytes) => Ok(Some(V::from_bytes(bytes)?)),
            None => Ok(None),
        }
    }

    pub fn put_all<K, V, T, U>(&mut self, entries: U) -> anyhow::Result<()>
    where
        K: IntoBytes,
        V: IntoBytes,
        T: LeftRight<L = K, R = V>,
        U: IntoIterator<Item = T>,
    {
        let mut batch = WriteBatch::default();
        for entry in entries.into_iter() {
            let (key, value) = entry.into_tuple();
            batch.put(key.into_bytes()?, value.into_bytes()?);
        }
        self.db.write(batch)?;
        Ok(())
    }
}