Add low pri pool ratio of block cache (#785)

* add low pri pool ratio Signed-off-by: Connor1996 <zbk602423539@gmail.com> --------- Signed-off-by: Connor1996 <zbk602423539@gmail.com>
tikv · Mar 12, 2024 · fd7ed25 · fd7ed25
1 parent 05081c2
commit fd7ed25
Show file tree

Hide file tree

Showing 4 changed files with 21 additions and 0 deletions.
diff --git a/librocksdb_sys/crocksdb/c.cc b/librocksdb_sys/crocksdb/c.cc
@@ -4283,6 +4283,11 @@ void crocksdb_lru_cache_options_set_high_pri_pool_ratio(
   opt->rep.high_pri_pool_ratio = high_pri_pool_ratio;
 }
 
+void crocksdb_lru_cache_options_set_low_pri_pool_ratio(
+    crocksdb_lru_cache_options_t* opt, double low_pri_pool_ratio) {
+  opt->rep.low_pri_pool_ratio = low_pri_pool_ratio;
+}
+
 void crocksdb_lru_cache_options_set_memory_allocator(
     crocksdb_lru_cache_options_t* opt, crocksdb_memory_allocator_t* allocator) {
   opt->rep.memory_allocator = allocator->rep;

diff --git a/librocksdb_sys/crocksdb/crocksdb/c.h b/librocksdb_sys/crocksdb/crocksdb/c.h
@@ -1756,6 +1756,9 @@ extern C_ROCKSDB_LIBRARY_API void
 crocksdb_lru_cache_options_set_high_pri_pool_ratio(
     crocksdb_lru_cache_options_t*, double);
 extern C_ROCKSDB_LIBRARY_API void
+crocksdb_lru_cache_options_set_low_pri_pool_ratio(crocksdb_lru_cache_options_t*,
+                                                  double);
+extern C_ROCKSDB_LIBRARY_API void
 crocksdb_lru_cache_options_set_memory_allocator(crocksdb_lru_cache_options_t*,
                                                 crocksdb_memory_allocator_t*);
 extern C_ROCKSDB_LIBRARY_API crocksdb_cache_t* crocksdb_cache_create_lru(

diff --git a/librocksdb_sys/src/lib.rs b/librocksdb_sys/src/lib.rs
@@ -647,6 +647,10 @@ extern "C" {
         opt: *mut DBLRUCacheOptions,
         high_pri_pool_ratio: c_double,
     );
+    pub fn crocksdb_lru_cache_options_set_low_pri_pool_ratio(
+        opt: *mut DBLRUCacheOptions,
+        low_pri_pool_ratio: c_double,
+    );
     pub fn crocksdb_lru_cache_options_set_memory_allocator(
         opt: *mut DBLRUCacheOptions,
         allocator: *mut DBMemoryAllocator,

diff --git a/src/rocksdb_options.rs b/src/rocksdb_options.rs
@@ -2480,6 +2480,15 @@ impl LRUCacheOptions {
         }
     }
 
+    pub fn set_low_pri_pool_ratio(&mut self, low_pri_pool_ratio: c_double) {
+        unsafe {
+            crocksdb_ffi::crocksdb_lru_cache_options_set_low_pri_pool_ratio(
+                self.inner,
+                low_pri_pool_ratio,
+            );
+        }
+    }
+
     pub fn set_memory_allocator(&mut self, allocator: MemoryAllocator) {
         unsafe {
             crocksdb_ffi::crocksdb_lru_cache_options_set_memory_allocator(