feat: re-schedule from error handler (#2023)

Signed-off-by: csviri <[email protected]>
Signed-off-by: Attila Mészáros <[email protected]>

Author: csviri

operator-framework-core/src/main/java/io/javaoperatorsdk/operator/api/reconciler/BaseControl.java

@@ -9,7 +9,7 @@ public abstract class BaseControl<T extends BaseControl<T>> {
   private Long scheduleDelay = null;
 
   public T rescheduleAfter(long delay) {
-    this.scheduleDelay = delay;
+    rescheduleAfter(Duration.ofMillis(delay));
     return (T) this;
   }
 

operator-framework-core/src/main/java/io/javaoperatorsdk/operator/api/reconciler/ErrorStatusUpdateControl.java

@@ -1,16 +1,17 @@
 package io.javaoperatorsdk.operator.api.reconciler;
 
+import java.time.Duration;
 import java.util.Optional;
 
 import io.fabric8.kubernetes.api.model.HasMetadata;
 
-public class ErrorStatusUpdateControl<P extends HasMetadata> {
+public class ErrorStatusUpdateControl<P extends HasMetadata>
+    extends BaseControl<ErrorStatusUpdateControl<P>> {
 
   private final P resource;
   private final boolean patch;
   private boolean noRetry = false;
 
-
   public static <T extends HasMetadata> ErrorStatusUpdateControl<T> patchStatus(T resource) {
     return new ErrorStatusUpdateControl<>(resource, true);
   }
@@ -49,4 +50,16 @@ public boolean isNoRetry() {
   public boolean isPatch() {
     return patch;
   }
+
+  /**
+   * If re-scheduled using this method, it is not considered as retry, it effectively cancels retry.
+   *
+   * @param delay for next execution
+   * @return ErrorStatusUpdateControl
+   */
+  @Override
+  public ErrorStatusUpdateControl<P> rescheduleAfter(Duration delay) {
+    withNoRetry();
+    return super.rescheduleAfter(delay);
+  }
 }

operator-framework-core/src/main/java/io/javaoperatorsdk/operator/processing/event/ReconciliationDispatcher.java

@@ -201,13 +201,17 @@ public boolean isLastAttempt() {
                   .updateStatus(errorStatusUpdateControl.getResource().orElseThrow());
         }
         if (errorStatusUpdateControl.isNoRetry()) {
+          PostExecutionControl<P> postExecutionControl;
           if (updatedResource != null) {
-            return errorStatusUpdateControl.isPatch()
+            postExecutionControl = errorStatusUpdateControl.isPatch()
                 ? PostExecutionControl.customResourceStatusPatched(updatedResource)
                 : PostExecutionControl.customResourceUpdated(updatedResource);
           } else {
-            return PostExecutionControl.defaultDispatch();
+            postExecutionControl = PostExecutionControl.defaultDispatch();
           }
+          errorStatusUpdateControl.getScheduleDelay()
+              .ifPresent(postExecutionControl::withReSchedule);
+          return postExecutionControl;
         }
       } catch (RuntimeException ex) {
         log.error("Error during error status handling.", ex);

operator-framework-core/src/test/java/io/javaoperatorsdk/operator/processing/event/ReconciliationDispatcherTest.java

@@ -673,6 +673,24 @@ void retriesAddingFinalizer() {
     verify(customResourceFacade, times(2)).updateResource(any());
   }
 
+  @Test
+  void reSchedulesFromErrorHandler() {
+    var delay = 1000L;
+    testCustomResource.addFinalizer(DEFAULT_FINALIZER);
+    reconciler.reconcile = (r, c) -> {
+      throw new IllegalStateException("Error Status Test");
+    };
+    reconciler.errorHandler =
+        (r, ri, e) -> ErrorStatusUpdateControl.<TestCustomResource>noStatusUpdate()
+            .rescheduleAfter(delay);
+
+    var res = reconciliationDispatcher.handleExecution(
+        new ExecutionScope(null).setResource(testCustomResource));
+
+    assertThat(res.getReScheduleDelay()).contains(delay);
+    assertThat(res.getRuntimeException()).isEmpty();
+  }
+
   private ObservedGenCustomResource createObservedGenCustomResource() {
     ObservedGenCustomResource observedGenCustomResource = new ObservedGenCustomResource();
     observedGenCustomResource.setMetadata(new ObjectMeta());