Add RocksDB.waitForCompact() method

Author: mike-evolved

java/rocksjni/rocksjni.cc

@@ -3134,6 +3134,24 @@ void Java_org_rocksdb_RocksDB_syncWal(JNIEnv* env, jclass, jlong jdb_handle) {
   }
 }
 
+/*
+ * Class:     org_rocksdb_RocksDB
+ * Method:    waitForCompact
+ * Signature: (JJ)V
+ */
+void Java_org_rocksdb_RocksDB_waitForCompact(
+    JNIEnv* env, jclass, jlong jdb_handle,
+    jlong jwait_for_compact_options_handle) {
+  auto* db = reinterpret_cast<ROCKSDB_NAMESPACE::DB*>(jdb_handle);
+  auto* wait_for_compact_options =
+      reinterpret_cast<ROCKSDB_NAMESPACE::WaitForCompactOptions*>(
+          jwait_for_compact_options_handle);
+  auto s = db->WaitForCompact(*wait_for_compact_options);
+  if (!s.ok()) {
+    ROCKSDB_NAMESPACE::RocksDBExceptionJni::ThrowNew(env, s);
+  }
+}
+
 /*
  * Class:     org_rocksdb_RocksDB
  * Method:    getLatestSequenceNumber

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

@@ -3839,6 +3839,18 @@ public void compactRange(
         columnFamilyHandle == null ? 0 : columnFamilyHandle.nativeHandle_);
   }
 
+  /**
+   * Wait for all pending flush and compaction jobs to complete.
+   * Warning: May block indefinitely if writes continue. Use timeout to prevent this.
+   *
+   * @param waitForCompactOptions controls timeout, flush inclusion, and abort behavior
+   * @throws RocksDBException if wait fails or DB is shutting down
+   */
+  public void waitForCompact(final WaitForCompactOptions waitForCompactOptions)
+      throws RocksDBException {
+    waitForCompact(nativeHandle_, waitForCompactOptions.nativeHandle_);
+  }
+
   /**
    * ClipColumnFamily() will clip the entries in the CF according to the range
    * [begin_key, end_key). Returns OK on success, and a non-OK status on error.
@@ -5049,6 +5061,16 @@ private static native void flush(final long handle, final long flushOptHandle,
   private static native void flushWal(final long handle, final boolean sync)
       throws RocksDBException;
   private static native void syncWal(final long handle) throws RocksDBException;
+  private static native void waitForCompact(
+      final long handle, final long waitForCompactOptionsHandle) throws RocksDBException;
+  private static native String getDbSessionId(final long handle) throws RocksDBException;
+  private static native void lockWAL(final long handle) throws RocksDBException;
+  private static native void unlockWAL(final long handle) throws RocksDBException;
+  private static native void increaseFullHistoryTsLow(
+      final long handle, final long columnFamilyHandle, final byte[] tsLow)
+      throws RocksDBException;
+  private static native byte[] getFullHistoryTsLow(
+      final long handle, final long columnFamilyHandle) throws RocksDBException;
   private static native long getLatestSequenceNumber(final long handle);
   private static native void disableFileDeletions(long handle) throws RocksDBException;
   private static native void enableFileDeletions(long handle) throws RocksDBException;