如何在Serverless中使用if-else？多阶段Lambda环境配置咨询

一、在Serverless中使用if-else语句 #

1. 配置文件（serverless.yml）中的条件判断 #

Serverless Framework支持在配置文件中通过内置的cond语法实现条件分支，效果类似if-else：

• 基础格式：${cond:逻辑表达式, 满足时的值, 不满足时的值}
• 示例：根据阶段配置Lambda内存与环境变量

custom:
  stage: ${opt:stage, 'dev'} # 默认使用dev阶段

functions:
  myService:
    handler: src/index.main
    memorySize: ${cond:${self:custom.stage} == 'prod', 2048, 1024} # 生产环境分配2048M内存，其他环境用1024M
    environment:
      DB_CONN: ${cond:${self:custom.stage} == 'prod', 'prod-db-addr', 'dev-db-addr'}

复杂多分支可嵌套cond表达式：

environment:
  LOG_LEVEL: ${cond:${self:custom.stage} == 'prod', 'error', ${cond:${self:custom.stage} == 'staging', 'warn', 'debug'}}

2. Lambda函数代码中的if-else #

在Lambda业务代码里，直接用对应编程语言的if-else语法即可，以Node.js为例：

exports.main = async (event) => {
  const currentStage = process.env.STAGE;
  let result;

  if (currentStage === 'prod') {
    result = { statusCode: 200, body: '生产环境响应' };
  } else if (currentStage === 'staging') {
    result = { statusCode: 200, body: '预发布环境响应' };
  } else {
    result = { statusCode: 200, body: '开发环境响应' };
  }

  return result;
};

二、多阶段环境配置方案 #

1. 基础阶段指定 #

部署时通过--stage参数直接指定目标环境：

# 部署到生产环境
serverless deploy --stage prod
# 部署到预发布环境
serverless deploy --stage staging

2. 拆分环境配置文件 #

将不同环境的配置拆分到独立文件，比如config.dev.yml、config.prod.yml，然后在serverless.yml中引入：

custom:
  stage: ${opt:stage, 'dev'}
  envConfig: ${file(./config.${self:custom.stage}.yml)}

functions:
  myService:
    handler: src/index.main
    environment: ${self:custom.envConfig.environment}

config.dev.yml示例：

environment:
  DB_CONN: 'dev-db.example.com'
  LOG_LEVEL: 'debug'

config.prod.yml示例：

environment:
  DB_CONN: 'prod-db.example.com'
  LOG_LEVEL: 'error'

3. 环境变量动态覆盖 #

部署时通过系统环境变量传递配置参数，灵活覆盖默认值：

STAGE=prod DB_CONN=custom-prod-db serverless deploy

在serverless.yml中读取这些变量：

environment:
  STAGE: ${env:STAGE, 'dev'}
  DB_CONN: ${env:DB_CONN, 'dev-db.example.com'}

4. 阶段特定资源配置 #

如果不同阶段需要创建差异化资源（如不同的S3桶），可通过Conditions实现：

resources:
  Resources:
    DevStorageBucket:
      Type: AWS::S3::Bucket
      Condition: IsDevStage
      Properties:
        BucketName: dev-app-storage
    ProdStorageBucket:
      Type: AWS::S3::Bucket
      Condition: IsProdStage
      Properties:
        BucketName: prod-app-storage

Conditions:
  IsDevStage:
    Fn::Equals: [${self:custom.stage}, 'dev']
  IsProdStage:
    Fn::Equals: [${self:custom.stage}, 'prod']

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