commit d66b39bc5d24a3fef79d71dfc296019ce50bd036
author Murtadha Hubail <mhubail@apache.org> Tue Sep 03 17:04:57 2024 +0300
committer Murtadha Hubail <mhubail@apache.org> Tue Sep 03 17:04:57 2024 +0300
tree a8cbc84225f45a67f205c48ed2b3f18bbf75c54e
parent 09a1024be5026a3d246bca348a01b57d0beb8ca4
parent 3f351864ca31c5e9a7e5ae73cf4481aff6087b06

Merge branch 'gerrit/trinity' into 'gerrit/goldfish'

Ext-ref: MB-63306
Change-Id: I399db805a33cd124c8b433b0bf4e8665a1b3d19b

asterixdb/asterix-common/src/main/java/org/apache/asterix/common/context/DatasetInfo.java

diff --git a/asterixdb/asterix-common/src/main/java/org/apache/asterix/common/context/DatasetInfo.java b/asterixdb/asterix-common/src/main/java/org/apache/asterix/common/context/DatasetInfo.java
index c8db4cd..952185b 100644
--- a/asterixdb/asterix-common/src/main/java/org/apache/asterix/common/context/DatasetInfo.java
+++ b/asterixdb/asterix-common/src/main/java/org/apache/asterix/common/context/DatasetInfo.java
@@ -292,6 +292,20 @@
         }
     }
 
+    public void waitForFlushes() throws HyracksDataException {
+        logManager.log(waitLog);
+        synchronized (this) {
+            while (pendingFlushes > 0) {
+                try {
+                    wait();
+                } catch (InterruptedException e) {
+                    Thread.currentThread().interrupt();
+                    throw HyracksDataException.create(e);
+                }
+            }
+        }
+    }
+
     public synchronized int getPendingFlushes() {
         return pendingFlushes;
     }