Add vector memtable option (#181)

librocksdb_sys/crocksdb/c.cc

@@ -129,6 +129,7 @@ using rocksdb::ExternalSstFilePropertyNames;
 using rocksdb::DecodeFixed32;
 using rocksdb::DecodeFixed64;
 using rocksdb::PutFixed64;
+using rocksdb::VectorRepFactory;
 
 using std::shared_ptr;
 
@@ -2457,6 +2458,10 @@ void crocksdb_options_set_ratelimiter(crocksdb_options_t *opt, crocksdb_ratelimi
   limiter->rep = nullptr;
 }
 
+void crocksdb_options_set_vector_memtable_factory(crocksdb_options_t* opt, uint64_t reserved_bytes) {
+  opt->rep.memtable_factory.reset(new VectorRepFactory(reserved_bytes));
+}
+
 crocksdb_ratelimiter_t* crocksdb_ratelimiter_create(
     int64_t rate_bytes_per_sec,
     int64_t refill_period_us,

librocksdb_sys/crocksdb/crocksdb/c.h

@@ -977,6 +977,8 @@ extern C_ROCKSDB_LIBRARY_API void crocksdb_options_set_fifo_compaction_options(
     crocksdb_options_t* opt, crocksdb_fifo_compaction_options_t* fifo);
 extern C_ROCKSDB_LIBRARY_API void crocksdb_options_set_ratelimiter(
     crocksdb_options_t* opt, crocksdb_ratelimiter_t* limiter);
+extern C_ROCKSDB_LIBRARY_API void crocksdb_options_set_vector_memtable_factory(
+    crocksdb_options_t* opt, uint64_t reserved_bytes);
 
 enum {
   compaction_by_compensated_size = 0,

librocksdb_sys/src/lib.rs

@@ -547,6 +547,7 @@ extern "C" {
         target_size: *const u64,
         num_paths: c_int,
     );
+    pub fn crocksdb_options_set_vector_memtable_factory(options: *mut Options, reserved_bytes: u64);
     pub fn crocksdb_filterpolicy_create_bloom_full(bits_per_key: c_int) -> *mut DBFilterPolicy;
     pub fn crocksdb_filterpolicy_create_bloom(bits_per_key: c_int) -> *mut DBFilterPolicy;
     pub fn crocksdb_open(

src/rocksdb_options.rs

@@ -1270,6 +1270,12 @@ impl ColumnFamilyOptions {
             crocksdb_ffi::crocksdb_options_set_fifo_compaction_options(self.inner, fifo_opts.inner);
         }
     }
+
+    pub fn set_vector_memtable_factory(&mut self, reserved_bytes: u64) {
+        unsafe {
+            crocksdb_ffi::crocksdb_options_set_vector_memtable_factory(self.inner, reserved_bytes);
+        }
+    }
 }
 
 // ColumnFamilyDescriptor is a pair of column family's name and options.

tests/test_rocksdb_options.rs

@@ -668,3 +668,34 @@ fn test_readoptions_max_bytes_for_level_multiplier() {
     cf_opts.set_max_bytes_for_level_multiplier(8);
     assert_eq!(cf_opts.get_max_bytes_for_level_multiplier(), 8);
 }
+
+#[test]
+fn test_vector_memtable_factory_options() {
+    let path = TempDir::new("_rust_rocksdb_vector_memtable_factory_options").unwrap();
+    let path_str = path.path().to_str().unwrap();
+
+    let mut opts = DBOptions::new();
+    opts.create_if_missing(true);
+    opts.allow_concurrent_memtable_write(false);
+
+    let mut cf_opts = ColumnFamilyOptions::new();
+    cf_opts.set_vector_memtable_factory(4096);
+
+    let db = DB::open_cf(opts.clone(), path_str, vec![("default", cf_opts)]).unwrap();
+    db.put(b"k1", b"v1").unwrap();
+    db.put(b"k2", b"v2").unwrap();
+    assert_eq!(db.get(b"k1").unwrap().unwrap(), b"v1");
+    assert_eq!(db.get(b"k2").unwrap().unwrap(), b"v2");
+    db.flush(true).unwrap();
+
+    let mut iter = db.iter();
+    iter.seek(SeekKey::Start);
+    assert!(iter.valid());
+    assert_eq!(iter.key(), b"k1");
+    assert_eq!(iter.value(), b"v1");
+    assert!(iter.next());
+    assert_eq!(iter.key(), b"k2");
+    assert_eq!(iter.value(), b"v2");
+    assert!(!iter.next());
+    assert!(!iter.valid());
+}