Go语言AWS Lambda初始化全局状态：用func init()还是func main()？

结论先行 #

你提出的把初始化逻辑从init()迁移到main()的方案完全可行，没有额外的严重副作用，反而更利于单元测试，是Go语言开发Lambda的常用最佳实践之一。

核心原理 #

Lambda Go运行时的冷启动逻辑是：完整启动你的Go可执行程序，依次执行所有包的init()函数后进入main()函数，直到你调用lambda.Start()之后，程序才会进入事件循环，开始监听并处理后续的Lambda触发请求。

所以你放在lambda.Start()之前的所有main()内代码，和init()函数的执行时机、执行次数完全一致：仅冷启动时执行一次，后续容器热复用处理请求时不会重复执行，全局变量跨调用共享的效果没有任何差异。

两种方案的差异对比 #

仅有的微小注意事项 #

• 所有初始化逻辑必须放在lambda.Start()调用之前，否则不会生效
• 初始化出错时的退出行为和init()完全一致：冷启动直接失败，Lambda返回初始化错误，不会进入请求处理逻辑

可选优化建议 #

如果想要进一步优化可测试性，还可以放弃全局变量，改用闭包注入依赖，示例如下：

func newLambdaHandler(myObjects []*s3.Object) func() (int, error) {
    invokeCount := 0
    return func() (int, error) {
        invokeCount = invokeCount + 1
        log.Print(myObjects)
        return invokeCount, nil
    }
}

func main() {
    // 初始化逻辑不变
    svc := s3.New(session.New())
    input := &s3.ListObjectsV2Input{
        Bucket: aws.String("examplebucket"),
    }
    result, _ := svc.ListObjectsV2(input)
    handler := newLambdaHandler(result.Contents)
    lambda.Start(handler)
}

这种写法连全局变量都不需要，单元测试时可以直接调用newLambdaHandler传入测试数据得到handler函数，完全不依赖初始化逻辑。

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