step functions: associate_with_parent broken with jsonata

[nywhere]

Describe the bug

The current value is still:

"AWS_STEP_FUNCTIONS_STARTED_BY_EXECUTION_ID": "$$.Execution.Id"

It should be: "{% $states.context.Execution.Id %}"

Regression Issue

• Select this option if this issue appears to be a regression.

Last Known Working CDK Version

No response

Expected Behavior

associate_with_parent should work

Current Behavior

associate_with_parent does not work

Reproduction Steps

sfn_tasks.StepFunctionsStartExecution(
    ...
    query_language=sfn.QueryLanguage.JSONATA,
    associate_with_parent=True,
    ...
)

Possible Solution

No response

Additional Information/Context

No response

CDK CLI Version

2.1005.0

Framework Version

No response

Node.js Version

v18.16.0

OS

Mac 15.3.2

Language

Python

Language Version

3.13

Other information

No response

[pahud]

Current/Expected Behavior

Code Analysis

Looking at the source code in packages/aws-cdk-lib/aws-stepfunctions-tasks/lib/stepfunctions/start-execution.ts, I found the issue in the _renderTask method:

aws-cdk/packages/aws-cdk-lib/aws-stepfunctions-tasks/lib/stepfunctions/start-execution.ts

Lines 123 to 131 in 38d82c4

	let input: any;
	if (this.props.associateWithParent) {
	const associateWithParentEntry = {
	AWS_STEP_FUNCTIONS_STARTED_BY_EXECUTION_ID: sfn.JsonPath.stringAt('$$.Execution.Id'),
	};
	input = this.props.input ? { ...this.props.input.value, ...associateWithParentEntry } : associateWithParentEntry;
	} else {
	input = this.props.input ? this.props.input.value: sfn.TaskInput.fromJsonPathAt('$').value;
	}

The implementation always uses JSON Path format (sfn.JsonPath.stringAt('$$.Execution.Id')) regardless of the query language set. It should check the query language and use the appropriate format:

For JSON Path: sfn.JsonPath.stringAt('$$.Execution.Id')
For JSONata: A reference to {% $states.context.Execution.Id %}

Solution

To fix this issue, the implementation needs to be modified to check the query language and set the appropriate syntax for the execution ID reference. The function should use the appropriate format based on the query language being used.

For JSONata, it should use:

AWS_STEP_FUNCTIONS_STARTED_BY_EXECUTION_ID: "{% $states.context.Execution.Id %}"

Instead of:

AWS_STEP_FUNCTIONS_STARTED_BY_EXECUTION_ID: "$$.Execution.Id"