Add consumer and JavaContextFunction shortcuts

Signed-off-by: Ricardo Zanini <[email protected]>

Author: ricardozanini

experimental/fluent/func/src/main/java/io/serverlessworkflow/fluent/func/FuncCallTaskBuilder.java

@@ -17,9 +17,11 @@
 
 import io.serverlessworkflow.api.types.func.CallJava;
 import io.serverlessworkflow.api.types.func.CallTaskJava;
+import io.serverlessworkflow.api.types.func.JavaContextFunction;
 import io.serverlessworkflow.fluent.func.spi.ConditionalTaskBuilder;
 import io.serverlessworkflow.fluent.func.spi.FuncTaskTransformations;
 import io.serverlessworkflow.fluent.spec.TaskBaseBuilder;
+import java.util.function.Consumer;
 import java.util.function.Function;
 
 public class FuncCallTaskBuilder extends TaskBaseBuilder<FuncCallTaskBuilder>
@@ -48,6 +50,31 @@ public <T, V> FuncCallTaskBuilder function(Function<T, V> function, Class<T> arg
     return this;
   }
 
+  public <T, V> FuncCallTaskBuilder function(JavaContextFunction<T, V> function) {
+    return function(function, null);
+  }
+
+  public <T, V> FuncCallTaskBuilder function(
+      JavaContextFunction<T, V> function, Class<T> argClass) {
+    this.callTaskJava = new CallTaskJava(CallJava.function(function, argClass));
+    super.setTask(this.callTaskJava.getCallJava());
+    return this;
+  }
+
+  /** Accept a side-effect Consumer; engine should pass input through unchanged. */
+  public <T> FuncCallTaskBuilder consumer(Consumer<T> consumer) {
+    this.callTaskJava = new CallTaskJava(CallJava.consumer(consumer));
+    super.setTask(this.callTaskJava.getCallJava());
+    return this;
+  }
+
+  /** Accept a Consumer with explicit input type hint. */
+  public <T> FuncCallTaskBuilder consumer(Consumer<T> consumer, Class<T> argClass) {
+    this.callTaskJava = new CallTaskJava(CallJava.consumer(consumer, argClass));
+    super.setTask(this.callTaskJava.getCallJava());
+    return this;
+  }
+
   public CallTaskJava build() {
     return this.callTaskJava;
   }

experimental/fluent/func/src/main/java/io/serverlessworkflow/fluent/func/dsl/ConsumeStep.java

@@ -0,0 +1,58 @@
+/*
+ * Copyright 2020-Present The Serverless Workflow Specification Authors
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package io.serverlessworkflow.fluent.func.dsl;
+
+import io.serverlessworkflow.fluent.func.FuncCallTaskBuilder;
+import io.serverlessworkflow.fluent.func.FuncTaskItemListBuilder;
+import java.util.function.Consumer;
+
+public final class ConsumeStep<T> extends Step<ConsumeStep<T>, FuncCallTaskBuilder> {
+  private final String name; // may be null
+  private final Consumer<T> consumer;
+  private final Class<T> argClass; // may be null
+
+  ConsumeStep(Consumer<T> consumer, Class<T> argClass) {
+    this(null, consumer, argClass);
+  }
+
+  ConsumeStep(String name, Consumer<T> consumer, Class<T> argClass) {
+    this.name = name;
+    this.consumer = consumer;
+    this.argClass = argClass;
+  }
+
+  @Override
+  protected void configure(
+      FuncTaskItemListBuilder list, java.util.function.Consumer<FuncCallTaskBuilder> post) {
+    if (name == null) {
+      list.callFn(
+          cb -> {
+            // prefer the typed consumer if your builder supports it; otherwise fallback:
+            if (argClass != null) cb.consumer(consumer, argClass);
+            else cb.consumer(consumer);
+            post.accept(cb);
+          });
+    } else {
+      list.callFn(
+          name,
+          cb -> {
+            if (argClass != null) cb.consumer(consumer, argClass);
+            else cb.consumer(consumer);
+            post.accept(cb);
+          });
+    }
+  }
+}

experimental/fluent/func/src/main/java/io/serverlessworkflow/fluent/func/dsl/CtxBiFunction.java

@@ -0,0 +1,29 @@
+/*
+ * Copyright 2020-Present The Serverless Workflow Specification Authors
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package io.serverlessworkflow.fluent.func.dsl;
+
+import io.serverlessworkflow.impl.WorkflowContextData;
+
+/**
+ * Functions that expect a {@link WorkflowContextData} injection in runtime
+ *
+ * @param <T> The task payload input
+ * @param <R> The task result output
+ */
+@FunctionalInterface
+public interface CtxBiFunction<T, R> {
+  R apply(WorkflowContextData context, T payload);
+}

experimental/fluent/func/src/main/java/io/serverlessworkflow/fluent/func/dsl/FuncCallStep.java

@@ -15,42 +15,61 @@
  */
 package io.serverlessworkflow.fluent.func.dsl;
 
+import io.serverlessworkflow.api.types.func.JavaContextFunction;
 import io.serverlessworkflow.fluent.func.FuncCallTaskBuilder;
 import io.serverlessworkflow.fluent.func.FuncTaskItemListBuilder;
 import java.util.function.Consumer;
 import java.util.function.Function;
 
-// FuncCallStep
 public final class FuncCallStep<T, R> extends Step<FuncCallStep<T, R>, FuncCallTaskBuilder> {
-  private final String name; // may be null
+
+  private final String name;
   private final Function<T, R> fn;
+  private final JavaContextFunction<T, R> ctxFn;
   private final Class<T> argClass;
 
+  /** Function<T,R> variant (unnamed). */
   FuncCallStep(Function<T, R> fn, Class<T> argClass) {
     this(null, fn, argClass);
   }
 
+  /** Function<T,R> variant (named). */
   FuncCallStep(String name, Function<T, R> fn, Class<T> argClass) {
     this.name = name;
     this.fn = fn;
+    this.ctxFn = null;
+    this.argClass = argClass;
+  }
+
+  /** JavaContextFunction<T,R> variant (unnamed). */
+  FuncCallStep(JavaContextFunction<T, R> ctxFn, Class<T> argClass) {
+    this(null, ctxFn, argClass);
+  }
+
+  /** JavaContextFunction<T,R> variant (named). */
+  FuncCallStep(String name, JavaContextFunction<T, R> ctxFn, Class<T> argClass) {
+    this.name = name;
+    this.fn = null;
+    this.ctxFn = ctxFn;
     this.argClass = argClass;
   }
 
   @Override
   protected void configure(FuncTaskItemListBuilder list, Consumer<FuncCallTaskBuilder> post) {
-    if (name == null) {
-      list.callFn(
-          cb -> {
+    final Consumer<FuncCallTaskBuilder> apply =
+        cb -> {
+          if (ctxFn != null) {
+            cb.function(ctxFn, argClass);
+          } else {
             cb.function(fn, argClass);
-            post.accept(cb);
-          });
+          }
+          post.accept(cb);
+        };
+
+    if (name == null) {
+      list.callFn(apply);
     } else {
-      list.callFn(
-          name,
-          cb -> {
-            cb.function(fn, argClass);
-            post.accept(cb);
-          });
+      list.callFn(name, apply);
     }
   }
 }