options: add compression option

librocksdb_sys/src/lib.rs

@@ -178,7 +178,7 @@ extern "C" {
     pub fn rocksdb_options_set_compaction_style(options: DBOptions,
                                                 cs: DBCompactionStyle);
     pub fn rocksdb_options_set_compression(options: DBOptions,
-                                           compression_style_no: c_int);
+                                           compression_style_no: DBCompressionType);
     pub fn rocksdb_options_set_max_background_compactions(
         options: DBOptions, max_bg_compactions: c_int);
     pub fn rocksdb_options_set_max_background_flushes(options: DBOptions,

src/lib.rs

@@ -26,7 +26,7 @@ pub mod merge_operator;
 pub mod comparator;
 
 pub use librocksdb_sys::{DBCompactionStyle, DBComparator, new_bloom_filter,
-                         self as rocksdb_ffi};
+                         self as rocksdb_ffi, DBCompressionType};
 pub use rocksdb::{DB, DBIterator, DBVector, Direction, IteratorMode, Writable,
                   WriteBatch};
 pub use rocksdb_options::{BlockBasedOptions, Options, WriteOptions};

src/rocksdb_options.rs

@@ -16,7 +16,7 @@ use libc::{c_int, size_t};
 use std::ffi::CString;
 use std::mem;
 
-use rocksdb_ffi;
+use rocksdb_ffi::{self, DBCompressionType};
 use merge_operator::{self, MergeOperatorCallback, full_merge_callback,
                      partial_merge_callback};
 use comparator::{self, ComparatorCallback, compare_callback};
@@ -148,6 +148,12 @@ impl Options {
         }
     }
     
+    pub fn compression(&mut self, t: DBCompressionType) {
+        unsafe {
+            rocksdb_ffi::rocksdb_options_set_compression(self.inner, t);
+        }
+    }
+
     pub fn add_merge_operator(&mut self, name: &str, merge_fn: MergeFn) {
         let cb = Box::new(MergeOperatorCallback {
             name: CString::new(name.as_bytes()).unwrap(),