Implemented GetworkflowProducedEvents and GetworkflowProducedEventCount utility methods

Signed-off-by: manick02 <[email protected]>

Author: manick02

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

@@ -16,6 +16,7 @@
 package io.serverlessworkflow.utils;
 
 import io.serverlessworkflow.api.Workflow;
+import io.serverlessworkflow.api.end.End;
 import io.serverlessworkflow.api.events.EventDefinition;
 import io.serverlessworkflow.api.interfaces.State;
 import io.serverlessworkflow.api.start.Start;
@@ -145,7 +146,22 @@ public static List<EventDefinition> getWorkflowConsumedEvents(Workflow workflow)
    * @return Returns {@code List<EventDefinition>}
    */
   public static List<EventDefinition> getWorkflowProducedEvents(Workflow workflow) {
-    return null;
+    if (workflow == null || workflow.getStates() == null || workflow.getStates().size() == 0) {
+      return null;
+    }
+    List<EventDefinition> definedProducedEvents =
+        getDefinedEvents(workflow, EventDefinition.Kind.PRODUCED);
+    Set<String> uniqueEvents = new HashSet<>();
+    for (State state : workflow.getStates()) {
+      End end = state.getEnd();
+      if (end != null || end.getProduceEvents() != null || end.getProduceEvents().size() != 0) {
+        end.getProduceEvents()
+            .forEach(produceEvent -> uniqueEvents.add(produceEvent.getEventRef()));
+      }
+    }
+    return definedProducedEvents.stream()
+        .filter(eventDefinition -> uniqueEvents.contains(eventDefinition.getName()))
+        .collect(Collectors.toList());
   }
 
   /**
@@ -202,4 +218,13 @@ public static int getWorkflowConsumedEventsCount(Workflow workflow) {
     List<EventDefinition> workflowConsumedEvents = getWorkflowConsumedEvents(workflow);
     return workflowConsumedEvents == null ? 0 : workflowConsumedEvents.size();
   }
+
+  /**
+   * @return Returns {@code int} Count of the workflow produced events. <strong>Does not</strong>
+   *     consider sub-workflows in the count
+   */
+  public static int getWorkflowProducedEventsCount(Workflow workflow) {
+    List<EventDefinition> workflowProducedEvents = getWorkflowProducedEvents(workflow);
+    return workflowProducedEvents == null ? 0 : workflowProducedEvents.size();
+  }
 }

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

@@ -50,4 +50,27 @@ public void testGetDefinedConsumedEventsCount(String workflowEvents) {
     int workflowConsumedEventsCount = WorkflowUtils.getWorkflowConsumedEventsCount(workflow);
     Arrays.asList(expectedEventsCount, workflowConsumedEventsCount);
   }
+
+  @ParameterizedTest
+  @ValueSource(strings = {"/events/workflowwithproducedevents.yml"})
+  public void testGetWorkflowProducedEvents(String workflowProducedEvents) {
+    int expectedEventsCount = 1;
+    Collection<String> expectedProducedEvent = Arrays.asList("ApplicationSubmitted");
+    Workflow workflow = TestUtils.createWorkflowFromTestResource(workflowProducedEvents);
+    List<EventDefinition> producedEvents = WorkflowUtils.getWorkflowProducedEvents(workflow);
+    assertNotNull(producedEvents);
+    assertEquals(expectedEventsCount, producedEvents.size());
+    for (EventDefinition producedEvent : producedEvents) {
+      assertTrue(expectedProducedEvent.contains(producedEvent.getName()));
+    }
+  }
+
+  @ParameterizedTest
+  @ValueSource(strings = {"/events/workflowwithproducedevents.yml"})
+  public void testGetWorkflowProducedEventsCount(String workflowProducedEvents) {
+    int expectedEventsCount = 1;
+    Workflow workflow = TestUtils.createWorkflowFromTestResource(workflowProducedEvents);
+    int producedEventsCount = WorkflowUtils.getWorkflowProducedEventsCount(workflow);
+    assertEquals(expectedEventsCount, producedEventsCount);
+  }
 }

utils/src/test/resources/events/workflowwithproducedevents.yml

@@ -0,0 +1,59 @@
+id: finalizeCollegeApplication
+name: Finalize College Application
+version: '1.0'
+specVersion: '0.7'
+events:
+  - name: ApplicationSubmitted
+    type: org.application.submitted
+    source: applicationsource
+    kind: produced
+    correlation:
+      - contextAttributeName: applicantId
+  - name: SATScoresReceived
+    type: org.application.satscores
+    source: applicationsource
+    correlation:
+      - contextAttributeName: applicantId
+  - name: RecommendationLetterReceived
+    type: org.application.recommendationLetter
+    source: applicationsource
+    correlation:
+      - contextAttributeName: applicantId
+functions:
+  - name: finalizeApplicationFunction
+    operation: http://myapis.org/collegeapplicationapi.json#finalize
+states:
+  - name: FinalizeApplication
+    type: event
+    exclusive: false
+    onEvents:
+      - eventRefs:
+          - ApplicationSubmitted
+          - SATScoresReceived
+          - RecommendationLetterReceived
+        actions:
+          - functionRef:
+              refName: finalizeApplicationFunction
+              arguments:
+                student: "${ .applicantId }"
+    end:
+      terminate: true
+      produceEvents:
+        - eventRef: ApplicationSubmitted
+          data: "${ .provisionedOrders }"
+
+  - name: CancelApplication
+    type: event
+    exclusive: false
+    onEvents:
+      - eventRefs:
+          - ApplicationSubmitted
+          - SATScoresReceived
+          - RecommendationLetterReceived
+        actions:
+          - functionRef:
+              refName: finalizeApplicationFunction
+              arguments:
+                student: "${ .applicantId }"
+    end:
+      terminate: true