Golang AWS Lambda中spf13/viper找不到配置文件的问题求助

问题根源 #

AWS Lambda（包括SAM本地运行环境）的工作目录固定为/var/task，你的现有流程存在两个核心问题：

1. 构建命令仅编译生成二进制文件到./bin目录，但未将config/application.yaml复制到部署包中；
2. 代码中使用的../config/相对路径在Lambda环境中无效——Lambda运行时不存在上层目录结构。

解决方案 #

1. 调整构建流程，确保配置文件随二进制文件打包 #

修改构建步骤，编译后将配置文件复制到二进制文件所在的bin目录：

# 编译Lambda二进制文件
GOARCH=amd64 GOOS=linux go build -mod=vendor -o ./bin/<lambda-function-name> -ldflags="-s -w" ./lambda/main.go
# 复制配置文件到bin目录，与二进制文件同路径
cp ./config/application.yaml ./bin/

同时修改Viper的配置路径，移除无效的上层目录引用：

v := viper.New()

v.SetConfigName("application")
v.SetConfigType("yaml")
v.AddConfigPath(".") // 现在配置文件与二进制同目录，此路径可直接读取
v.AutomaticEnv()

err := v.ReadInConfig()
if err != nil {
    fmt.Println("fatal error config file: default \n", err)
    os.Exit(1)
}

2. 更新SAM模板，确保配置文件被包含 #

在deployment/template.yml的Lambda函数定义中，将CodeUri指向包含二进制和配置文件的bin目录，确保SAM打包时将配置文件纳入部署包：

Resources:
  YourLambdaFunction:
    Type: AWS::Serverless::Function
    Properties:
      CodeUri: ./bin/  # 指向包含二进制与配置文件的目录
      Handler: <lambda-function-name>
      Runtime: go1.x
      # 其他Lambda配置项...

3. 优化Viper初始化逻辑（可选） #

避免在每次请求的Handler中重复初始化Viper，将初始化逻辑放在全局init函数中，提升性能：

var v *viper.Viper

func init() {
    v = viper.New()
    v.SetConfigName("application")
    v.SetConfigType("yaml")
    v.AddConfigPath(".")
    v.AutomaticEnv()
    
    err := v.ReadInConfig()
    if err != nil {
        fmt.Println("fatal error config file: default \n", err)
        os.Exit(1)
    }
}

func main() {
    lambda.Start(Handler)
}

func Handler(request events.APIGatewayProxyRequest) (*events.APIGatewayProxyResponse, error) {
    env := v.GetString("app.env")
    fmt.Println("Environment : ", env)
    return handlers.Join(request)
}

4. 本地测试验证 #

重新执行构建、打包流程后，运行SAM本地命令测试：

sam local start-api -t deployment/template.yml

此时Viper即可正常读取application.yaml配置文件。

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