Add RocksDB.flushWal(FlushWALOptions) overload

Author: mike-evolved

java/rocksjni/rocksjni.cc

@@ -3112,15 +3112,33 @@ void Java_org_rocksdb_RocksDB_flush(JNIEnv* env, jclass, jlong jdb_handle,
  * Method:    flushWal
  * Signature: (JZ)V
  */
-void Java_org_rocksdb_RocksDB_flushWal(JNIEnv* env, jclass, jlong jdb_handle,
-                                       jboolean jsync) {
+void Java_org_rocksdb_RocksDB_flushWal__JZ(JNIEnv* env, jclass,
+                                           jlong jdb_handle, jboolean jsync) {
   auto* db = reinterpret_cast<ROCKSDB_NAMESPACE::DB*>(jdb_handle);
   auto s = db->FlushWAL(jsync == JNI_TRUE);
   if (!s.ok()) {
     ROCKSDB_NAMESPACE::RocksDBExceptionJni::ThrowNew(env, s);
   }
 }
 
+/*
+ * Class:     org_rocksdb_RocksDB
+ * Method:    flushWal
+ * Signature: (JJ)V
+ */
+void Java_org_rocksdb_RocksDB_flushWal__JJ(JNIEnv* env, jclass,
+                                           jlong jdb_handle,
+                                           jlong jflush_wal_options_handle) {
+  auto* db = reinterpret_cast<ROCKSDB_NAMESPACE::DB*>(jdb_handle);
+  auto* flush_wal_options =
+      reinterpret_cast<ROCKSDB_NAMESPACE::FlushWALOptions*>(
+          jflush_wal_options_handle);
+  auto s = db->FlushWAL(*flush_wal_options);
+  if (!s.ok()) {
+    ROCKSDB_NAMESPACE::RocksDBExceptionJni::ThrowNew(env, s);
+  }
+}
+
 /*
  * Class:     org_rocksdb_RocksDB
  * Method:    syncWal

java/src/main/java/org/rocksdb/RocksDB.java

@@ -4272,6 +4272,17 @@ public void flushWal(final boolean sync) throws RocksDBException {
     flushWal(nativeHandle_, sync);
   }
 
+  /**
+   * Flush the WAL buffer to disk with advanced control.
+   * Use this for fine-grained control over sync behavior and I/O rate limiting.
+   *
+   * @param flushWALOptions controls sync and rate limiter priority
+   * @throws RocksDBException if flush fails
+   */
+  public void flushWal(final FlushWALOptions flushWALOptions) throws RocksDBException {
+    flushWal(nativeHandle_, flushWALOptions.nativeHandle_);
+  }
+
   /**
    * Sync the WAL.
    * <p>
@@ -5048,6 +5059,8 @@ private static native void flush(final long handle, final long flushOptHandle,
       /* @Nullable */ final long[] cfHandles) throws RocksDBException;
   private static native void flushWal(final long handle, final boolean sync)
       throws RocksDBException;
+  private static native void flushWal(final long handle, final long flushWALOptionsHandle)
+      throws RocksDBException;
   private static native void syncWal(final long handle) throws RocksDBException;
   private static native long getLatestSequenceNumber(final long handle);
   private static native void disableFileDeletions(long handle) throws RocksDBException;