Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Original file line number Diff line number Diff line change
Expand Up @@ -40,6 +40,11 @@ public class AllocationRoutedStep extends ClusterStateWaitStep {
super(key, nextStepKey);
}

@Override
public boolean isRetryable() {
return true;
}

@Override
public Result isConditionMet(Index index, ClusterState clusterState) {
IndexMetadata idxMeta = clusterState.metadata().index(index);
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -18,4 +18,10 @@ public ErrorStep(StepKey key) {
throw new IllegalArgumentException("An error step must have a step key whose step name is " + NAME);
}
}

@Override
public boolean isRetryable() {
// this is marker step so it doesn't make sense to be retryable
return false;
}
}
Original file line number Diff line number Diff line change
Expand Up @@ -88,6 +88,11 @@ public ClusterState performAction(Index index, ClusterState clusterState) {
return newClusterStateBuilder.build();
}

@Override
public boolean isRetryable() {
return true;
}

@Override
public int hashCode() {
return Objects.hash(super.hashCode(), snapshotRepository);
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -20,4 +20,10 @@ public PhaseCompleteStep(StepKey key, StepKey nextStepKey) {
public static PhaseCompleteStep finalStep(String phase) {
return new PhaseCompleteStep(new StepKey(phase, NAME, NAME), null);
}

@Override
public boolean isRetryable() {
// this is marker step so it doesn't make sense to be retryable
return false;
}
}
Original file line number Diff line number Diff line change
Expand Up @@ -42,9 +42,7 @@ public StepKey getNextStepKey() {
/**
* Indicates if the step can be automatically retried when it encounters an execution error.
*/
public boolean isRetryable() {
return false;
}
public abstract boolean isRetryable();

@Override
public int hashCode() {
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -17,4 +17,10 @@ public class TerminalPolicyStep extends Step {
TerminalPolicyStep(StepKey key, StepKey nextStepKey) {
super(key, nextStepKey);
}

@Override
public boolean isRetryable() {
// this is marker step so it doesn't make sense to be retryable
return false;
}
}
Original file line number Diff line number Diff line change
Expand Up @@ -30,6 +30,11 @@ final class WaitForIndexingCompleteStep extends ClusterStateWaitStep {
super(key, nextStepKey);
}

@Override
public boolean isRetryable() {
return true;
}

@Override
public Result isConditionMet(Index index, ClusterState clusterState) {
IndexMetadata followerIndex = clusterState.metadata().index(index);
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -53,6 +53,11 @@ protected IndexMetadata getIndexMetadata() {

public void testBranchStepKeyIsTheWrappedStepKey() {
AsyncActionStep stepToExecute = new AsyncActionStep(randomStepKey(), randomStepKey(), client) {
@Override
public boolean isRetryable() {
return true;
}

@Override
public void performAction(IndexMetadata indexMetadata, ClusterState currentClusterState, ClusterStateObserver observer,
Listener listener) {
Expand All @@ -65,6 +70,11 @@ public void performAction(IndexMetadata indexMetadata, ClusterState currentClust

public void testBranchStepNextKeyOnCompleteResponse() {
AsyncActionStep stepToExecute = new AsyncActionStep(randomStepKey(), randomStepKey(), client) {
@Override
public boolean isRetryable() {
return true;
}

@Override
public void performAction(IndexMetadata indexMetadata, ClusterState currentClusterState, ClusterStateObserver observer,
Listener listener) {
Expand All @@ -91,6 +101,11 @@ public void onFailure(Exception e) {

public void testBranchStepNextKeyOnInCompleteResponse() {
AsyncActionStep stepToExecute = new AsyncActionStep(randomStepKey(), randomStepKey(), client) {
@Override
public boolean isRetryable() {
return true;
}

@Override
public void performAction(IndexMetadata indexMetadata, ClusterState currentClusterState, ClusterStateObserver observer,
Listener listener) {
Expand Down Expand Up @@ -120,6 +135,11 @@ public void onFailure(Exception e) {
public void testBranchStepPropagatesFailure() {
NullPointerException failException = new NullPointerException("fail");
AsyncActionStep stepToExecute = new AsyncActionStep(randomStepKey(), randomStepKey(), client) {
@Override
public boolean isRetryable() {
return true;
}

@Override
public void performAction(IndexMetadata indexMetadata, ClusterState currentClusterState, ClusterStateObserver observer,
Listener listener) {
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -19,6 +19,11 @@ public MockStep(StepKey stepKey, Step.StepKey nextStepKey) {
super(stepKey, nextStepKey);
}

@Override
public boolean isRetryable() {
return false;
}

public MockStep(Step other) {
super(other.getKey(), other.getNextStepKey());
}
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -473,6 +473,11 @@ public ObservableClusterStateWaitStep(StepKey current, StepKey next) {
super(current, next);
}

@Override
public boolean isRetryable() {
return false;
}

public ObservableClusterStateWaitStep(StreamInput in) throws IOException {
this(new StepKey(in.readString(), in.readString(), in.readString()), readOptionalNextStepKey(in));
}
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -888,6 +888,11 @@ static class MockAsyncActionStep extends AsyncActionStep {
super(key, nextStepKey, null);
}

@Override
public boolean isRetryable() {
return false;
}

void setException(Exception exception) {
this.exception = exception;
}
Expand Down Expand Up @@ -937,6 +942,11 @@ static class MockAsyncWaitStep extends AsyncWaitStep {
super(key, nextStepKey, null);
}

@Override
public boolean isRetryable() {
return false;
}

void setException(Exception exception) {
this.exception = exception;
}
Expand Down Expand Up @@ -974,6 +984,11 @@ static class MockClusterStateActionStep extends ClusterStateActionStep {
super(key, nextStepKey);
}

@Override
public boolean isRetryable() {
return false;
}

public void setException(RuntimeException exception) {
this.exception = exception;
}
Expand Down Expand Up @@ -1010,6 +1025,11 @@ static class MockClusterStateWaitStep extends ClusterStateWaitStep {
super(key, nextStepKey);
}

@Override
public boolean isRetryable() {
return false;
}

public void setException(RuntimeException exception) {
this.exception = exception;
}
Expand Down