Implemented GetActionsForFunctionDefinition #127

Signed-off-by: manick02 <manickavasagam.sundaram@gmail.com>

Author: manick02

utils/src/main/java/io/serverlessworkflow/utils/WorkflowUtils.java

@@ -258,6 +258,97 @@ public static FunctionDefinition getFunctionDefinitionsForAction(
     return functionDefinition.isPresent() ? functionDefinition.get() : null;
   }
 
+  /** @return : Returns @{code List<Action>} which uses a function defintion */
+  public static List<Action> getActionsForFunctionDefinition(
+      Workflow workflow, String functionDefinitionName) {
+    if (!hasFunctionDefs(workflow, functionDefinitionName)) return null;
+    return getActionsWhichUsesFunctionDefinition(workflow, functionDefinitionName);
+  }
+
+  private static List<Action> getActionsWhichUsesFunctionDefinition(
+      Workflow workflow, String functionDefinitionName) {
+    List<Action> actions = new ArrayList<>();
+    for (State state : workflow.getStates()) {
+      if (state instanceof EventState) {
+        EventState eventState = (EventState) state;
+        List<OnEvents> onEvents = eventState.getOnEvents();
+        if (onEvents != null) {
+          for (OnEvents onEvent : onEvents) {
+            if (onEvent != null) {
+              List<Action> onEventActions = onEvent.getActions();
+              if (onEventActions != null) {
+                for (Action onEventAction : onEventActions) {
+                  if (doesActionUsesFuntionDefinition(functionDefinitionName, onEventAction))
+                    actions.add(onEventAction);
+                }
+              }
+            }
+          }
+        }
+      } else if (state instanceof CallbackState) {
+        CallbackState callbackState = (CallbackState) state;
+        final Action callbackStateAction = callbackState.getAction();
+        if (doesActionUsesFuntionDefinition(functionDefinitionName, callbackStateAction)) {
+          actions.add(callbackStateAction);
+        }
+
+      } else if (state instanceof OperationState) {
+        OperationState operationState = (OperationState) state;
+        final List<Action> operationStateActions = operationState.getActions();
+        if (operationStateActions != null) {
+          for (Action operationStateAction : operationStateActions) {
+            if (doesActionUsesFuntionDefinition(functionDefinitionName, operationStateAction)) {
+              actions.add(operationStateAction);
+            }
+          }
+        }
+      } else if (state instanceof ParallelState) {
+        ParallelState parallelState = (ParallelState) state;
+        List<Branch> parallelStateBranches = parallelState.getBranches();
+        if (parallelStateBranches != null) {
+          for (Branch branch : parallelStateBranches) {
+            List<Action> branchActions = branch.getActions();
+            if (branchActions != null) {
+              for (Action branchAction : branchActions) {
+                if (doesActionUsesFuntionDefinition(functionDefinitionName, branchAction)) {
+                  actions.add(branchAction);
+                }
+              }
+            }
+          }
+        }
+      } else if (state instanceof ForEachState) {
+        ForEachState forEachState = (ForEachState) state;
+        List<Action> forEachStateActions = forEachState.getActions();
+        if (forEachStateActions != null) {
+          for (Action forEachStateAction : forEachStateActions) {
+            if (doesActionUsesFuntionDefinition(functionDefinitionName, forEachStateAction)) {
+              actions.add(forEachStateAction);
+            }
+          }
+        }
+      }
+    }
+
+    return actions;
+  }
+
+  private static boolean doesActionUsesFuntionDefinition(
+      String functionDefinitionName, Action forEachStateAction) {
+    return forEachStateAction != null
+        && forEachStateAction.getFunctionRef() != null
+        && forEachStateAction.getFunctionRef().getRefName() != null
+        && forEachStateAction.getFunctionRef().getRefName().equals(functionDefinitionName);
+  }
+
+  private static boolean hasFunctionDefs(Workflow workflow, String functionDefinitionName) {
+    if (!hasFunctionDefs(workflow)) return false;
+    List<FunctionDefinition> functionDefs = workflow.getFunctions().getFunctionDefs();
+    return functionDefs.stream()
+        .anyMatch(
+            functionDefinition -> functionDefinition.getName().equals(functionDefinitionName));
+  }
+
   private static FunctionRef getFunctionRefFromAction(Workflow workflow, String action) {
     if (!hasStates(workflow)) return null;
 

utils/src/test/java/io/serverlessworkflow/util/FunctionDefinitionsTest.java

@@ -19,9 +19,11 @@
 import static org.junit.jupiter.api.Assertions.*;
 
 import io.serverlessworkflow.api.Workflow;
+import io.serverlessworkflow.api.actions.Action;
 import io.serverlessworkflow.api.functions.FunctionDefinition;
 import io.serverlessworkflow.util.testutil.TestUtils;
 import io.serverlessworkflow.utils.WorkflowUtils;
+import java.util.List;
 import org.junit.jupiter.params.ParameterizedTest;
 import org.junit.jupiter.params.provider.ValueSource;
 
@@ -43,7 +45,6 @@ public void testFunctionDefsForAction(String funcDefinitions) {
   @ValueSource(strings = {"/funcdefinitiontest/functiondefinition.yml"})
   public void testFunctionDefsForActionNotPresent(String funcDefinitions) {
     String actionLookUp = "finalizeApplicationFunctionNotPresent";
-    int expectedCount = 0;
     Workflow workflow = TestUtils.createWorkflowFromTestResource(funcDefinitions);
     FunctionDefinition finalizeApplicationFunctionDefinition =
         WorkflowUtils.getFunctionDefinitionsForAction(workflow, actionLookUp);
@@ -55,4 +56,15 @@ public void testFunctionDefsForActionNotPresent(String funcDefinitions) {
   public void testFunctionDefsForNullWorkflow(String funcDefinitions) {
     assertNull(WorkflowUtils.getFunctionDefinitionsForAction(null, "TestAction"));
   }
+
+  @ParameterizedTest
+  @ValueSource(strings = {"/funcdefinitiontest/functiondefinition.yml"})
+  public void testGetActionsForFunctionDefinition(String funcDefinitions) {
+    String functionRefName = "finalizeApplicationFunction";
+    int expectedActionCount = 2;
+    Workflow workflow = TestUtils.createWorkflowFromTestResource(funcDefinitions);
+    List<Action> actionsForFunctionDefinition =
+        WorkflowUtils.getActionsForFunctionDefinition(workflow, functionRefName);
+    assertEquals(expectedActionCount, actionsForFunctionDefinition.size());
+  }
 }