Use Phaser in onRegionError

apache · Sep 2, 2024 · 5573a60 · 5573a60
1 parent 82cef8b
commit 5573a60
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/hbase-client/src/main/java/org/apache/hadoop/hbase/client/AsyncTableImpl.java b/hbase-client/src/main/java/org/apache/hadoop/hbase/client/AsyncTableImpl.java
@@ -316,7 +316,14 @@ public void onRegionComplete(RegionInfo region, R resp) {
 
       @Override
       public void onRegionError(RegionInfo region, Throwable error) {
-        pool.execute(context.wrap(() -> callback.onRegionError(region, error)));
+        regionCompletesInProgress.register();
+        pool.execute(context.wrap(() -> {
+          try {
+            callback.onRegionError(region, error);
+          } finally {
+            regionCompletesInProgress.arriveAndDeregister();
+          }
+        }));
       }
 
       @Override