set num levels (#45)

librocksdb_sys/src/lib.rs

@@ -199,6 +199,7 @@ extern "C" {
     pub fn rocksdb_options_set_wal_recovery_mode(options: *mut DBOptions, mode: DBRecoveryMode);
     pub fn rocksdb_options_enable_statistics(options: *mut DBOptions);
     pub fn rocksdb_options_set_stats_dump_period_sec(options: *mut DBOptions, v: usize);
+    pub fn rocksdb_options_set_num_levels(options: *mut DBOptions, v: c_int);
     pub fn rocksdb_filterpolicy_create_bloom_full(bits_per_key: c_int)
                                                 -> *mut DBFilterPolicy;
     pub fn rocksdb_filterpolicy_create_bloom(bits_per_key: c_int)

src/rocksdb_options.rs

@@ -563,6 +563,12 @@ impl Options {
                                                       period);
         }
     }
+
+    pub fn set_num_levels(&mut self, n: c_int) {
+        unsafe {
+            rocksdb_ffi::rocksdb_options_set_num_levels(self.inner, n);
+        }
+    }
 }
 
 pub struct FlushOptions {

test/test.rs

@@ -6,3 +6,4 @@ mod test_multithreaded;
 mod test_column_family;
 mod test_compaction_filter;
 mod test_compact_range;
+mod test_rocksdb_options;

test/test_rocksdb_options.rs

+use tempdir::TempDir;
+use rocksdb::{DB, Options};
+
+
+#[test]
+fn test_set_num_levels() {
+	let path = TempDir::new("_rust_rocksdb_test_set_num_levels").expect("");
+	let mut opts = Options::new();
+	opts.create_if_missing(true);
+	opts.set_num_levels(2);
+	let db = DB::open(&opts, path.path().to_str().unwrap()).unwrap();
+	drop(db);
+}