fluidB
/
fluidb
1
0
Fork 0
Code Pull Requests Activity

macro

Browse Source
main
Ziyang Hu 2 years ago
parent eddf6a5573
commit b9ebb8e533
1 changed files with 86 additions and 81 deletions
Show Stats Download Patch File Download Diff File

167
cozorocks/src/lib.rs
Unescape Escape View File

@ -31,6 +31,7 @@ include_cpp! {
}
pub use ffi::rocksdb::Options;
pub use ffi::rocksdb::PinnableSlice;
pub use ffi::rocksdb::ReadOptions;
pub use ffi::rocksdb::Slice;
pub use ffi::rocksdb::Snapshot;
@ -50,6 +51,7 @@ pub struct DbStatus {
pub severity: StatusSeverity,
}
#[inline(always)]
fn convert_status(status: &ffi::rocksdb::Status) -> DbStatus {
let code = status.code();
let subcode = status.subcode();
@ -61,113 +63,112 @@ fn convert_status(status: &ffi::rocksdb::Status) -> DbStatus {
}
}
fn convert_slice(src: impl AsRef<[u8]>) -> Slice {
let src = src.as_ref();
#[inline(always)]
fn convert_slice(src: &[u8]) -> Slice {
Slice {
data_: src.as_ptr() as *const c_char,
size_: src.len(),
}
}
#[inline(always)]
fn convert_slice_back(src: &Slice) -> &[u8] {
unsafe { std::slice::from_raw_parts(src.data() as *const u8, src.size()) }
}
#[inline]
pub fn put(
db: Pin<&mut DB>,
opts: &WriteOptions,
key: impl AsRef<[u8]>,
val: impl AsRef<[u8]>,
) -> DbStatus {
let key = convert_slice(key);
let val = convert_slice(val);
let key = convert_slice(key.as_ref());
let val = convert_slice(val.as_ref());
moveit! { let status = db.Put2(opts, &key, &val); }
convert_status(&status)
}
#[macro_export]
macro_rules! let_pinnable_slice {
($i:ident) => {
$crate::moveit! {
let mut $i = $crate::PinnableSlice::new();
}
};
}
#[macro_export]
macro_rules! let_write_opts {
($i:ident = [$( $opt:ident ),*]) => {
($i:ident = {$( $opt_name:ident => $opt_val:expr ),*}) => {
$crate::moveit! {
let mut $i = $crate::WriteOptions::new();
}
$(
match stringify!($opt) {
"sync" => {
$crate::ffi::rocksdb_additions::set_w_opts_sync($i.as_mut(), true);
}
"no_sync" => {
$crate::ffi::rocksdb_additions::set_w_opts_sync($i.as_mut(), false);
}
"disable_wal" => {
$crate::ffi::rocksdb_additions::set_w_opts_disable_wal($i.as_mut(), true);
}
"no_disable_wal" => {
$crate::ffi::rocksdb_additions::set_w_opts_disable_wal($i.as_mut(), false);
}
"low_pri" => {
$crate::ffi::rocksdb_additions::set_w_opts_low_pri($i.as_mut(), true);
}
"no_set_w_opts_low_pri" => {
$crate::ffi::rocksdb_additions::set_w_opts_low_pri($i.as_mut(), false);
}
_ => panic!("unknown option to let_write_opts: {}", stringify!($i))
};
)*
}
let_write_opts! { @let_opts, $i, $( $opt_name, $opt_val, )* }
};
( @let_opts, $i:ident, ) => {};
( @let_opts, $i:ident, sync, $val:expr, $( $opt_name:ident, $opt_val:expr, )* ) => {
$crate::ffi::rocksdb_additions::set_w_opts_sync($i.as_mut(), $val);
let_write_opts! { @let_opts, $i, $( $opt_name, $opt_val, )* }
};
( @let_opts, $i:ident, disable_wal, $val:expr, $( $opt_name:ident, $opt_val:expr, )* ) => {
$crate::ffi::rocksdb_additions::set_w_opts_disable_wal($i.as_mut(), $val);
let_write_opts! { @let_opts, $i, $( $opt_name, $opt_val, )* }
};
( @let_opts, $i:ident, low_pri, $val:expr, $( $opt_name:ident, $opt_val:expr, )* ) => {
$crate::ffi::rocksdb_additions::set_w_opts_low_pri($i.as_mut(), $val);
let_write_opts! { @let_opts, $i, $( $opt_name, $opt_val, )* }
};
}
#[macro_export]
macro_rules! let_read_opts {
($i:ident = [$( $opt:ident ),*]) => {
($i:ident = {$( $opt_name:ident => $opt_val:expr ),*}) => {
$crate::moveit! {
let mut $i = $crate::ReadOptions::new();
}
$(
match stringify!($opt) {
"total_order_seek" => {
$crate::ffi::rocksdb_additions::set_r_opts_total_order_seek($i.as_mut(), true);
}
"no_total_order_seek" => {
$crate::ffi::rocksdb_additions::set_r_opts_total_order_seek($i.as_mut(), false);
}
"auto_prefix_mode" => {
$crate::ffi::rocksdb_additions::set_r_opts_auto_prefix_mode($i.as_mut(), true);
}
"no_auto_prefix_mode" => {
$crate::ffi::rocksdb_additions::set_r_opts_auto_prefix_mode($i.as_mut(), false);
}
"prefix_same_as_start" => {
$crate::ffi::rocksdb_additions::set_r_opts_prefix_same_as_start($i.as_mut(), true);
}
"no_prefix_same_as_start" => {
$crate::ffi::rocksdb_additions::set_r_opts_prefix_same_as_start($i.as_mut(), false);
}
"tailing" => {
$crate::ffi::rocksdb_additions::set_r_opts_tailing($i.as_mut(), true);
}
"no_tailing" => {
$crate::ffi::rocksdb_additions::set_r_opts_tailing($i.as_mut(), false);
}
"pin_data" => {
$crate::ffi::rocksdb_additions::set_r_opts_pin_data($i.as_mut(), true);
}
"no_pin_data" => {
$crate::ffi::rocksdb_additions::set_r_opts_pin_data($i.as_mut(), false);
}
"verify_checksums" => {
$crate::ffi::rocksdb_additions::set_r_opts_verify_checksums($i.as_mut(), true);
}
"no_verify_checksums" => {
$crate::ffi::rocksdb_additions::set_r_opts_verify_checksums($i.as_mut(), false);
}
"fill_cache" => {
$crate::ffi::rocksdb_additions::set_r_opts_fill_cache($i.as_mut(), true);
}
"fill_cache" => {
$crate::ffi::rocksdb_additions::set_r_opts_fill_cache($i.as_mut(), false);
}
_ => panic!("unknown option to let_read_opts: {}", stringify!($i))
};
)*
let_read_opts! { @let_opts, $i, $( $opt_name, $opt_val, )* }
};
( @let_opts, $i:ident, ) => {};
( @let_opts, $i:ident, lower_bound, $val:expr, $( $opt_name:ident, $opt_val:expr, )* ) => {
$crate::ffi::rocksdb_additions::set_iterate_lower_bound($i.as_mut(), &$crate::convert_slice($val.as_ref()));
let_read_opts! { @let_opts, $i, $( $opt_name, $opt_val, )* }
};
( @let_opts, $i:ident, upper_bound, $val:expr, $( $opt_name:ident, $opt_val:expr, )* ) => {
$crate::ffi::rocksdb_additions::set_iterate_upper_bound($i.as_mut(), &$crate::convert_slice($val.as_ref()));
let_read_opts! { @let_opts, $i, $( $opt_name, $opt_val, )* }
};
( @let_opts, $i:ident, snapshot, $val:expr, $( $opt_name:ident, $opt_val:expr, )* ) => {
$crate::ffi::rocksdb_additions::set_snapshot($i.as_mut(), $val);
let_read_opts! { @let_opts, $i, $( $opt_name, $opt_val, )* }
};
( @let_opts, $i:ident, total_order_seek, $val:expr, $( $opt_name:ident, $opt_val:expr, )* ) => {
$crate::ffi::rocksdb_additions::set_r_opts_total_order_seek($i.as_mut(), $val);
let_read_opts! { @let_opts, $i, $( $opt_name, $opt_val, )* }
};
( @let_opts, $i:ident, auto_prefix_mode, $val:expr, $( $opt_name:ident, $opt_val:expr, )* ) => {
$crate::ffi::rocksdb_additions::set_r_opts_auto_prefix_mode($i.as_mut(), $val);
let_read_opts! { @let_opts, $i, $( $opt_name, $opt_val, )* }
};
( @let_opts, $i:ident, prefix_same_as_start, $val:expr, $( $opt_name:ident, $opt_val:expr, )* ) => {
$crate::ffi::rocksdb_additions::set_r_opts_prefix_same_as_start($i.as_mut(), $val);
let_read_opts! { @let_opts, $i, $( $opt_name, $opt_val, )* }
};
( @let_opts, $i:ident, tailing, $val:expr, $( $opt_name:ident, $opt_val:expr, )* ) => {
$crate::ffi::rocksdb_additions::set_r_opts_tailing($i.as_mut(), $val);
let_read_opts! { @let_opts, $i, $( $opt_name, $opt_val, )* }
};
( @let_opts, $i:ident, pin_data, $val:expr, $( $opt_name:ident, $opt_val:expr, )* ) => {
$crate::ffi::rocksdb_additions::set_r_opts_pin_data($i.as_mut(), $val);
let_read_opts! { @let_opts, $i, $( $opt_name, $opt_val, )* }
};
( @let_opts, $i:ident, verify_checksums, $val:expr, $( $opt_name:ident, $opt_val:expr, )* ) => {
$crate::ffi::rocksdb_additions::set_r_opts_verify_checksums($i.as_mut(), $val);
let_read_opts! { @let_opts, $i, $( $opt_name, $opt_val, )* }
};
( @let_opts, $i:ident, fill_cache, $val:expr, $( $opt_name:ident, $opt_val:expr, )* ) => {
$crate::ffi::rocksdb_additions::set_r_opts_fill_cache($i.as_mut(), $val);
let_read_opts! { @let_opts, $i, $( $opt_name, $opt_val, )* }
};
}
@ -184,8 +185,12 @@ mod tests {
let mut g_opts = Options::new().within_unique_ptr();
g_opts.pin_mut().OptimizeForSmallDb();
let_read_opts!(r_opts = []);
let_write_opts!(w_opts = [disable_wal]);
let lower_bound = "lower".as_bytes();
let upper_bound = "upper".as_bytes();
let should_tail = true;
let_read_opts!(r_opts = { lower_bound => lower_bound, tailing => should_tail, upper_bound => upper_bound });
let_write_opts!(w_opts = { disable_wal => true });
}
dbg!(size_of::<ReadOptions>());
}

Write Preview
Loading…
Cancel
Save