如何通过AWS CLI以编程方式判断Lambda是否在本地运行

无需通过AWS CLI实现该需求，直接使用Lambda运行时自动注入的专属环境变量即可区分环境，不需要添加任何额外请求参数，性能和可靠性都优于CLI调用方案。

判断依据 #

AWS云环境下运行的Lambda会自动注入多个独有的环境变量，本地模拟环境（SAM CLI、LocalStack等）不会完整生成符合规则的变量，常用判断条件如下：

• AWS_LAMBDA_FUNCTION_NAME：云环境下会自动填充部署时定义的Lambda函数名称，本地模拟环境通常不存在该变量，或为自定义测试值
• AWS_EXECUTION_ENV：云环境下该值固定以AWS_Lambda_为前缀，后缀对应使用的运行时版本，比如AWS_Lambda_java11、AWS_Lambda_python3.10，本地环境一般不会设置该变量或值不符合该格式规则

代码实现示例 #

Python 示例 #

import os

def is_cloud_lambda_env() -> bool:
    return (
        "AWS_LAMBDA_FUNCTION_NAME" in os.environ
        and os.environ.get("AWS_EXECUTION_ENV", "").startswith("AWS_Lambda_")
    )

def lambda_handler(event, context):
    if is_cloud_lambda_env():
        # 仅AWS云环境执行的业务逻辑
        pass
    # 所有环境通用的业务逻辑
    return {}

Node.js 示例 #

const isCloudLambdaEnv = () => {
  return process.env.AWS_LAMBDA_FUNCTION_NAME !== undefined
    && process.env.AWS_EXECUTION_ENV?.startsWith("AWS_Lambda_")
}

exports.handler = async (event) => {
  if (isCloudLambdaEnv()) {
    // 仅AWS云环境执行的业务逻辑
  }
  // 所有环境通用的业务逻辑
  return { statusCode: 200 }
}

不推荐使用AWS CLI判断的原因 #

• CLI调用属于网络IO操作，会额外增加Lambda执行耗时，还可能受网络波动影响可靠性
• 如果本地开发环境配置了AWS访问凭证，CLI调用也可以正常返回结果，无法准确区分实际运行环境
• 环境变量判断是无依赖的本地同步操作，执行耗时可忽略，可靠性更高

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