如何通过AWS CDK（Python）向Lambda Layer传递变量？

Lambda Layer本身确实不支持直接设置环境变量，不过可以通过以下两种常用方法实现需求：

方法1：构建时生成配置文件 #

在CDK打包Layer代码前，动态生成包含目标变量的配置文件，Layer中的代码直接读取该文件即可获取变量值。

CDK代码示例 #

from aws_cdk import (
    aws_lambda as _lambda,
    Stack,
    AssetCode,
)
from constructs import Construct
import os
import pathlib

class CdkAppStack(Stack):
    def __init__(self, scope: Construct, construct_id: str, **kwargs) -> None:
        super().__init__(scope, construct_id, **kwargs)

        MY_VAR = 'foobar'
        # 生成配置文件到Layer代码目录
        layer_config_path = pathlib.Path('../my_lib/config.py')
        with open(layer_config_path, 'w') as f:
            f.write(f"MY_VAR = '{MY_VAR}'\n")
        
        my_lambda_layer = _lambda.LayerVersion(self, 'MyLambdaLayer',
                                               code=AssetCode(os.path.join('..', 'my_lib')),
                                               )

Layer代码使用示例 #

在../my_lib的模块中直接导入配置文件读取变量：

from config import MY_VAR

def some_logic():
    print(f"获取到的变量值：{MY_VAR}")

注意：如果使用版本控制，建议将自动生成的config.py加入忽略列表，避免本地文件冲突。

方法2：通过关联的Lambda函数传递环境变量 #

Lambda Layer会共享其附加的Lambda函数的执行环境，因此可以在Lambda函数中设置环境变量，Layer中的代码直接读取函数的环境变量即可。

CDK代码示例 #

from aws_cdk import (
    aws_lambda as _lambda,
    Stack,
    AssetCode,
)
from constructs import Construct
import os

class CdkAppStack(Stack):
    def __init__(self, scope: Construct, construct_id: str, **kwargs) -> None:
        super().__init__(scope, construct_id, **kwargs)

        MY_VAR = 'foobar'
        # 正常定义Lambda Layer
        my_lambda_layer = _lambda.LayerVersion(self, 'MyLambdaLayer',
                                               code=AssetCode(os.path.join('..', 'my_lib')),
                                               )
        # 创建关联Layer的Lambda函数，设置环境变量
        lambda_function = _lambda.Function(self, "MyLambdaFunction",
                                           runtime=_lambda.Runtime.PYTHON_3_11,
                                           handler="handler.lambda_handler",
                                           code=_lambda.Code.from_asset("lambda"),
                                           layers=[my_lambda_layer],
                                           environment={
                                               'MY_VAR': MY_VAR,
                                           },
                                           )

Layer代码使用示例 #

在../my_lib的模块中通过os.environ读取变量：

import os

def get_my_var():
    # 可设置默认值避免变量不存在时报错
    return os.environ.get('MY_VAR', 'default_value')

方法对比 #

• 方法1适合变量是Layer自身的固定配置，与Lambda函数无关的场景，变量值在构建时确定。
• 方法2适合变量与Lambda函数绑定，不同函数可给同一Layer传递不同值的场景，灵活性更高。

内容的提问来源于stack exchange，提问作者P D