BaseContext.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.logging.DurableLogger;

public interface BaseContext {
    ThreadLocal<BaseContext> CONTEXT = new ThreadLocal<>();

    /**
     * Gets the current context (DurableContext or StepContext) for this thread.
     *
     * @return the current context or null if not set
     */
    static BaseContext getCurrentContext() {
        return CONTEXT.get();
    }
    /**
     * Gets a logger with additional information of the current execution context.
     *
     * @return a DurableLogger instance
     */
    DurableLogger getLogger();

    /**
     * Gets a logger with additional information of the current execution context.
     *
     * @param delegate the logger to wrap
     * @return a DurableLogger instance
     */
    DurableLogger getLogger(Logger delegate);

    /**
     * Returns the AWS Lambda runtime context.
     *
     * @return the Lambda context
     */
    Context getLambdaContext();

    /**
     * Returns the current durable execution arn
     *
     * @return the execution arn
     */
    String getExecutionArn();

    /**
     * Returns the configuration for durable execution behavior.
     *
     * @return the durable configuration
     */
    DurableConfig getDurableConfig();

    /**
     * Gets the context ID for this context. Null for root context, operationId of the context operation for child
     * contexts.
     */
    String getContextId();

    /** Gets the context name for this context. Null for root context. */
    String getContextName();
}