Add workflow retention setting at service level (#547)

Author: slinkydeveloper

examples/src/main/java/my/restate/sdk/examples/LoanWorkflow.java

@@ -121,7 +121,11 @@ public Status getStatus(SharedWorkflowContext ctx) {
   }
 
   public static void main(String[] args) {
-    Endpoint endpoint = Endpoint.builder().bind(new LoanWorkflow()).bind(new MockBank()).build();
+    Endpoint endpoint =
+        Endpoint.builder()
+            .bind(new LoanWorkflow(), c -> c.workflowRetention(Duration.ofDays(10)))
+            .bind(new MockBank())
+            .build();
 
     RestateHttpServer.listen(endpoint);
 

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

@@ -88,6 +88,20 @@ var ServiceDefinition.Configurator.abortTimeout: Duration?
     this.abortTimeout(value?.toJavaDuration())
   }
 
+/**
+ * The retention duration for this workflow. This applies only to workflow services.
+ *
+ * *NOTE:* You can set this field only if you register this service against restate-server >= 1.4,
+ * otherwise the service discovery will fail.
+ */
+var ServiceDefinition.Configurator.workflowRetention: Duration?
+  get() {
+    return this.workflowRetention()?.toKotlinDuration()
+  }
+  set(value) {
+    this.workflowRetention(value?.toJavaDuration())
+  }
+
 /**
  * The retention duration of idempotent requests to this service.
  *

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

@@ -362,6 +362,43 @@ public Configurator abortTimeout(@Nullable Duration abortTimeout) {
       return this;
     }
 
+    /**
+     * @return configured workflow retention.
+     * @see #workflowRetention(Duration)
+     */
+    public @Nullable Duration workflowRetention() {
+      return handlers.values().stream()
+          .filter(h -> h.getHandlerType() == HandlerType.WORKFLOW)
+          .findFirst()
+          .orElseThrow(
+              () ->
+                  new IllegalArgumentException(
+                      "Workflow retention cannot be set for non-workflow services"))
+          .getWorkflowRetention();
+    }
+
+    /**
+     * The retention duration of idempotent requests to this workflow service. This applies only to
+     * workflow services.
+     *
+     * <p><b>NOTE:</b> You can set this field only if you register this service against
+     * restate-server >= 1.4, otherwise the service discovery will fail.
+     *
+     * @return this
+     */
+    public Configurator workflowRetention(@Nullable Duration workflowRetention) {
+      String workflowHandlerName =
+          handlers.entrySet().stream()
+              .filter(e -> e.getValue().getHandlerType() == HandlerType.WORKFLOW)
+              .findFirst()
+              .orElseThrow(
+                  () ->
+                      new IllegalArgumentException(
+                          "Workflow retention cannot be set for non-workflow services"))
+              .getKey();
+      return configureHandler(workflowHandlerName, ch -> ch.workflowRetention(workflowRetention));
+    }
+
     /**
      * @return configured idempotency retention.
      * @see #idempotencyRetention(Duration)