[WIP] Expose WAL compression option

Signed-off-by: Bisheng Huang <[email protected]>

.gitmodules

@@ -1,9 +1,9 @@
 [submodule "rocksdb"]
 	path = librocksdb_sys/rocksdb
-	url = https://github.com/v01dstar/rocksdb.git
-	branch = 8.10-tikv
+	url = https://github.com/hbisheng/rocksdb.git
+	branch = fix-pri
 
 [submodule "titan"]
 	path = librocksdb_sys/libtitan_sys/titan
-	url = https://github.com/v01dstar/titan.git
-	branch = upgrade-rocksdb
+	url = https://github.com/hbisheng/titan.git
+	branch = fix-pri

librocksdb_sys/crocksdb/c.cc

@@ -5070,6 +5070,14 @@ void crocksdb_options_set_min_level_to_compress(crocksdb_options_t* opt,
   }
 }
 
+uint32_t crocksdb_options_get_wal_compression(crocksdb_options_t* opt) {
+  return static_cast<uint32_t>(opt->rep.wal_compression);
+}
+
+void crocksdb_options_set_wal_compression(crocksdb_options_t* opt, int compression_type) {
+  opt->rep.wal_compression = static_cast<CompressionType>(compression_type);
+}
+
 size_t crocksdb_livefiles_count(const crocksdb_livefiles_t* lf) {
   return static_cast<int>(lf->rep.size());
 }

librocksdb_sys/crocksdb/crocksdb/c.h

@@ -1345,6 +1345,11 @@ extern C_ROCKSDB_LIBRARY_API void crocksdb_options_set_plain_table_factory(
 extern C_ROCKSDB_LIBRARY_API void crocksdb_options_set_min_level_to_compress(
     crocksdb_options_t* opt, int level);
 
+extern C_ROCKSDB_LIBRARY_API uint32_t crocksdb_options_get_wal_compression(
+    crocksdb_options_t* opt);
+extern C_ROCKSDB_LIBRARY_API void crocksdb_options_set_wal_compression(
+    crocksdb_options_t* opt, int compression_type);
+
 extern C_ROCKSDB_LIBRARY_API void crocksdb_options_set_memtable_huge_page_size(
     crocksdb_options_t*, size_t);
 

librocksdb_sys/src/lib.rs

@@ -847,6 +847,11 @@ extern "C" {
         zstd_max_train_bytes: c_int,
         parallel_threads: c_int,
     );
+    pub fn crocksdb_options_set_wal_compression(
+        options: *mut Options,
+        compression_style_no: DBCompressionType,
+    );
+    pub fn crocksdb_options_get_wal_compression(options: *mut Options) -> DBCompressionType;
     pub fn crocksdb_options_set_bottommost_compression_options(
         options: *mut Options,
         window_bits: c_int,

src/rocksdb_options.rs

@@ -1117,6 +1117,14 @@ impl DBOptions {
         }
     }
 
+    pub fn set_wal_compression(&self, c: DBCompressionType) {
+        unsafe { crocksdb_ffi::crocksdb_options_set_wal_compression(self.inner, c) }
+    }
+
+    pub fn get_wal_compression(&self) -> DBCompressionType {
+        unsafe { crocksdb_ffi::crocksdb_options_get_wal_compression(self.inner) }
+    }
+
     pub fn set_delayed_write_rate(&mut self, rate: u64) {
         unsafe {
             crocksdb_ffi::crocksdb_options_set_delayed_write_rate(self.inner, rate);