Revert "Avoid invalid retries on multiple replicas when querying (opensearch-project#17370)"

This reverts commit eba12fa.

Signed-off-by: Andrew Ross <andrross@amazon.com>

Author: andrross

CHANGELOG.md

@@ -12,7 +12,6 @@ The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/),
 - Add composite directory factory ([#17988](https://github.com/opensearch-project/OpenSearch/pull/17988))
 
 ### Changed
-- Avoid invalid retries in multiple replicas when querying [#17370](https://github.com/opensearch-project/OpenSearch/pull/17370)
 
 ### Dependencies
 - Bump `com.google.code.gson:gson` from 2.12.1 to 2.13.0 ([#17923](https://github.com/opensearch-project/OpenSearch/pull/17923))

libs/core/src/main/java/org/opensearch/OpenSearchException.java

@@ -302,12 +302,8 @@ public Map<String, List<String>> getHeaders() {
      * Returns the rest status code associated with this exception.
      */
     public RestStatus status() {
-        return status(this);
-    }
-
-    public static RestStatus status(Throwable t) {
-        Throwable cause = ExceptionsHelper.unwrapCause(t);
-        if (cause == t) {
+        Throwable cause = unwrapCause();
+        if (cause == this) {
             return RestStatus.INTERNAL_SERVER_ERROR;
         } else {
             return ExceptionsHelper.status(cause);

server/src/main/java/org/opensearch/action/search/AbstractSearchAsyncAction.java

@@ -514,19 +514,10 @@ private void onShardFailure(final int shardIndex, @Nullable SearchShardTarget sh
         // we do make sure to clean it on a successful response from a shard
         setPhaseResourceUsages();
         onShardFailure(shardIndex, shard, e);
+        SearchShardTarget nextShard = FailAwareWeightedRouting.getInstance()
+            .findNext(shardIt, clusterState, e, () -> totalOps.incrementAndGet());
 
-        final SearchShardTarget nextShard;
-        final boolean lastShard;
-        final int advanceShardCount;
-        if (TransportActions.isRetryableSearchException(e)) {
-            nextShard = FailAwareWeightedRouting.getInstance().findNext(shardIt, clusterState, e, () -> totalOps.incrementAndGet());
-            lastShard = nextShard == null;
-            advanceShardCount = 1;
-        } else {
-            nextShard = null;
-            lastShard = true;
-            advanceShardCount = remainingOpsCount(shardIt);
-        }
+        final boolean lastShard = nextShard == null;
         if (logger.isTraceEnabled()) {
             logger.trace(
                 () -> new ParameterizedMessage(
@@ -551,7 +542,7 @@ private void onShardFailure(final int shardIndex, @Nullable SearchShardTarget sh
         if (lastShard) {
             onShardGroupFailure(shardIndex, shard, e);
         }
-        final int totalOps = this.totalOps.addAndGet(advanceShardCount);
+        final int totalOps = this.totalOps.incrementAndGet();
         if (totalOps == expectedTotalOps) {
             try {
                 onPhaseDone();
@@ -570,14 +561,6 @@ private void onShardFailure(final int shardIndex, @Nullable SearchShardTarget sh
         }
     }
 
-    private int remainingOpsCount(SearchShardIterator shardsIt) {
-        if (shardsIt.skip()) {
-            return shardsIt.remaining();
-        } else {
-            return shardsIt.remaining() + 1;
-        }
-    }
-
     /**
      * Executed once for every {@link ShardId} that failed on all available shard routing.
      *
@@ -668,7 +651,12 @@ private void onShardResultConsumed(Result result, SearchShardIterator shardIt) {
     }
 
     private void successfulShardExecution(SearchShardIterator shardsIt) {
-        final int remainingOpsOnIterator = remainingOpsCount(shardsIt);
+        final int remainingOpsOnIterator;
+        if (shardsIt.skip()) {
+            remainingOpsOnIterator = shardsIt.remaining();
+        } else {
+            remainingOpsOnIterator = shardsIt.remaining() + 1;
+        }
         final int xTotalOps = totalOps.addAndGet(remainingOpsOnIterator);
         if (xTotalOps == expectedTotalOps) {
             try {

server/src/main/java/org/opensearch/action/support/TransportActions.java

@@ -34,11 +34,8 @@
 
 import org.apache.lucene.store.AlreadyClosedException;
 import org.opensearch.ExceptionsHelper;
-import org.opensearch.OpenSearchException;
 import org.opensearch.action.NoShardAvailableActionException;
 import org.opensearch.action.UnavailableShardsException;
-import org.opensearch.core.concurrency.OpenSearchRejectedExecutionException;
-import org.opensearch.core.tasks.TaskCancelledException;
 import org.opensearch.index.IndexNotFoundException;
 import org.opensearch.index.shard.IllegalIndexShardStateException;
 import org.opensearch.index.shard.ShardNotFoundException;
@@ -67,12 +64,4 @@ public static boolean isReadOverrideException(Exception e) {
         return !isShardNotAvailableException(e);
     }
 
-    public static boolean isRetryableSearchException(final Exception e) {
-        return (OpenSearchException.status(e).getStatusFamilyCode() != 4) && (e.getCause() instanceof TaskCancelledException == false)
-            // There exists a scenario where a primary shard (0 replicas) relocates and is in POST_RECOVERY on the
-            // target node but already deleted on the source node. Search request should still work.
-            || (e.getCause() instanceof IndexNotFoundException)
-            || (e.getCause() instanceof OpenSearchRejectedExecutionException);
-    }
-
 }

server/src/test/java/org/opensearch/action/search/AbstractSearchAsyncActionTests.java

@@ -49,7 +49,6 @@
 import org.opensearch.core.common.breaker.NoopCircuitBreaker;
 import org.opensearch.core.index.Index;
 import org.opensearch.core.index.shard.ShardId;
-import org.opensearch.core.tasks.TaskCancelledException;
 import org.opensearch.core.tasks.resourcetracker.TaskResourceInfo;
 import org.opensearch.core.tasks.resourcetracker.TaskResourceUsage;
 import org.opensearch.index.query.MatchAllQueryBuilder;
@@ -67,7 +66,6 @@
 import org.opensearch.threadpool.TestThreadPool;
 import org.opensearch.threadpool.ThreadPool;
 import org.opensearch.transport.Transport;
-import org.opensearch.transport.TransportException;
 import org.junit.After;
 import org.junit.Before;
 
@@ -138,7 +136,6 @@ private AbstractSearchAsyncAction<SearchPhaseResult> createAction(
             controlled,
             false,
             false,
-            false,
             expected,
             resourceUsage,
             new SearchShardIterator(null, null, Collections.emptyList(), null)
@@ -151,7 +148,6 @@ private AbstractSearchAsyncAction<SearchPhaseResult> createAction(
         ActionListener<SearchResponse> listener,
         final boolean controlled,
         final boolean failExecutePhaseOnShard,
-        final boolean throw4xxExceptionOnShard,
         final boolean catchExceptionWhenExecutePhaseOnShard,
         final AtomicLong expected,
         final TaskResourceUsage resourceUsage,
@@ -221,11 +217,7 @@ protected void executePhaseOnShard(
                 final SearchActionListener<SearchPhaseResult> listener
             ) {
                 if (failExecutePhaseOnShard) {
-                    if (throw4xxExceptionOnShard) {
-                        listener.onFailure(new TransportException(new TaskCancelledException(shardIt.shardId().toString())));
-                    } else {
-                        listener.onFailure(new ShardNotFoundException(shardIt.shardId()));
-                    }
+                    listener.onFailure(new ShardNotFoundException(shardIt.shardId()));
                 } else {
                     if (catchExceptionWhenExecutePhaseOnShard) {
                         try {
@@ -593,7 +585,6 @@ public void onFailure(Exception e) {
             false,
             true,
             false,
-            false,
             new AtomicLong(),
             new TaskResourceUsage(randomLong(), randomLong()),
             shards
@@ -610,62 +601,6 @@ public void onFailure(Exception e) {
         assertThat(searchResponse.getSuccessfulShards(), equalTo(0));
     }
 
-    public void testSkipInValidRetryInMultiReplicas() throws InterruptedException {
-        final Index index = new Index("test", UUID.randomUUID().toString());
-        final CountDownLatch latch = new CountDownLatch(1);
-        final AtomicBoolean fail = new AtomicBoolean(true);
-
-        List<String> targetNodeIds = List.of("n1", "n2", "n3");
-        final SearchShardIterator[] shards = IntStream.range(2, 4)
-            .mapToObj(i -> new SearchShardIterator(null, new ShardId(index, i), targetNodeIds, null, null, null))
-            .toArray(SearchShardIterator[]::new);
-
-        SearchRequest searchRequest = new SearchRequest().allowPartialSearchResults(true);
-        searchRequest.setMaxConcurrentShardRequests(1);
-
-        final ArraySearchPhaseResults<SearchPhaseResult> queryResult = new ArraySearchPhaseResults<>(shards.length);
-        AbstractSearchAsyncAction<SearchPhaseResult> action = createAction(
-            searchRequest,
-            queryResult,
-            new ActionListener<SearchResponse>() {
-                @Override
-                public void onResponse(SearchResponse response) {
-
-                }
-
-                @Override
-                public void onFailure(Exception e) {
-                    if (fail.compareAndExchange(true, false)) {
-                        try {
-                            throw new RuntimeException("Simulated exception");
-                        } finally {
-                            executor.submit(() -> latch.countDown());
-                        }
-                    }
-                }
-            },
-            false,
-            true,
-            true,
-            false,
-            new AtomicLong(),
-            new TaskResourceUsage(randomLong(), randomLong()),
-            shards
-        );
-        action.run();
-        assertTrue(latch.await(1, TimeUnit.SECONDS));
-        InternalSearchResponse internalSearchResponse = InternalSearchResponse.empty();
-        SearchResponse searchResponse = action.buildSearchResponse(internalSearchResponse, action.buildShardFailures(), null, null);
-        assertSame(searchResponse.getAggregations(), internalSearchResponse.aggregations());
-        assertSame(searchResponse.getSuggest(), internalSearchResponse.suggest());
-        assertSame(searchResponse.getProfileResults(), internalSearchResponse.profile());
-        assertSame(searchResponse.getHits(), internalSearchResponse.hits());
-        assertThat(searchResponse.getSuccessfulShards(), equalTo(0));
-        for (int i = 0; i < shards.length; i++) {
-            assertEquals(targetNodeIds.size() - 1, shards[i].remaining());
-        }
-    }
-
     public void testOnShardSuccessPhaseDoneFailure() throws InterruptedException {
         final Index index = new Index("test", UUID.randomUUID().toString());
         final CountDownLatch latch = new CountDownLatch(1);
@@ -698,7 +633,6 @@ public void onFailure(Exception e) {
             false,
             false,
             false,
-            false,
             new AtomicLong(),
             new TaskResourceUsage(randomLong(), randomLong()),
             shards
@@ -751,7 +685,6 @@ public void onFailure(Exception e) {
             },
             false,
             false,
-            false,
             catchExceptionWhenExecutePhaseOnShard,
             new AtomicLong(),
             new TaskResourceUsage(randomLong(), randomLong()),