Skip to content

Commit 1f556e5

Browse files
committed
generating metadata for action events
Signed-off-by: Pedro Escaleira <[email protected]>
1 parent 6a80846 commit 1f556e5

File tree

1 file changed

+11
-4
lines changed

1 file changed

+11
-4
lines changed

serverlessworkflow/sdk/state_machine_generator.py

Lines changed: 11 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -447,7 +447,11 @@ def generate_actions_info(
447447
ns := self.state_machine.state_cls(name)
448448
)
449449
ns.tags = ["event"]
450-
self.get_action_event(state=ns, e_name=name)
450+
self.get_action_event(
451+
state=ns,
452+
e_name=action.eventRef.triggerEventRef,
453+
er_name=action.eventRef.resultEventRef,
454+
)
451455
if name:
452456
if action_mode == "sequential":
453457
if i < len(actions) - 1:
@@ -499,7 +503,9 @@ def generate_actions_info(
499503
)
500504
ns.tags = ["event"]
501505
self.get_action_event(
502-
state=ns, e_name=next_name
506+
state=ns,
507+
e_name=action.eventRef.triggerEventRef,
508+
er_name=action.eventRef.resultEventRef,
503509
)
504510
self.state_machine.add_transition(
505511
trigger="",
@@ -536,13 +542,14 @@ def get_action_function(self, state: NestedState, f_name: str):
536542
state.metadata = {"function": current_function}
537543
break
538544

539-
def get_action_event(self, state: NestedState, e_name: str):
545+
def get_action_event(self, state: NestedState, e_name: str, er_name: str = ""):
540546
if self.workflow.events:
541547
for event in self.workflow.events:
542548
current_event = event.serialize().__dict__
543549
if current_event["name"] == e_name:
544550
state.metadata = {"event": current_event}
545-
break
551+
if current_event["name"] == er_name:
552+
state.metadata = {"result_event": current_event}
546553

547554
def subflow_state_name(self, action: Action, subflow: Workflow):
548555
return (

0 commit comments

Comments
 (0)