wal recovery mode (#34)

librocksdb_sys/src/lib.rs

@@ -96,6 +96,15 @@ pub enum DBUniversalCompactionStyle {
     rocksdb_total_size_compaction_stop_style = 1,
 }
 
+#[derive(Copy, Clone)]
+#[repr(C)]
+pub enum DBRecoveryMode {
+    TolerateCorruptedTailRecords = 0,
+    AbsoluteConsistency = 1,
+    PointInTime = 2,
+    SkipAnyCorruptedRecords = 3,
+}
+
 pub fn error_message(ptr: *const i8) -> String {
     let c_str = unsafe { CStr::from_ptr(ptr as *const _) };
     let s = from_utf8(c_str.to_bytes()).unwrap().to_owned();
@@ -200,6 +209,7 @@ extern "C" {
     pub fn rocksdb_options_set_disable_auto_compactions(options: DBOptions,
                                                         v: c_int);
     pub fn rocksdb_options_set_report_bg_io_stats(options: DBOptions, v: c_int);
+    pub fn rocksdb_options_set_wal_recovery_mode(options: DBOptions, mode: DBRecoveryMode);
     pub fn rocksdb_filterpolicy_create_bloom_full(bits_per_key: c_int)
                                                 -> DBFilterPolicy;
     pub fn rocksdb_filterpolicy_create_bloom(bits_per_key: c_int)

src/lib.rs

@@ -24,7 +24,7 @@ pub mod rocksdb_options;
 pub mod merge_operator;
 pub mod comparator;
 
-pub use librocksdb_sys::{DBCompactionStyle, DBComparator, DBCompressionType,
+pub use librocksdb_sys::{DBCompactionStyle, DBComparator, DBCompressionType, DBRecoveryMode,
                          new_bloom_filter, self as rocksdb_ffi};
 pub use rocksdb::{DB, DBIterator, DBVector, Kv, ReadOptions, SeekKey,
                   Writable, WriteBatch};

src/main.rs

@@ -145,6 +145,7 @@ fn main() {
 mod tests {
     use rocksdb::{BlockBasedOptions, DB, DBCompressionType, Options, ReadOptions, WriteOptions, SeekKey};
     use rocksdb::DBCompactionStyle::DBUniversal;
+    use rocksdb::DBRecoveryMode;
 
     #[allow(dead_code)]
     fn tuned_for_somebody_elses_disk(path: &str,
@@ -179,6 +180,7 @@ mod tests {
         opts.set_max_background_flushes(4);
         opts.set_filter_deletes(false);
         opts.set_report_bg_io_stats(true);
+        opts.set_wal_recovery_mode(DBRecoveryMode::PointInTime);
         opts.compression_per_level(&per_level_compression);
         blockopts.set_block_size(524288);
         blockopts.set_cache_index_and_filter_blocks(true);

src/rocksdb_options.rs

@@ -16,7 +16,7 @@ use libc::{c_int, size_t};
 use std::ffi::CString;
 use std::mem;
 
-use rocksdb_ffi::{self, DBCompressionType};
+use rocksdb_ffi::{self, DBCompressionType, DBRecoveryMode};
 use merge_operator::{self, MergeOperatorCallback, full_merge_callback,
                      partial_merge_callback};
 use comparator::{self, ComparatorCallback, compare_callback};
@@ -409,6 +409,12 @@ impl Options {
             }
         }
     }
+
+    pub fn set_wal_recovery_mode(&mut self, mode: DBRecoveryMode) {
+        unsafe {
+            rocksdb_ffi::rocksdb_options_set_wal_recovery_mode(self.inner, mode);
+        }
+    }
 }
 
 impl Default for WriteOptions {