如何在Go语言编写的AWS Lambda函数中获取Alias别名？

核心结论 #

你不可能在init()函数中拿到lambdaHandler里的req对象：

• init()是Go程序初始化阶段就会执行的逻辑，执行时机远早于main函数启动、Lambda运行时接收到用户请求的阶段，此时还没有任何请求产生，自然拿不到只有请求触发时才会生成的req参数。

两种实现方案 #

你要运行与Lambda Alias相关的初始化逻辑，可以根据你的使用场景选下面的方案： #

方案1：第一次请求时懒加载初始化（无额外配置，适配所有场景）

Lambda每次冷启动后，只会处理同一个别名的请求，你可以用sync.Once保证初始化逻辑只在第一次请求时执行一次，同时可以拿到请求或上下文中的Alias信息：

import (
	"context"
	"strings"
	"sync"

	"github.com/aws/aws-lambda-go/events"
	"github.com/aws/aws-lambda-go/lambdacontext"
)

var (
	aliasInitOnce sync.Once
	// 你需要初始化的和Alias相关的全局资源
)

// 你的Alias专属初始化逻辑
func initAliasRelated(alias string) {
	// 这里放你原来想放在init里的代码
}

func main() {
	adapter = chiproxy.New(r)
	lambda.Start(lambdaHandler)
}

func lambdaHandler(ctx context.Context, req events.APIGatewayProxyRequest) (events.APIGatewayProxyResponse, error) {
	// 初始化逻辑仅执行一次
	aliasInitOnce.Do(func() {
		// 方式A：从Lambda上下文的ARN解析别名，不需要依赖API网关的req
		lc, _ := lambdacontext.FromContext(ctx)
		arnParts := strings.Split(lc.InvokedFunctionArn, ":")
		if len(arnParts) >= 8 {
			alias := arnParts[7]
			initAliasRelated(alias)
		}

		// 方式B：如果你需要的信息在req里，直接从req中读取即可
		// yourField := req.XXX
		// initAliasRelated(yourField)
	})

	c, err := adapter.ProxyWithContext(ctx, req)
	return c, err
}

方案2：环境变量注入（性能最优，适配别名固定的场景）

如果你的Lambda别名是部署时就确定的固定值，你可以直接在Lambda控制台/部署配置中新增环境变量LAMBDA_ALIAS，值设为你的别名，此时你就可以直接在init()中获取并执行初始化逻辑：

import "os"

func init() {
	alias := os.Getenv("LAMBDA_ALIAS")
	// 在这里执行你的Alias专属初始化逻辑
}

func main() {
	adapter = chiproxy.New(r)
	lambda.Start(lambdaHandler)
}

func lambdaHandler(ctx context.Context, req events.APIGatewayProxyRequest) (events.APIGatewayProxyResponse, error) {
	c, err := adapter.ProxyWithContext(ctx, req)
	return c, err
}

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