Add option set_compaction_pri

Author: zaidoon1

src/db_options.rs

@@ -3529,6 +3529,28 @@ impl Options {
             ffi::rocksdb_options_set_avoid_unnecessary_blocking_io(self.inner, u8::from(val));
         }
     }
+
+    /// Sets the compaction priority.
+    ///
+    /// If level compaction_style =
+    /// kCompactionStyleLevel, for each level, which files are prioritized to be
+    /// picked to compact.
+    ///
+    /// Default: `DBCompactionPri::MinOverlappingRatio`
+    ///
+    /// # Examples
+    ///
+    /// ```
+    /// use rust_rocksdb::{Options, DBCompactionPri};
+    ///
+    /// let mut opts = Options::default();
+    /// opts.set_compaction_pri(DBCompactionPri::RoundRobinCompaction);
+    /// ```
+    pub fn set_compaction_pri(&mut self, pri: DBCompactionPri) {
+        unsafe {
+            ffi::rocksdb_options_set_compaction_pri(self.inner, pri as c_int);
+        }
+    }
 }
 
 impl Default for Options {
@@ -4166,6 +4188,16 @@ pub enum RateLimiterMode {
     KAllIo = 2,
 }
 
+#[derive(Debug, Copy, Clone, PartialEq, Eq)]
+#[cfg_attr(feature = "serde1", derive(serde::Serialize, serde::Deserialize))]
+pub enum DBCompactionPri {
+    ByCompensatedSize = ffi::rocksdb_k_by_compensated_size_compaction_pri as isize,
+    OldestLargestSeqFirst = ffi::rocksdb_k_oldest_largest_seq_first_compaction_pri as isize,
+    OldestSmallestSeqFirst = ffi::rocksdb_k_oldest_smallest_seq_first_compaction_pri as isize,
+    MinOverlappingRatio = ffi::rocksdb_k_min_overlapping_ratio_compaction_pri as isize,
+    RoundRobinCompaction = ffi::rocksdb_k_round_robin_compaction_pri as isize,
+}
+
 pub struct FifoCompactOptions {
     pub(crate) inner: *mut ffi::rocksdb_fifo_compaction_options_t,
 }

src/lib.rs

@@ -115,8 +115,8 @@ pub use crate::{
     },
     db_options::{
         BlockBasedIndexType, BlockBasedOptions, BottommostLevelCompaction, Cache, ChecksumType,
-        CompactOptions, CuckooTableOptions, DBCompactionStyle, DBCompressionType, DBPath,
-        DBRecoveryMode, DataBlockIndexType, FifoCompactOptions, FlushOptions,
+        CompactOptions, CuckooTableOptions, DBCompactionPri, DBCompactionStyle, DBCompressionType,
+        DBPath, DBRecoveryMode, DataBlockIndexType, FifoCompactOptions, FlushOptions,
         IngestExternalFileOptions, KeyEncodingType, LogLevel, MemtableFactory, Options,
         PlainTableFactoryOptions, RateLimiterMode, ReadOptions, ReadTier, UniversalCompactOptions,
         UniversalCompactionStopStyle, WaitForCompactOptions, WriteBufferManager, WriteOptions,

tests/test_rocksdb_options.rs

@@ -17,7 +17,8 @@ mod util;
 use std::{fs, io::Read as _};
 
 use rust_rocksdb::{
-    BlockBasedOptions, Cache, DBCompressionType, DataBlockIndexType, Env, Options, ReadOptions, DB,
+    BlockBasedOptions, Cache, DBCompactionPri, DBCompressionType, DataBlockIndexType, Env, Options,
+    ReadOptions, DB,
 };
 use util::DBPath;
 
@@ -351,3 +352,14 @@ fn test_set_ratelimiter() {
         assert_eq!(&*db.get(b"k2").unwrap().unwrap(), b"a");
     }
 }
+
+#[test]
+fn test_set_compaction_pri() {
+    let path = DBPath::new("_set_compaction_pri");
+    {
+        let mut opts = Options::default();
+        opts.create_if_missing(true);
+        opts.set_compaction_pri(DBCompactionPri::RoundRobinCompaction);
+        let _db = DB::open(&opts, &path).unwrap();
+    }
+}