BaseContextImpl.java

// Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
// SPDX-License-Identifier: Apache-2.0
package software.amazon.lambda.durable.context;

import com.amazonaws.services.lambda.runtime.Context;
import org.slf4j.Logger;
import software.amazon.lambda.durable.DurableConfig;
import software.amazon.lambda.durable.execution.ExecutionManager;
import software.amazon.lambda.durable.execution.ThreadType;
import software.amazon.lambda.durable.logging.DurableLogger;

public abstract class BaseContextImpl implements BaseContext {
    private final ExecutionManager executionManager;
    private final DurableConfig durableConfig;
    private final Context lambdaContext;
    private final String contextId;
    private final String contextName;
    private final ThreadType threadType;

    /**
     * Creates a new BaseContext instance.
     *
     * @param executionManager the execution manager for thread coordination and state management
     * @param durableConfig the durable execution configuration
     * @param lambdaContext the AWS Lambda runtime context
     * @param contextId the context ID, null for root context, set for child contexts
     * @param contextName the human-readable name for this context
     * @param threadType the type of thread this context runs on
     */
    protected BaseContextImpl(
            ExecutionManager executionManager,
            DurableConfig durableConfig,
            Context lambdaContext,
            String contextId,
            String contextName,
            ThreadType threadType) {
        this.executionManager = executionManager;
        this.durableConfig = durableConfig;
        this.lambdaContext = lambdaContext;
        this.contextId = contextId;
        this.contextName = contextName;
        this.threadType = threadType;
    }

    // =============== accessors ================

    /**
     * Returns the AWS Lambda runtime context.
     *
     * @return the Lambda context
     */
    @Override
    public Context getLambdaContext() {
        return lambdaContext;
    }

    /**
     * Returns metadata about the current durable execution.
     *
     * <p>The execution context provides information that remains constant throughout the execution lifecycle, such as
     * the durable execution ARN. This is useful for tracking execution progress, correlating logs, and referencing this
     * execution in external systems.
     *
     * @return the execution context
     */
    @Override
    public String getExecutionArn() {
        return executionManager.getDurableExecutionArn();
    }

    /**
     * Returns the configuration for durable execution behavior.
     *
     * @return the durable configuration
     */
    @Override
    public DurableConfig getDurableConfig() {
        return durableConfig;
    }

    // ============= internal utilities ===============

    /** Gets the context ID for this context. Null for root context, set for child contexts. */
    @Override
    public String getContextId() {
        return contextId;
    }

    /** Gets the context name for this context. Null for root context. */
    @Override
    public String getContextName() {
        return contextName;
    }

    public ExecutionManager getExecutionManager() {
        return executionManager;
    }

    /** Returns a durable logger for this context. */
    public DurableLogger getLogger() {
        return DurableLogger.INSTANCE;
    }

    /** Returns a durable logger for this context. */
    public DurableLogger getLogger(Logger delegate) {
        return new DurableLogger(delegate);
    }

    public static void setCurrentContext(BaseContext context) {
        CONTEXT.set(context);
    }
}