Add new retry policy options

Author: slinkydeveloper

sdk-api-kotlin/src/main/kotlin/dev/restate/sdk/kotlin/endpoint/endpoint.kt

@@ -10,6 +10,7 @@ package dev.restate.sdk.kotlin.endpoint
 
 import dev.restate.sdk.endpoint.Endpoint
 import dev.restate.sdk.endpoint.definition.HandlerDefinition
+import dev.restate.sdk.endpoint.definition.InvocationRetryPolicy
 import dev.restate.sdk.endpoint.definition.ServiceDefinition
 import kotlin.time.Duration
 import kotlin.time.toJavaDuration
@@ -150,6 +151,22 @@ var ServiceDefinition.Configurator.ingressPrivate: Boolean?
     this.ingressPrivate(value)
   }
 
+/**
+ * Retry policy used by Restate when invoking this service.
+ *
+ * <p><b>NOTE:</b> You can set this field only if you register this service against
+ * restate-server >= 1.5, otherwise the service discovery will fail.
+ *
+ * @see InvocationRetryPolicy
+ */
+var ServiceDefinition.Configurator.invocationRetryPolicy: InvocationRetryPolicy?
+  get() {
+    return this.invocationRetryPolicy()
+  }
+  set(value) {
+    this.invocationRetryPolicy(value)
+  }
+
 /**
  * Set the acceptable content type when ingesting HTTP requests. Wildcards can be used, e.g.
  * `application/*` or `*/*`.
@@ -298,3 +315,82 @@ var HandlerDefinition.Configurator.enableLazyState: Boolean?
   set(value) {
     this.enableLazyState(value)
   }
+
+/**
+ * Retry policy used by Restate when invoking this handler.
+ *
+ * <p><b>NOTE:</b> You can set this field only if you register this service against
+ * restate-server >= 1.5, otherwise the service discovery will fail.
+ *
+ * @see InvocationRetryPolicy
+ */
+var HandlerDefinition.Configurator.invocationRetryPolicy: InvocationRetryPolicy?
+  get() {
+    return this.invocationRetryPolicy()
+  }
+  set(value) {
+    this.invocationRetryPolicy(value)
+  }
+
+/** Initial delay before the first retry attempt. If unset, server defaults apply. */
+var InvocationRetryPolicy.Builder.initialInterval: Duration?
+  get() {
+    return this.initialInterval()?.toKotlinDuration()
+  }
+  set(value) {
+    this.initialInterval(value?.toJavaDuration())
+  }
+
+/** Exponential backoff multiplier used to compute the next retry delay. */
+var InvocationRetryPolicy.Builder.exponentiationFactor: Double?
+  get() {
+    return this.exponentiationFactor()
+  }
+  set(value) {
+    this.exponentiationFactor(value)
+  }
+
+/** Upper bound for any computed retry delay. */
+var InvocationRetryPolicy.Builder.maxInterval: Duration?
+  get() {
+    return this.maxInterval()?.toKotlinDuration()
+  }
+  set(value) {
+    this.maxInterval(value?.toJavaDuration())
+  }
+
+/**
+ * Maximum number of attempts before giving up retrying.
+ *
+ * The initial call counts as the first attempt; retries increment the count by 1. When giving up,
+ * the behavior defined with [onMaxAttempts] will be applied.
+ *
+ * @see InvocationRetryPolicy.OnMaxAttempts
+ */
+var InvocationRetryPolicy.Builder.maxAttempts: Int?
+  get() {
+    return this.maxAttempts()
+  }
+  set(value) {
+    this.maxAttempts(value)
+  }
+
+/**
+ * Behavior when reaching max attempts.
+ *
+ * @see InvocationRetryPolicy.OnMaxAttempts
+ */
+var InvocationRetryPolicy.Builder.onMaxAttempts: InvocationRetryPolicy.OnMaxAttempts?
+  get() {
+    return this.onMaxAttempts()
+  }
+  set(value) {
+    this.onMaxAttempts(value)
+  }
+
+/** [InvocationRetryPolicy] builder function. */
+fun invocationRetryPolicy(init: InvocationRetryPolicy.Builder.() -> Unit): InvocationRetryPolicy {
+  val builder = InvocationRetryPolicy.builder()
+  builder.init()
+  return builder.build()
+}

sdk-common/src/main/java/dev/restate/sdk/endpoint/definition/HandlerDefinition.java

@@ -35,6 +35,7 @@ public final class HandlerDefinition<REQ, RES> {
   private final @Nullable Duration journalRetention;
   private final @Nullable Boolean ingressPrivate;
   private final @Nullable Boolean enableLazyState;
+  private final @Nullable InvocationRetryPolicy invocationRetryPolicy;
 
   HandlerDefinition(
       String name,
@@ -51,7 +52,8 @@ public final class HandlerDefinition<REQ, RES> {
       @Nullable Duration workflowRetention,
       @Nullable Duration journalRetention,
       @Nullable Boolean ingressPrivate,
-      @Nullable Boolean enableLazyState) {
+      @Nullable Boolean enableLazyState,
+      @Nullable InvocationRetryPolicy invocationRetryPolicy) {
     this.name = name;
     this.handlerType = handlerType;
     this.acceptContentType = acceptContentType;
@@ -67,6 +69,7 @@ public final class HandlerDefinition<REQ, RES> {
     this.journalRetention = journalRetention;
     this.ingressPrivate = ingressPrivate;
     this.enableLazyState = enableLazyState;
+    this.invocationRetryPolicy = invocationRetryPolicy;
   }
 
   /**
@@ -181,6 +184,14 @@ public HandlerRunner<REQ, RES> getRunner() {
     return enableLazyState;
   }
 
+  /**
+   * @return Retry policy for all requests to this handler
+   * @see Configurator#invocationRetryPolicy(InvocationRetryPolicy)
+   */
+  public @Nullable InvocationRetryPolicy getInvocationRetryPolicy() {
+    return invocationRetryPolicy;
+  }
+
   public HandlerDefinition<REQ, RES> withAcceptContentType(String acceptContentType) {
     return new HandlerDefinition<>(
         name,
@@ -197,7 +208,8 @@ public HandlerDefinition<REQ, RES> withAcceptContentType(String acceptContentTyp
         workflowRetention,
         journalRetention,
         ingressPrivate,
-        enableLazyState);
+        enableLazyState,
+        invocationRetryPolicy);
   }
 
   public HandlerDefinition<REQ, RES> withDocumentation(@Nullable String documentation) {
@@ -216,7 +228,8 @@ public HandlerDefinition<REQ, RES> withDocumentation(@Nullable String documentat
         journalRetention,
         workflowRetention,
         ingressPrivate,
-        enableLazyState);
+        enableLazyState,
+        invocationRetryPolicy);
   }
 
   public HandlerDefinition<REQ, RES> withMetadata(Map<String, String> metadata) {
@@ -235,7 +248,8 @@ public HandlerDefinition<REQ, RES> withMetadata(Map<String, String> metadata) {
         workflowRetention,
         journalRetention,
         ingressPrivate,
-        enableLazyState);
+        enableLazyState,
+        invocationRetryPolicy);
   }
 
   /**
@@ -255,7 +269,8 @@ public HandlerDefinition<REQ, RES> configure(
             workflowRetention,
             journalRetention,
             ingressPrivate,
-            enableLazyState);
+            enableLazyState,
+            invocationRetryPolicy);
     configurator.accept(configuratorObj);
 
     return new HandlerDefinition<>(
@@ -273,7 +288,8 @@ public HandlerDefinition<REQ, RES> configure(
         configuratorObj.workflowRetention,
         configuratorObj.journalRetention,
         configuratorObj.ingressPrivate,
-        configuratorObj.enableLazyState);
+        configuratorObj.enableLazyState,
+        configuratorObj.invocationRetryPolicy);
   }
 
   /** Configurator for a {@link HandlerDefinition}. */
@@ -290,6 +306,7 @@ public static final class Configurator {
     private @Nullable Duration journalRetention;
     private @Nullable Boolean ingressPrivate;
     private @Nullable Boolean enableLazyState;
+    private @Nullable InvocationRetryPolicy invocationRetryPolicy;
 
     private Configurator(
         HandlerType handlerType,
@@ -302,7 +319,8 @@ private Configurator(
         @Nullable Duration workflowRetention,
         @Nullable Duration journalRetention,
         @Nullable Boolean ingressPrivate,
-        @Nullable Boolean enableLazyState) {
+        @Nullable Boolean enableLazyState,
+        @Nullable InvocationRetryPolicy invocationRetryPolicy) {
       this.handlerType = handlerType;
       this.acceptContentType = acceptContentType;
       this.documentation = documentation;
@@ -314,6 +332,7 @@ private Configurator(
       this.journalRetention = journalRetention;
       this.ingressPrivate = ingressPrivate;
       this.enableLazyState = enableLazyState;
+      this.invocationRetryPolicy = invocationRetryPolicy;
     }
 
     /**
@@ -555,6 +574,37 @@ public Configurator enableLazyState(@Nullable Boolean enableLazyState) {
       this.enableLazyState = enableLazyState;
       return this;
     }
+
+    /**
+     * @return configured invocation retry policy
+     * @see #invocationRetryPolicy(InvocationRetryPolicy)
+     */
+    public @Nullable InvocationRetryPolicy invocationRetryPolicy() {
+      return invocationRetryPolicy;
+    }
+
+    /**
+     * Retry policy used by Restate when invoking this handler.
+     *
+     * <p><b>NOTE:</b> You can set this field only if you register this service against
+     * restate-server >= 1.5, otherwise the service discovery will fail.
+     *
+     * @see InvocationRetryPolicy
+     * @return this
+     */
+    public Configurator invocationRetryPolicy(
+        @Nullable InvocationRetryPolicy invocationRetryPolicy) {
+      this.invocationRetryPolicy = invocationRetryPolicy;
+      return this;
+    }
+
+    /**
+     * @see #invocationRetryPolicy(InvocationRetryPolicy)
+     */
+    public Configurator invocationRetryPolicy(InvocationRetryPolicy.Builder invocationRetryPolicy) {
+      this.invocationRetryPolicy = invocationRetryPolicy.build();
+      return this;
+    }
   }
 
   public static <REQ, RES> HandlerDefinition<REQ, RES> of(
@@ -578,6 +628,7 @@ public static <REQ, RES> HandlerDefinition<REQ, RES> of(
         null,
         null,
         null,
+        null,
         null);
   }
 
@@ -598,7 +649,8 @@ && getHandlerType() == that.getHandlerType()
         && Objects.equals(getWorkflowRetention(), that.getWorkflowRetention())
         && Objects.equals(getJournalRetention(), that.getJournalRetention())
         && Objects.equals(getIngressPrivate(), that.getIngressPrivate())
-        && Objects.equals(getEnableLazyState(), that.getEnableLazyState());
+        && Objects.equals(getEnableLazyState(), that.getEnableLazyState())
+        && Objects.equals(getInvocationRetryPolicy(), that.getInvocationRetryPolicy());
   }
 
   @Override
@@ -618,6 +670,7 @@ public int hashCode() {
         getWorkflowRetention(),
         getJournalRetention(),
         getIngressPrivate(),
-        getEnableLazyState());
+        getEnableLazyState(),
+        getInvocationRetryPolicy());
   }
 }