Merge pull request #10 from BusyJay/busyjay/add-base-size-opt

option: add max_bytes_for_level_base option

librocksdb_sys/src/lib.rs

@@ -147,6 +147,7 @@ extern "C" {
     pub fn rocksdb_options_set_bytes_per_sync(options: DBOptions, bytes: u64);
     pub fn rocksdb_options_set_disable_data_sync(options: DBOptions,
                                                  v: c_int);
+    pub fn rocksdb_options_set_allow_os_buffer(options: DBOptions, is_allow: bool);
     pub fn rocksdb_options_optimize_for_point_lookup(options: DBOptions,
                                                      block_cache_size_mb: u64);
     pub fn rocksdb_options_set_table_cache_numshardbits(options: DBOptions,
@@ -167,6 +168,8 @@ extern "C" {
                                                      bytes: u64);
     pub fn rocksdb_options_set_target_file_size_multiplier(options: DBOptions,
                                                            mul: c_int);
+    pub fn rocksdb_options_set_max_bytes_for_level_base(options: DBOptions, bytes: u64);
+    pub fn rocksdb_options_set_max_bytes_for_level_multiplier(options: DBOptions, mul: c_int);
     pub fn rocksdb_options_set_max_log_file_size(options: DBOptions,
                                                  bytes: u64);
     pub fn rocksdb_options_set_max_manifest_file_size(options: DBOptions,

src/rocksdb_options.rs

@@ -82,6 +82,15 @@ impl BlockBasedOptions {
                                                                     size);
         }
     }
+
+    pub fn set_lru_cache(&mut self, size: size_t) {
+        let cache = rocksdb_ffi::new_cache(size);
+        unsafe {
+            // because cache is wrapped in shared_ptr, so we don't need to call
+            // rocksdb_cache_destroy explicitly.
+            rocksdb_ffi::rocksdb_block_based_options_set_block_cache(self.inner, cache);
+        }
+    }
 }
 
 // TODO figure out how to create these in a Rusty way
@@ -232,6 +241,13 @@ impl Options {
         }
     }
 
+    pub fn allow_os_buffer(&mut self, is_allow: bool) {
+        unsafe {
+            rocksdb_ffi::rocksdb_options_set_allow_os_buffer(self.inner,
+                                                             is_allow);
+        }
+    }
+
     pub fn set_table_cache_num_shard_bits(&mut self, nbits: c_int) {
         unsafe {
             rocksdb_ffi::rocksdb_options_set_table_cache_numshardbits(self.inner,
@@ -260,6 +276,18 @@ impl Options {
         }
     }
 
+    pub fn set_max_bytes_for_level_base(&mut self, size: u64) {
+        unsafe {
+            rocksdb_ffi::rocksdb_options_set_max_bytes_for_level_base(self.inner, size);
+        }
+    }
+
+    pub fn set_max_bytes_for_level_multiplier(&mut self, mul: i32) {
+        unsafe {
+            rocksdb_ffi::rocksdb_options_set_max_bytes_for_level_multiplier(self.inner, mul);
+        }
+    }
+
     pub fn set_target_file_size_base(&mut self, size: u64) {
         unsafe {
             rocksdb_ffi::rocksdb_options_set_target_file_size_base(self.inner,