chore: ensure the event group id is kept for async workflow (#7668)

* chore: ensure the event group id is kept for async workflow

* revert test

* fix tests

---------

Co-authored-by: Oli Juhl <59018053+olivermrbl@users.noreply.github.com>
Co-authored-by: Riqwan Thamir <rmthamir@gmail.com>

packages/core/orchestration/src/transaction/transaction-orchestrator.ts

@@ -28,6 +28,10 @@ export type TransactionFlow = {
   options?: TransactionModelOptions
   definition: TransactionStepsDefinition
   transactionId: string
+  metadata?: {
+    eventGroupId?: string
+    [key: string]: unknown
+  }
   hasAsyncSteps: boolean
   hasFailedSteps: boolean
   hasWaitingSteps: boolean
@@ -839,7 +843,7 @@ export class TransactionOrchestrator extends EventEmitter {
     await this.executeNext(transaction)
   }
 
-  private createTransactionFlow(transactionId: string): TransactionFlow {
+  private createTransactionFlow(transactionId: string, flowMetadata?: TransactionFlow['metadata']): TransactionFlow {
     const [steps, features] = TransactionOrchestrator.buildSteps(
       this.definition
     )
@@ -864,6 +868,7 @@ export class TransactionOrchestrator extends EventEmitter {
       modelId: this.id,
       options: this.options,
       transactionId: transactionId,
+      metadata: flowMetadata,
       hasAsyncSteps,
       hasFailedSteps: false,
       hasSkippedSteps: false,
@@ -1006,11 +1011,13 @@ export class TransactionOrchestrator extends EventEmitter {
    * @param transactionId - unique identifier of the transaction
    * @param handler - function to handle action of the transaction
    * @param payload - payload to be passed to all the transaction steps
+   * @param flowMetadata - flow metadata which can include event group id for example
    */
   public async beginTransaction(
     transactionId: string,
     handler: TransactionStepHandler,
-    payload?: unknown
+    payload?: unknown,
+    flowMetadata?: TransactionFlow["metadata"]
   ): Promise<DistributedTransaction> {
     const existingTransaction =
       await TransactionOrchestrator.loadTransactionById(this.id, transactionId)
@@ -1018,7 +1025,7 @@ export class TransactionOrchestrator extends EventEmitter {
     let newTransaction = false
     let modelFlow: TransactionFlow
     if (!existingTransaction) {
-      modelFlow = this.createTransactionFlow(transactionId)
+      modelFlow = this.createTransactionFlow(transactionId, flowMetadata)
       newTransaction = true
     } else {
       modelFlow = existingTransaction.flow

packages/core/orchestration/src/workflow/local-workflow.ts

@@ -13,6 +13,7 @@ import {
   DistributedTransaction,
   DistributedTransactionEvent,
   DistributedTransactionEvents,
+  TransactionFlow,
   TransactionModelOptions,
   TransactionOrchestrator,
   TransactionStepsDefinition,
@@ -332,7 +333,8 @@ export class LocalWorkflow {
     uniqueTransactionId: string,
     input?: unknown,
     context?: Context,
-    subscribe?: DistributedTransactionEvents
+    subscribe?: DistributedTransactionEvents,
+    flowMetadata?: TransactionFlow["metadata"]
   ) {
     if (this.flow.hasChanges) {
       this.commit()
@@ -343,7 +345,8 @@ export class LocalWorkflow {
     const transaction = await orchestrator.beginTransaction(
       uniqueTransactionId,
       handler(this.container_, context),
-      input
+      input,
+      flowMetadata
     )
 
     const { cleanUpEventListeners } = this.registerEventCallbacks({

packages/core/workflows-sdk/src/helper/workflow-export.ts

@@ -1,10 +1,12 @@
 import { MedusaModule } from "@medusajs/modules-sdk"
 import {
+  DistributedTransaction,
+  DistributedTransactionEvents,
   LocalWorkflow,
   TransactionHandlerType,
   TransactionState,
 } from "@medusajs/orchestration"
-import { LoadedModule, MedusaContainer } from "@medusajs/types"
+import { Context, LoadedModule, MedusaContainer } from "@medusajs/types"
 import { isPresent, MedusaContextType } from "@medusajs/utils"
 import { EOL } from "os"
 import { ulid } from "ulid"
@@ -59,7 +61,10 @@ function createContextualWorkflowRunner<
       isCancel = false,
       container: executionContainer,
     },
-    ...args
+    transactionOrIdOrIdempotencyKey: DistributedTransaction | string,
+    input: unknown,
+    context: Context,
+    events: DistributedTransactionEvents | undefined = {}
   ) => {
     if (!executionContainer) {
       const container_ = flow.container as MedusaContainer
@@ -74,6 +79,18 @@ function createContextualWorkflowRunner<
       flow.container = executionContainer
     }
 
+    const { eventGroupId } = context
+    const flowMetadata = {
+      eventGroupId,
+    }
+
+    const args = [
+      transactionOrIdOrIdempotencyKey,
+      input,
+      context,
+      events,
+      flowMetadata,
+    ]
     const transaction = await method.apply(method, args)
 
     let errors = transaction.getErrors(TransactionHandlerType.INVOKE)
@@ -136,7 +153,7 @@ function createContextualWorkflowRunner<
 
     const context = {
       ...outerContext,
-      __type: MedusaContextType,
+      __type: MedusaContextType as Context["__type"],
     }
 
     context.transactionId ??= ulid()
@@ -191,7 +208,7 @@ function createContextualWorkflowRunner<
     const context = {
       ...outerContext,
       transactionId,
-      __type: MedusaContextType,
+      __type: MedusaContextType as Context["__type"],
     }
 
     context.eventGroupId ??= ulid()
@@ -229,7 +246,7 @@ function createContextualWorkflowRunner<
     const context = {
       ...outerContext,
       transactionId,
-      __type: MedusaContextType,
+      __type: MedusaContextType as Context["__type"],
     }
 
     context.eventGroupId ??= ulid()
@@ -262,7 +279,7 @@ function createContextualWorkflowRunner<
     const context = {
       ...outerContext,
       transactionId,
-      __type: MedusaContextType,
+      __type: MedusaContextType as Context["__type"],
     }
 
     context.eventGroupId ??= ulid()
@@ -275,7 +292,8 @@ function createContextualWorkflowRunner<
         isCancel: true,
         container,
       },
-      transaction ?? transactionId,
+      transaction ?? transactionId!,
+      undefined,
       context,
       events
     )

packages/core/workflows-sdk/src/utils/composer/create-step.ts

@@ -135,7 +135,9 @@ function applyStep<
           attempt: metadata.attempt,
           container: stepArguments.container,
           metadata,
-          eventGroupId: stepArguments.context!.eventGroupId,
+          eventGroupId:
+            stepArguments.transaction.getFlow()?.metadata?.eventGroupId ??
+            stepArguments.context!.eventGroupId,
           transactionId: stepArguments.context!.transactionId,
           context: stepArguments.context!,
         }