diff --git a/experimental/lambda/src/main/java/io/serverlessworkflow/impl/executors/func/JavaFilterFunctionCallExecutor.java b/experimental/lambda/src/main/java/io/serverlessworkflow/impl/executors/func/JavaFilterFunctionCallExecutor.java new file mode 100644 index 00000000..765177dc --- /dev/null +++ b/experimental/lambda/src/main/java/io/serverlessworkflow/impl/executors/func/JavaFilterFunctionCallExecutor.java @@ -0,0 +1,55 @@ +/* + * 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.impl.executors.func; + +import io.serverlessworkflow.api.types.TaskBase; +import io.serverlessworkflow.api.types.func.CallJava; +import io.serverlessworkflow.api.types.func.JavaFilterFunction; +import io.serverlessworkflow.impl.TaskContext; +import io.serverlessworkflow.impl.WorkflowContext; +import io.serverlessworkflow.impl.WorkflowDefinition; +import io.serverlessworkflow.impl.WorkflowModel; +import io.serverlessworkflow.impl.WorkflowModelFactory; +import io.serverlessworkflow.impl.executors.CallableTask; +import java.util.Optional; +import java.util.concurrent.CompletableFuture; + +public class JavaFilterFunctionCallExecutor + implements CallableTask> { + + private JavaFilterFunction function; + private Optional> inputClass = Optional.empty(); + + @Override + public void init(CallJava.CallJavaFilterFunction task, WorkflowDefinition definition) { + this.function = task.function(); + this.inputClass = task.inputClass(); + } + + @Override + public CompletableFuture apply( + WorkflowContext workflowContext, TaskContext taskContext, WorkflowModel input) { + WorkflowModelFactory mf = workflowContext.definition().application().modelFactory(); + T typedIn = JavaFuncUtils.convertT(input, inputClass); + V out = function.apply(typedIn, workflowContext, taskContext); + return CompletableFuture.completedFuture(mf.fromAny(input, out)); + } + + @Override + public boolean accept(Class clazz) { + return CallJava.CallJavaFilterFunction.class.isAssignableFrom(clazz); + } +} diff --git a/experimental/lambda/src/main/resources/META-INF/services/io.serverlessworkflow.impl.executors.CallableTask b/experimental/lambda/src/main/resources/META-INF/services/io.serverlessworkflow.impl.executors.CallableTask index 59f3338a..a22a1553 100644 --- a/experimental/lambda/src/main/resources/META-INF/services/io.serverlessworkflow.impl.executors.CallableTask +++ b/experimental/lambda/src/main/resources/META-INF/services/io.serverlessworkflow.impl.executors.CallableTask @@ -2,4 +2,5 @@ io.serverlessworkflow.impl.executors.func.JavaLoopFunctionIndexCallExecutor io.serverlessworkflow.impl.executors.func.JavaLoopFunctionCallExecutor io.serverlessworkflow.impl.executors.func.JavaFunctionCallExecutor io.serverlessworkflow.impl.executors.func.JavaConsumerCallExecutor -io.serverlessworkflow.impl.executors.func.JavaContextFunctionCallExecutor \ No newline at end of file +io.serverlessworkflow.impl.executors.func.JavaContextFunctionCallExecutor +io.serverlessworkflow.impl.executors.func.JavaFilterFunctionCallExecutor