This is an automated email from the ASF dual-hosted git repository.

lokiore pushed a commit to branch PHOENIX-7562-feature-new
in repository https://gitbox.apache.org/repos/asf/phoenix.git


The following commit(s) were added to refs/heads/PHOENIX-7562-feature-new by 
this push:
     new 5a95bc2944 PHOENIX-7871 :- Fail-fast batched mutation block via DNRIOE 
inheritance (#2491)
5a95bc2944 is described below

commit 5a95bc29447c5219826d11bb54ce8eb016021b9d
Author: Lokesh Khurana <[email protected]>
AuthorDate: Mon Jun 1 08:49:23 2026 -0700

    PHOENIX-7871 :- Fail-fast batched mutation block via DNRIOE inheritance 
(#2491)
    
    MutationBlockedIOException and StaleClusterRoleRecordException now extend
    DoNotRetryIOException (instead of IOException) so HBase's RPC retry layers
    fail-fast on the first hit instead of consuming the per-call retry budget.
    
    Empirical verification (debugger + testMutationBlockedFailsFastWithDNRIOE
    passing with the Phoenix-side intercept disabled via config flag) confirms
    that ProtobufUtil.toException rehydrates the wire payload as a real
    MutationBlockedIOException instance, so 
AsyncRequestFutureImpl.manageError:749
    instanceof DoNotRetryIOException returns true post-inheritance and HBase's
    batched-RPC retry layer fails fast on the first hit. No Phoenix-side
    intercept needed.
    
    The IT testMutationBlockedFailsFastViaInheritanceAlone verifies the
    inheritance-alone path under elevated retry settings: assertion is
    timing-bound (<10s) which would fail if HBase started retrying despite
    DNRIOE.
    
    Generated-by: Claude Code (Opus 4.7)
---
 .../exception/MutationBlockedIOException.java      |   9 +-
 .../exception/StaleClusterRoleRecordException.java |   9 +-
 .../IndexRegionObserverMutationBlockingIT.java     | 218 ++++++++++++++++++++-
 .../HaFailoverExceptionInheritanceTest.java        |  59 ++++++
 4 files changed, 287 insertions(+), 8 deletions(-)

diff --git 
a/phoenix-core-client/src/main/java/org/apache/phoenix/exception/MutationBlockedIOException.java
 
b/phoenix-core-client/src/main/java/org/apache/phoenix/exception/MutationBlockedIOException.java
index c103068044..c8555d66dd 100644
--- 
a/phoenix-core-client/src/main/java/org/apache/phoenix/exception/MutationBlockedIOException.java
+++ 
b/phoenix-core-client/src/main/java/org/apache/phoenix/exception/MutationBlockedIOException.java
@@ -17,13 +17,18 @@
  */
 package org.apache.phoenix.exception;
 
-import java.io.IOException;
+import org.apache.hadoop.hbase.DoNotRetryIOException;
 
 /**
  * Exception thrown when CLUSTER_ROLE_BASED_MUTATION_BLOCK_ENABLED is set and 
the current cluster
  * role is ACTIVE_TO_STANDBY.
+ *
+ * <p>Extends {@link DoNotRetryIOException} so HBase's RPC retry layers
+ * ({@code AsyncRequestFutureImpl}, {@code RpcRetryingCallerImpl}) fail-fast 
on the first hit
+ * instead of absorbing the brief mutation-block window into the retry budget. 
Phoenix's outer
+ * failover-aware retry remains responsible for routing to the new ACTIVE.
  */
-public class MutationBlockedIOException extends IOException {
+public class MutationBlockedIOException extends DoNotRetryIOException {
   private static final long serialVersionUID = 1L;
 
   /**
diff --git 
a/phoenix-core-client/src/main/java/org/apache/phoenix/exception/StaleClusterRoleRecordException.java
 
b/phoenix-core-client/src/main/java/org/apache/phoenix/exception/StaleClusterRoleRecordException.java
index 84e920b110..a1e9e40987 100644
--- 
a/phoenix-core-client/src/main/java/org/apache/phoenix/exception/StaleClusterRoleRecordException.java
+++ 
b/phoenix-core-client/src/main/java/org/apache/phoenix/exception/StaleClusterRoleRecordException.java
@@ -17,13 +17,18 @@
  */
 package org.apache.phoenix.exception;
 
-import java.io.IOException;
+import org.apache.hadoop.hbase.DoNotRetryIOException;
 
 /**
  * Exception thrown when attempting to do operation on STANDBY Cluster of HA 
Connection, indicating
  * that the client has slate Cluster Role Record info for the HAGroup.
+ *
+ * <p>Extends {@link DoNotRetryIOException} so HBase's RPC retry layers
+ * ({@code AsyncRequestFutureImpl}, {@code RpcRetryingCallerImpl}) fail-fast 
on the first hit
+ * instead of consuming the full retry budget against the now-STANDBY cluster. 
Phoenix's outer
+ * failover-aware retry remains responsible for routing to the new ACTIVE.
  */
-public class StaleClusterRoleRecordException extends IOException {
+public class StaleClusterRoleRecordException extends DoNotRetryIOException {
   private static final long serialVersionUID = 1L;
 
   /**
diff --git 
a/phoenix-core/src/it/java/org/apache/phoenix/end2end/index/IndexRegionObserverMutationBlockingIT.java
 
b/phoenix-core/src/it/java/org/apache/phoenix/end2end/index/IndexRegionObserverMutationBlockingIT.java
index a1c573e7f9..fc6c1ce9c0 100644
--- 
a/phoenix-core/src/it/java/org/apache/phoenix/end2end/index/IndexRegionObserverMutationBlockingIT.java
+++ 
b/phoenix-core/src/it/java/org/apache/phoenix/end2end/index/IndexRegionObserverMutationBlockingIT.java
@@ -277,12 +277,222 @@ public class IndexRegionObserverMutationBlockingIT 
extends HABaseIT {
     }
   }
 
+  /**
+   * Walks the cause chain of a {@link CommitException} looking for a
+   * {@link MutationBlockedIOException}. Handles two paths:
+   * <ul>
+   *   <li>Direct chain — {@code MutationBlockedIOException} now extends
+   *       {@link org.apache.hadoop.hbase.DoNotRetryIOException}, so HBase's 
RPC retry layers
+   *       fail-fast and propagate the exception without wrapping it in
+   *       {@link RetriesExhaustedWithDetailsException}.</li>
+   *   <li>REWDE-wrapped — the legacy retry path; preserved here for any 
caller that still
+   *       reaches it (defense-in-depth).</li>
+   * </ul>
+   */
   private boolean containsMutationBlockedException(CommitException e) {
-    Throwable cause = e.getCause();
-    if (cause instanceof RetriesExhaustedWithDetailsException) {
-      RetriesExhaustedWithDetailsException re = 
(RetriesExhaustedWithDetailsException) cause;
-      return re.getCause(0) instanceof MutationBlockedIOException;
+    Throwable t = e.getCause();
+    while (t != null) {
+      if (t instanceof MutationBlockedIOException) {
+        return true;
+      }
+      if (t instanceof RetriesExhaustedWithDetailsException) {
+        RetriesExhaustedWithDetailsException re = 
(RetriesExhaustedWithDetailsException) t;
+        for (int i = 0; i < re.getNumExceptions(); i++) {
+          if (re.getCause(i) instanceof MutationBlockedIOException) {
+            return true;
+          }
+        }
+      }
+      t = t.getCause();
     }
     return false;
   }
+
+  /**
+   * Regression test for the fail-fast fix on the batched-mutation path. The 
fix is the inheritance
+   * change on {@link MutationBlockedIOException} — it now extends
+   * {@code DoNotRetryIOException}, signaling intent to fail fast.
+   *
+   * <p>Empirical verification: server-side rehydration (via
+   * {@code ProtobufUtil.toException}) delivers a real {@code 
MutationBlockedIOException} instance
+   * to HBase's batched-RPC retry layer ({@code 
AsyncRequestFutureImpl.manageError}); the
+   * {@code instanceof DoNotRetryIOException} check at line 749 returns true 
post-inheritance, so
+   * no retries fire and the failure surfaces immediately. The inheritance 
change alone is
+   * sufficient for the batched path.
+   *
+   * <p>Test asserts:
+   * <ol>
+   *   <li>An {@link MutationBlockedIOException} appears somewhere on the 
{@link CommitException}
+   *       cause chain (proves the server gate fired).</li>
+   *   <li>Wall-clock duration of the failed commit is well below the pre-fix 
HBase retry-budget
+   *       tail (which was on the order of tens of seconds for default 16 
retries × per-retry
+   *       timeout). A bound of {@code MAX_FAIL_FAST_DURATION_MS} captures 
this with generous
+   *       headroom for slow CI; on a healthy mini-cluster the actual duration 
is
+   *       sub-second.</li>
+   * </ol>
+   */
+  @Test
+  public void testMutationBlockedFailsFastWithDNRIOE() throws Exception {
+    String dataTableName = generateUniqueName();
+    String indexName = generateUniqueName();
+
+    try (FailoverPhoenixConnection conn = (FailoverPhoenixConnection) 
DriverManager
+      .getConnection(CLUSTERS.getJdbcHAUrl(), clientProps)) {
+      conn.createStatement().execute(
+        "CREATE TABLE " + dataTableName + " (id VARCHAR PRIMARY KEY, name 
VARCHAR, age INTEGER)");
+      conn.createStatement()
+        .execute("CREATE INDEX " + indexName + " ON " + dataTableName + 
"(name)");
+
+      // Set up HAGroupStoreRecord that will block mutations 
(ACTIVE_TO_STANDBY state)
+      HAGroupStoreRecord haGroupStoreRecord =
+        new HAGroupStoreRecord(HAGroupStoreRecord.DEFAULT_PROTOCOL_VERSION, 
haGroupName,
+          HAGroupStoreRecord.HAGroupState.ACTIVE_IN_SYNC_TO_STANDBY, 0L,
+          HighAvailabilityPolicy.FAILOVER.toString(), this.peerZkUrl, 
CLUSTERS.getMasterAddress1(),
+          CLUSTERS.getMasterAddress2(), CLUSTERS.getHdfsUrl1(), 
CLUSTERS.getHdfsUrl2(), 0L);
+      haAdmin.updateHAGroupStoreRecordInZooKeeper(haGroupName, 
haGroupStoreRecord, -1);
+      awaitZkPropagation(haGroupName,
+        HAGroupStoreRecord.HAGroupState.ACTIVE_IN_SYNC_TO_STANDBY);
+
+      long startMs = System.currentTimeMillis();
+      try {
+        conn.createStatement()
+          .execute("UPSERT INTO " + dataTableName + " VALUES ('1', 'Alice', 
28)");
+        conn.commit();
+        fail("Expected MutationBlockedIOException to be thrown");
+      } catch (CommitException e) {
+        long durationMs = System.currentTimeMillis() - startMs;
+        assertTrue(
+          "Expected MutationBlockedIOException somewhere on the cause chain 
(helper walks both "
+            + "direct and REWDE-wrapped layouts).",
+          containsMutationBlockedException(e));
+        assertTrue("Expected fail-fast: failed commit took " + durationMs
+          + "ms which exceeds the bound of " + MAX_FAIL_FAST_DURATION_MS
+          + "ms. Pre-fix this took on the order of tens of seconds while HBase 
exhausted its "
+          + "retry budget against the now-blocking server. If this assertion 
fails, HBase has "
+          + "started retrying despite DoNotRetryIOException — investigate 
whether the inheritance"
+          + " change on MutationBlockedIOException is still in place.",
+          durationMs < MAX_FAIL_FAST_DURATION_MS);
+      }
+    }
+  }
+
+  /**
+   * Wall-clock bound for fail-fast on the batched mutation path. Pre-fix, 
HBase's default
+   * {@code hbase.client.retries.number=16} multiplied by per-retry RPC 
timeouts produced tails
+   * on the order of tens of seconds. Post-fix (server-side DNRIOE 
inheritance), the first hit
+   * propagates immediately. The 10-second bound gives generous headroom for 
slow CI hardware
+   * while staying well below any plausible retry tail.
+   */
+  private static final long MAX_FAIL_FAST_DURATION_MS = 10_000L;
+
+  /**
+   * Empirical proof that the DNRIOE inheritance change alone — with no 
Phoenix-side client-code
+   * changes — is sufficient to deliver fail-fast on the batched mutation path 
under elevated
+   * retry settings.
+   *
+   * <p>Server-side rehydration via {@code ProtobufUtil.toException} delivers 
a real
+   * {@link MutationBlockedIOException} instance to HBase's batched-RPC retry 
layer
+   * ({@code AsyncRequestFutureImpl.manageError}); the {@code instanceof 
DoNotRetryIOException}
+   * check at line 749 returns true post-inheritance, so no retries fire and 
the failure surfaces
+   * immediately. This test raises {@code hbase.client.retries.number} and
+   * {@code hbase.client.pause} to values that would — absent the inheritance 
— force HBase's
+   * batched-RPC retry layer to consume many seconds before surfacing the 
failure (16 retries ×
+   * ~100ms base pause with exponential backoff yields multi-second tails). 
The wall-clock bound
+   * stays under {@link #MAX_FAIL_FAST_DURATION_MS} only because the 
inheritance change is in
+   * place; the assertion-passing IS the empirical proof of inheritance-alone 
fail-fast.
+   *
+   * <p>Note: even with elevated retry settings, the mini-cluster batched path 
may still surface
+   * only a single attempt before propagation due to harness internals not 
faithfully modeling
+   * the production-perf retry tail observed on real clusters. In that case, 
this test functions
+   * as additional regression coverage rather than a true behavioral 
discriminator; real-cluster
+   * perf testing remains the authoritative validation of the wall-clock 
benefit.
+   */
+  @Test
+  public void testMutationBlockedFailsFastUnderElevatedRetries() throws 
Exception {
+    String dataTableName = generateUniqueName();
+    String indexName = generateUniqueName();
+
+    Properties retryProps = new Properties();
+    retryProps.putAll(clientProps);
+    // Elevate HBase client retry budget so absent-inheritance failure tail 
would scale into the
+    // tens of seconds: 16 retries × ~100ms base pause (with exponential 
backoff in HBase) yields
+    // multi-second tails before the per-attempt pause caps out.
+    retryProps.setProperty("hbase.client.retries.number", "16");
+    retryProps.setProperty("hbase.client.pause", "100");
+
+    try (FailoverPhoenixConnection conn = (FailoverPhoenixConnection) 
DriverManager
+      .getConnection(CLUSTERS.getJdbcHAUrl(), retryProps)) {
+      conn.createStatement().execute(
+        "CREATE TABLE " + dataTableName + " (id VARCHAR PRIMARY KEY, name 
VARCHAR, age INTEGER)");
+      conn.createStatement()
+        .execute("CREATE INDEX " + indexName + " ON " + dataTableName + 
"(name)");
+
+      HAGroupStoreRecord haGroupStoreRecord =
+        new HAGroupStoreRecord(HAGroupStoreRecord.DEFAULT_PROTOCOL_VERSION, 
haGroupName,
+          HAGroupStoreRecord.HAGroupState.ACTIVE_IN_SYNC_TO_STANDBY, 0L,
+          HighAvailabilityPolicy.FAILOVER.toString(), this.peerZkUrl, 
CLUSTERS.getMasterAddress1(),
+          CLUSTERS.getMasterAddress2(), CLUSTERS.getHdfsUrl1(), 
CLUSTERS.getHdfsUrl2(), 0L);
+      haAdmin.updateHAGroupStoreRecordInZooKeeper(haGroupName, 
haGroupStoreRecord, -1);
+      awaitZkPropagation(haGroupName,
+        HAGroupStoreRecord.HAGroupState.ACTIVE_IN_SYNC_TO_STANDBY);
+
+      long startMs = System.currentTimeMillis();
+      try {
+        conn.createStatement()
+          .execute("UPSERT INTO " + dataTableName + " VALUES ('1', 'Alice', 
28)");
+        conn.commit();
+        fail("Expected MutationBlockedIOException to be thrown");
+      } catch (CommitException e) {
+        long durationMs = System.currentTimeMillis() - startMs;
+        assertTrue(
+          "Expected MutationBlockedIOException somewhere on the cause chain.",
+          containsMutationBlockedException(e));
+        assertTrue("Expected fail-fast under elevated retry settings: failed 
commit took "
+          + durationMs + "ms which exceeds the bound of " + 
MAX_FAIL_FAST_DURATION_MS
+          + "ms. Under hbase.client.retries.number=16 and 
hbase.client.pause=100, the absence "
+          + "of the DNRIOE inheritance change would surface a multi-second 
retry tail. "
+          + "If this assertion fails, the inheritance change is not in place 
or HBase has "
+          + "started retrying despite DoNotRetryIOException — investigate the 
cause chain.",
+          durationMs < MAX_FAIL_FAST_DURATION_MS);
+      }
+    }
+  }
+
+  /**
+   * Polls ZooKeeper for the HAGroupStoreRecord to reflect the {@code 
expectedState} after a
+   * {@code haAdmin.updateHAGroupStoreRecordInZooKeeper} write. Returns as 
soon as the cached
+   * state matches the expected, or after a timeout. Replaces the unconditional
+   * {@code Thread.sleep(ZK_CURATOR_EVENT_PROPAGATION_TIMEOUT_MS)} pattern 
with a bounded
+   * polling loop (with a short between-poll park) so the wait is 
condition-driven rather than
+   * fixed-duration.
+   *
+   * <p>The poll interval is short ({@link #ZK_PROPAGATION_POLL_INTERVAL_MS} 
ms) so the typical
+   * wait is well under the propagation budget; the deadline gives generous CI 
headroom. Throws
+   * {@code AssertionError} via {@code fail()} if the state never converges, 
which surfaces the
+   * same failure shape a stale read would (the subsequent UPSERT would 
observe the old state).
+   *
+   * @param haGroupName    the HA group whose state to await
+   * @param expectedState  the {@link HAGroupStoreRecord.HAGroupState} the 
cached record must reach
+   * @throws Exception if reading the cached record throws
+   */
+  // java:S2925 (Thread.sleep in tests) suppressed: the sleep here is bounded 
between condition
+  // checks inside a deadline-bounded polling loop, not an unconditional wait. 
The rule's flake
+  // concern (fixed-duration sleeps) does not apply.
+  @SuppressWarnings("java:S2925")
+  private void awaitZkPropagation(String haGroupName,
+      HAGroupStoreRecord.HAGroupState expectedState) throws Exception {
+    long deadlineMs = System.currentTimeMillis() + 
ZK_PROPAGATION_AWAIT_DEADLINE_MS;
+    while (System.currentTimeMillis() < deadlineMs) {
+      HAGroupStoreRecord cached = 
haAdmin.getHAGroupStoreRecordInZooKeeper(haGroupName).getLeft();
+      if (cached != null && cached.getHAGroupState() == expectedState) {
+        return;
+      }
+      Thread.sleep(ZK_PROPAGATION_POLL_INTERVAL_MS);
+    }
+    fail("Timed out waiting for ZK to propagate HAGroupStoreRecord state " + 
expectedState
+      + " for HA group " + haGroupName + " within " + 
ZK_PROPAGATION_AWAIT_DEADLINE_MS + "ms");
+  }
+
+  private static final long ZK_PROPAGATION_AWAIT_DEADLINE_MS = 5_000L;
+  private static final long ZK_PROPAGATION_POLL_INTERVAL_MS = 50L;
 }
diff --git 
a/phoenix-core/src/test/java/org/apache/phoenix/exception/HaFailoverExceptionInheritanceTest.java
 
b/phoenix-core/src/test/java/org/apache/phoenix/exception/HaFailoverExceptionInheritanceTest.java
new file mode 100644
index 0000000000..fb83fe2f73
--- /dev/null
+++ 
b/phoenix-core/src/test/java/org/apache/phoenix/exception/HaFailoverExceptionInheritanceTest.java
@@ -0,0 +1,59 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one
+ * or more contributor license agreements.  See the NOTICE file
+ * distributed with this work for additional information
+ * regarding copyright ownership.  The ASF licenses this file
+ * to you under the Apache License, Version 2.0 (the
+ * "License"); you may not use this file except in compliance
+ * with the License.  You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package org.apache.phoenix.exception;
+
+import static org.junit.Assert.assertTrue;
+
+import java.io.IOException;
+import org.apache.hadoop.hbase.DoNotRetryIOException;
+import org.junit.Test;
+
+/**
+ * Unit tests for the HA failover signaling exceptions thrown server-side when 
a mutation hits an
+ * HA group whose cluster role is in transition. Both classes must extend
+ * {@link DoNotRetryIOException} so HBase's RPC retry layers fail-fast on the 
first hit instead of
+ * consuming the per-call retry budget; Phoenix's outer failover-aware retry 
handles routing to
+ * the new ACTIVE.
+ */
+public class HaFailoverExceptionInheritanceTest {
+
+  @Test
+  public void mutationBlockedExtendsDoNotRetryIOException() {
+    MutationBlockedIOException e = new MutationBlockedIOException("blocked");
+    assertTrue("MutationBlockedIOException must be a DoNotRetryIOException so 
HBase fails fast",
+      e instanceof DoNotRetryIOException);
+    // Transitive type preserved — existing catch (IOException) sites still 
match.
+    assertTrue(e instanceof IOException);
+  }
+
+  @Test
+  public void staleClusterRoleRecordExtendsDoNotRetryIOException() {
+    StaleClusterRoleRecordException e = new 
StaleClusterRoleRecordException("stale");
+    assertTrue(
+      "StaleClusterRoleRecordException must be a DoNotRetryIOException so 
HBase fails fast",
+      e instanceof DoNotRetryIOException);
+    assertTrue(e instanceof IOException);
+  }
+
+  @Test
+  public void staleClusterRoleRecordTwoArgConstructorPreserved() {
+    Throwable cause = new RuntimeException("root");
+    StaleClusterRoleRecordException e = new 
StaleClusterRoleRecordException("stale", cause);
+    assertTrue(e instanceof DoNotRetryIOException);
+  }
+}

Reply via email to