如何使用CDK Aspects获取作用域内Lambda函数的所有环境变量

处理L2 Lambda Function构造 #

Lambda的L2构造lambda.Function属于高层抽象，没有直接暴露environment属性，需要通过其底层生成的L1构造CfnFunction来获取环境变量：

class LambdaEnvAspect implements IAspect {
  visit(node: IConstruct): void {
    // 匹配L2 Lambda Function
    if (node instanceof lambda.Function) {
      // 获取对应的L1 CloudFormation资源
      const cfnLambda = node.node.defaultChild as lambda.CfnFunction;
      // 检查是否存在环境变量
      if (cfnLambda.environment?.variables) {
        console.log(`Lambda ${node.node.id} 的环境变量：`, cfnLambda.environment.variables);
      }
    }
  }
}

处理L1 CfnFunction构造 #

对于直接使用L1构造lambda.CfnFunction的情况，environment属性是CfnEnvironmentProperty类型，其中的variables字段就是键值对形式的环境变量对象：

class LambdaEnvAspect implements IAspect {
  visit(node: IConstruct): void {
    // 匹配L1 CfnFunction
    if (node instanceof lambda.CfnFunction) {
      if (node.environment?.variables) {
        console.log(`L1 Lambda ${node.node.id} 的环境变量：`, node.environment.variables);
      }
    }
  }
}

合并两种情况的完整Aspect #

可以把两种判断逻辑合并，一次性处理所有Lambda资源：

class LambdaEnvAspect implements IAspect {
  visit(node: IConstruct): void {
    let cfnLambda: lambda.CfnFunction | undefined;

    // 处理L2 Lambda Function
    if (node instanceof lambda.Function) {
      cfnLambda = node.node.defaultChild as lambda.CfnFunction;
    }
    // 处理L1 CfnFunction
    else if (node instanceof lambda.CfnFunction) {
      cfnLambda = node;
    }

    // 输出环境变量
    if (cfnLambda?.environment?.variables) {
      console.log(`Lambda ${cfnLambda.node.id} 的环境变量：`, cfnLambda.environment.variables);
    }
  }
}

注意事项 #

• 如果Lambda未配置任何环境变量，environment属性会是undefined，建议使用可选链操作符?.避免运行时错误。
• 确保使用稳定版CDK，不同版本的L2构造在defaultChild返回值上可能存在细微差异。

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