如何使用CDK创建携带不同环境变量配置的Lambda版本

核心原理 #

Lambda的Version本质是函数当前配置（含代码、环境变量、运行时等）的不可变快照，本身不支持单独修改环境变量。你为同一个Function实例创建的所有版本，都会继承该实例当前的环境变量配置，因此无法通过同一个基础函数的不同版本实现不同环境变量的配置。

方案1：抽象公共配置，创建多个独立Lambda实例 #

适合不同版本的配置（权限、超时、环境变量等）差异较大的场景，所有实例共用同一份代码包，仅自定义部分参数。

// 抽象公共配置的工厂函数
function createDummyLambda(scope: Construct, id: string, customEnv: Record<string, string>) {
  return new Function(scope, `dummy-lambda-${id}`, {
    functionName: `dummy-lambda-${id}`,
    code: Code.fromAsset('../dummy-lambda/src'), // 共用同一份代码资源
    handler: 'index.lambda_handler',
    timeout: Duration.seconds(900),
    runtime: Runtime.PYTHON_3_8,
    role: snapshotRole,
    environment : {
      'region': props.env.region,
      ...customEnv // 合并传入的自定义环境变量
    }
  });
}

// 实例化不同配置的Lambda函数
const ciLambda = createDummyLambda(this, 'ci', {
  'utility': 'CI专用参数',
  'siteUrl': 'https://ci.xxx.com'
});
const testLambda = createDummyLambda(this, 'test', {
  'utility': '测试环境参数',
  'siteUrl': 'https://test.xxx.com'
});
const prodLambda = createDummyLambda(this, 'prod', {
  'utility': '生产环境参数',
  'siteUrl': 'https://prod.xxx.com'
});

// 分别为每个实例创建版本
const ciVersion = new Version(this, 'ci-dummy-ver', {
  lambda: ciLambda,
  description: 'CI环境专用版本'
});
const testVersion = new Version(this, 'test-dummy-ver', {
  lambda: testLambda,
  description: '测试环境专用版本'
});
const prodVersion = new Version(this, 'prod-dummy-ver', {
  lambda: prodLambda,
  description: '生产环境专用版本'
});

方案2：使用Lambda别名覆盖环境变量 #

适合代码完全一致，仅需要区分少量环境变量的场景，仅需维护一份函数实例，通过别名实现不同环境变量的覆盖。

// 创建基础Lambda函数，仅配置公共参数
const dummyLambda = new Function(this, 'dummy-lambda', {
  functionName: 'dummy-lambda',
  code: Code.fromAsset('../dummy-lambda/src'),
  handler: 'index.lambda_handler',
  timeout: Duration.seconds(900),
  runtime: Runtime.PYTHON_3_8,
  role: snapshotRole,
  environment : {
    'region': props.env.region // 公共环境变量
  }
});

// 先创建基础版本
const baseVersion = new Version(this, 'base-dummy-ver', {
  lambda: dummyLambda,
  description: '通用基础版本'
});

// 创建CI别名，覆盖自定义环境变量
const ciAlias = new Alias(this, 'ci-alias', {
  aliasName: 'ci',
  version: baseVersion,
  environment: {
    'utility': 'CI专用参数',
    'siteUrl': 'https://ci.xxx.com'
  }
});

// 创建测试环境别名，覆盖不同的环境变量
const testAlias = new Alias(this, 'test-alias', {
  aliasName: 'test',
  version: baseVersion,
  environment: {
    'utility': '测试环境参数',
    'siteUrl': 'https://test.xxx.com'
  }
});

// 创建生产环境别名
const prodAlias = new Alias(this, 'prod-alias', {
  aliasName: 'prod',
  version: baseVersion,
  environment: {
    'utility': '生产环境参数',
    'siteUrl': 'https://prod.xxx.com'
  }
});

后续你直接调用对应别名的ARN即可触发对应配置的Lambda执行。

内容的提问来源于stack exchange，提问作者Deepanshu Kalra