Add Fluent DSL Equivalents for Try/Catch/Retry Workflows

Signed-off-by: Matheus André <matheusandr2@gmail.com>

Author: matheusandre1

fluent/spec/src/main/java/io/serverlessworkflow/fluent/spec/BaseTryTaskBuilder.java

@@ -113,6 +113,11 @@ public TryTaskCatchBuilder<T> retry(Consumer<RetryPolicyBuilder> consumer) {
       return this;
     }
 
+    public TryTaskCatchBuilder<T> retry(String reference) {
+      this.tryTaskCatch.setRetry(new Retry().withRetryPolicyReference(reference));
+      return this;
+    }
+
     public TryTaskCatchBuilder<T> errorsWith(Consumer<CatchErrorsBuilder> consumer) {
       final CatchErrorsBuilder catchErrorsBuilder = new CatchErrorsBuilder();
       consumer.accept(catchErrorsBuilder);

fluent/spec/src/main/java/io/serverlessworkflow/fluent/spec/UseBuilder.java

@@ -19,13 +19,18 @@
 import io.serverlessworkflow.api.types.ErrorDetails;
 import io.serverlessworkflow.api.types.ErrorTitle;
 import io.serverlessworkflow.api.types.ErrorType;
+import io.serverlessworkflow.api.types.RetryPolicy;
 import io.serverlessworkflow.api.types.UriTemplate;
 import io.serverlessworkflow.api.types.Use;
 import io.serverlessworkflow.api.types.UseAuthentications;
 import io.serverlessworkflow.api.types.UseErrors;
+import io.serverlessworkflow.api.types.UseRetries;
+import io.serverlessworkflow.fluent.spec.BaseTryTaskBuilder.RetryPolicyBuilder;
 import java.net.URI;
 import java.net.URISyntaxException;
+import java.util.LinkedHashMap;
 import java.util.List;
+import java.util.Map;
 import java.util.function.Consumer;
 
 public class UseBuilder {
@@ -66,6 +71,17 @@ public UseBuilder errors(Consumer<UseErrorsBuilder> errorsConsumer) {
     return this;
   }
 
+  public UseBuilder retries(Consumer<UseRetriesBuilder> retriesConsumer) {
+    final UseRetriesBuilder retriesBuilder = new UseRetriesBuilder();
+    retriesConsumer.accept(retriesBuilder);
+    final UseRetries built = retriesBuilder.build();
+    if (this.use.getRetries() == null) {
+      this.use.setRetries(new UseRetries());
+    }
+    this.use.getRetries().getAdditionalProperties().putAll(built.getAdditionalProperties());
+    return this;
+  }
+
   public Use build() {
     return use;
   }
@@ -132,4 +148,23 @@ public Error build() {
       return error;
     }
   }
+
+  public static final class UseRetriesBuilder {
+    private final Map<String, RetryPolicy> retries = new LinkedHashMap<>();
+
+    UseRetriesBuilder() {}
+
+    public UseRetriesBuilder retry(String name, Consumer<RetryPolicyBuilder> configurer) {
+      final RetryPolicyBuilder policyBuilder = new RetryPolicyBuilder();
+      configurer.accept(policyBuilder);
+      this.retries.put(name, policyBuilder.build());
+      return this;
+    }
+
+    public UseRetries build() {
+      final UseRetries useRetries = new UseRetries();
+      useRetries.getAdditionalProperties().putAll(this.retries);
+      return useRetries;
+    }
+  }
 }

fluent/spec/src/main/java/io/serverlessworkflow/fluent/spec/dsl/BaseCallHttpSpec.java

@@ -122,6 +122,11 @@ default SELF query(String name, String value) {
     return self();
   }
 
+  default SELF redirect(boolean redirect) {
+    steps().add(c -> c.redirect(redirect));
+    return self();
+  }
+
   default void accept(CallHttpTaskFluent<?> b) {
     for (var s : steps()) {
       s.accept(b);

fluent/spec/src/main/java/io/serverlessworkflow/fluent/spec/dsl/RetrySpec.java

@@ -15,6 +15,7 @@
  */
 package io.serverlessworkflow.fluent.spec.dsl;
 
+import io.serverlessworkflow.fluent.spec.DurationInlineBuilder;
 import io.serverlessworkflow.fluent.spec.TryTaskBuilder;
 import io.serverlessworkflow.fluent.spec.configurers.RetryConfigurer;
 import java.util.LinkedList;
@@ -51,11 +52,31 @@ public RetrySpec limit(Consumer<TryTaskBuilder.RetryLimitBuilder> retry) {
     return this;
   }
 
+  public RetrySpec delay(String expression) {
+    steps.add(r -> r.delay(expression));
+    return this;
+  }
+
+  public RetrySpec delay(Consumer<DurationInlineBuilder> duration) {
+    steps.add(r -> r.delay(duration));
+    return this;
+  }
+
   public RetrySpec backoff(Consumer<TryTaskBuilder.BackoffBuilder> backoff) {
     steps.add(r -> r.backoff(backoff));
     return this;
   }
 
+  public RetrySpec backoffExponential() {
+    steps.add(r -> r.backoff(b -> b.exponential("e", "1.5")));
+    return this;
+  }
+
+  public RetrySpec backoffConstant() {
+    steps.add(r -> r.backoff(b -> b.constant("c", "10")));
+    return this;
+  }
+
   public RetrySpec jitter(Consumer<TryTaskBuilder.RetryPolicyJitterBuilder> jitter) {
     steps.add(r -> r.jitter(jitter));
     return this;

fluent/spec/src/main/java/io/serverlessworkflow/fluent/spec/dsl/TryCatchSpec.java

@@ -64,6 +64,16 @@ public TryCatchSpec errors(Consumer<TryTaskBuilder.CatchErrorsBuilder> errors) {
     return this;
   }
 
+  public TryCatchSpec as(String errorVarName) {
+    steps.add(t -> t.as(errorVarName));
+    return this;
+  }
+
+  public TryCatchSpec retry(String reference) {
+    steps.add(t -> t.retry(reference));
+    return this;
+  }
+
   public RetrySpec retry() {
     return retry;
   }

fluent/spec/src/main/java/io/serverlessworkflow/fluent/spec/dsl/UseSpec.java

@@ -46,6 +46,11 @@ public UseSpec errors(Consumer<UseBuilder.UseErrorsBuilder> errorsConsumer) {
     return this;
   }
 
+  public UseSpec retries(Consumer<UseBuilder.UseRetriesBuilder> retriesConsumer) {
+    steps.add(u -> u.retries(retriesConsumer));
+    return this;
+  }
+
   @Override
   public void accept(UseBuilder useBuilder) {
     steps.forEach(step -> step.accept(useBuilder));