[feature] add an optional stepContext parameter

[zhongkechen]

By submitting this pull request, I confirm that my contribution is made under the terms of the Apache 2.0 license.

Issue Link, if available

Add StepContext #80

Description

Add an optional stepContext to the user provided step function, which allows users to get the logger and additional execution context from it. Now any logger from whichever context behaves exactly the same.

        var greeting = context.step("create-greeting", String.class, ctx -> {
            ctx.getLogger().info("Creating greeting message");
            return "Hello, " + input.getName();
        });

equivalent to

        var greeting = context.step("create-greeting", String.class, () -> {
            context.getLogger().info("Creating greeting message");
            return "Hello, " + input.getName();
        });

Demo/Screenshots

step logger

{
    "@timestamp": "2026-02-27T00:25:46.759Z",
    "ecs.version": "1.2.0",
    "log.level": "INFO",
    "message": "Transforming greeting to uppercase",
    "process.thread.name": "durable-exec-16",
    "log.logger": "software.amazon.lambda.durable.DurableContext",
    "attempt": "0",
    "durableExecutionArn": "arn:aws:lambda:us-east-2:961306377283:function:logging-example:$LATEST/durable-execution/6a56c3ef-4a07-4ea1-8500-82bcfe388859/cb7be3f9-3b5a-31bc-8797-9ba38d44055a",
    "operationId": "2",
    "operationName": "transform",
    "requestId": "da78b44c-d811-4431-ad27-1280d6d1c2c8"
}

execution logger:

{
    "@timestamp": "2026-02-27T00:25:46.846Z",
    "ecs.version": "1.2.0",
    "log.level": "INFO",
    "message": "Completed processing, result: HELLO, WORLD!",
    "process.thread.name": "durable-exec-15",
    "log.logger": "software.amazon.lambda.durable.DurableContext",
    "durableExecutionArn": "arn:aws:lambda:us-east-2:961306377283:function:logging-example:$LATEST/durable-execution/6a56c3ef-4a07-4ea1-8500-82bcfe388859/cb7be3f9-3b5a-31bc-8797-9ba38d44055a",
    "requestId": "da78b44c-d811-4431-ad27-1280d6d1c2c8"
}

• Updated operations to use DurableContext parameter
• Made root context id consistent ("Root" -> null)

Checklist

• I have filled out every section of the PR template
• I have thoroughly tested this change

Testing

Unit Tests

Have unit tests been written for these changes? Updated

Integration Tests

Have integration tests been written for these changes? Updated

Examples

Has a new example been added for the change? (if applicable) Updated

[zhongkechen]

Even though the new parameter is optional, it's still breaking some edge cases like this.

[phipag]

Nice work!