如何在 AWS Lambda 中创建身份验证中间件

Question

我正在使用 AWS Cognito 对我的用户进行身份验证，一旦通过身份验证，他们就可以调用我的 API (API Gateway + Lambda)。我正在使用无服务器框架完成所有这些工作。一旦通过身份验证，当他们调用需要此身份验证的端点时，我的 lambda 将通过request.RequestContext.Authorizer["claims"]. 我有创建一个身份验证中间件以将当前用户注入上下文的想法。但我确定我做错了什么（或者可以改进）。怎么运行的：我的 lambda.go:package mainimport (    "context"    "github.com/aws/aws-lambda-go/events"    "github.com/aws/aws-lambda-go/lambda"    "github.com/company/api/middlewares")func Handler(ctx context.Context, request events.APIGatewayProxyRequest) (events.APIGatewayProxyResponse, error) {    fmt.Println(ctx.user)    return events.APIGatewayProxyResponse{}, nil}func main() {    lambda.Start(        middlewares.Authentication(Handler),    )}中间件/authentication.gopackage middlewaresimport (    "context"    "github.com/aws/aws-lambda-go/events"    "github.com/company/api/models")func Authentication(next func(context.Context, events.APIGatewayProxyRequest) (events.APIGatewayProxyResponse, error)) func(context.Context, events.APIGatewayProxyRequest) (events.APIGatewayProxyResponse, error) {    var user models.User    return func(ctx context.Context, request events.APIGatewayProxyRequest) (events.APIGatewayProxyResponse, error) {        claims := request.RequestContext.Authorizer["claims"]        // Find user by claims properties.        if err := user.Current(claims); err != nil {            return events.APIGatewayProxyResponse{}, err        }        ctx.user = user        return next(ctx, request)    }}型号/user.go：package modelsimport (    "github.com/jinzhu/gorm"    "github.com/mitchellh/mapstructure")type User struct {    gorm.Model    // Override ID cause we are using cognito.    Email string `gorm:"primary_key,not null"`    Site  Site}我有两个问题：这是定义接收函数并返回另一个函数的函数（身份验证函数）的正确方法吗？因为它太冗长了，我觉得这是错误的。有没有办法增加ctx一个user属性？我正在尝试的方式，我看到了错误ctx.user undefined (type context.Context has no field or method user)。

Answer 1

关于使用中间件的第一个问题：

这种方法当然没有错。如果您定义函数类型并使用定义的名称，函数看起来可能会好一些。net/http做同样的事情HandlerFunc：

type HandlerFunc func(ResponseWriter, *Request)

这将使中间件的签名更加合理：

func AuthMiddleware(nextHop HandlerFunc) HandlerFunc

编辑：lambda 库没有为函数签名定义这样的类型吗？我希望有一个存在。

另外我不知道后缀Middleware在你的情况下是否有意义，但我认为一些后缀应该对你有意义，以便为函数名称提供更多上下文并使其更易于理解。AuthenticationMiddleware可能是一个例子。

编辑：刚刚看到包名。LGTM 真的。

第二个问题：

还有一个常见的陷阱：context.WithValue返回要使用的新上下文。因此，您不应期望传递的参数上下文发生变化，而应使用返回的新参数上下文。