mirror of
https://github.com/serai-dex/serai.git
synced 2025-12-08 20:29:23 +00:00
6d5851a9ee
zstd was recommended for the base layer only, due to its CPU requirements. That was a misreading on mhy behalf. lz4 gets ~5% better compression than snappy with ~30% faster performance. zstd does ~25% better than lz4 yet at ~30% of the performance.
43 lines
1.5 KiB
Rust
43 lines
1.5 KiB
Rust
use std::sync::Arc;
|
|
|
|
use rocksdb::{DBCompressionType, ThreadMode, SingleThreaded, Options, Transaction, TransactionDB};
|
|
|
|
use crate::*;
|
|
|
|
impl<T: ThreadMode> Get for Transaction<'_, TransactionDB<T>> {
|
|
fn get(&self, key: impl AsRef<[u8]>) -> Option<Vec<u8>> {
|
|
self.get(key).expect("couldn't read from RocksDB via transaction")
|
|
}
|
|
}
|
|
impl<T: ThreadMode> DbTxn for Transaction<'_, TransactionDB<T>> {
|
|
fn put(&mut self, key: impl AsRef<[u8]>, value: impl AsRef<[u8]>) {
|
|
Transaction::put(self, key, value).expect("couldn't write to RocksDB via transaction")
|
|
}
|
|
fn del(&mut self, key: impl AsRef<[u8]>) {
|
|
self.delete(key).expect("couldn't delete from RocksDB via transaction")
|
|
}
|
|
fn commit(self) {
|
|
Transaction::commit(self).expect("couldn't commit to RocksDB via transaction")
|
|
}
|
|
}
|
|
|
|
impl<T: ThreadMode> Get for Arc<TransactionDB<T>> {
|
|
fn get(&self, key: impl AsRef<[u8]>) -> Option<Vec<u8>> {
|
|
TransactionDB::get(self, key).expect("couldn't read from RocksDB")
|
|
}
|
|
}
|
|
impl<T: ThreadMode + 'static> Db for Arc<TransactionDB<T>> {
|
|
type Transaction<'a> = Transaction<'a, TransactionDB<T>>;
|
|
fn txn(&mut self) -> Self::Transaction<'_> {
|
|
self.transaction()
|
|
}
|
|
}
|
|
|
|
pub type RocksDB = Arc<TransactionDB<SingleThreaded>>;
|
|
pub fn new_rocksdb(path: &str) -> RocksDB {
|
|
let mut options = Options::default();
|
|
options.create_if_missing(true);
|
|
options.set_compression_type(DBCompressionType::Lz4);
|
|
Arc::new(TransactionDB::open(&options, &Default::default(), path).unwrap())
|
|
}
|