amazon-web-services - 在 Go 中为 AWS Lambda 指定多个事件处理程序

Question

通常，Go 中的 AWS Lambda 事件处理程序代码(使用无服务器框架)编码为:

package main

import (
  "fmt"
  "context"
  "github.com/aws/aws-lambda-go/lambda"
)

type MyEvent struct {
  Name string `json:"name"`
}

func HandleRequest(ctx context.Context, name MyEvent) (string, error) {
  return fmt.Sprintf("Hello %s!", name.Name ), nil
}

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

serverless.yml 文件包含如下部分:

  skeleton-go-get:
    name: skeleton-go-get
    runtime: go1.x
    handler: go-handler  # <- This specifies a file, not a function.
    events:
      - http:
          path: skeleton/go
          method: get

^ 创建一个请求处理程序...但现在我希望我的一个 Go 脚本/程序包含 HTTP GET 和 POST 请求的事件处理程序，而不是每个无服务器函数使用一个 Go 程序文件。

这在 Node.js、Ruby、Python 等语言中确实是可能的，通过 serverless.yml 指定处理程序文件中的哪个函数将用于哪个无服务器函数。例如(对于 Python 函数):

[...]
functions:
  skeleton-python-get:
    name: skeleton-python-get
    handler: python-handler.handle_get  # <- Specifies the HTTP GET handler.
    events:
      - http:
          path: skeleton/python
          method: get
  skeleton-python-post:
    name: skeleton-python-post
    handler: python-handler.handle_post  # <- Specifies the HTTP POST handler.
    events:
      - http:
          path: skeleton/python
          method: post

我无法让同样的技巧适用于 Go。我尝试在 main() 中包含正确的请求，但无济于事:

func HandleGetRequest(ctx context.Context, name MyEvent) (string, error) {
  return fmt.Sprintf("Hello %s!", name.Name ), nil
}

func HandlePostRequest(ctx context.Context, name MyEvent) (string, error) {
  return fmt.Sprintf("Hello %s!", name.Name ), nil
}

func main() {
  lambda.Start(HandleGetRequest)
  lambda.Start(HandlePostRequest)  // <- Attempt to add another handler.
}

并且在 serverless.yml 文件中为 Go 处理程序指定多个事件处理函数也不起作用:该函数不是处理程序声明的有效部分。

  skeleton-go-get:
    name: skeleton-go-get
    runtime: go1.x
    handler: go-handler.HandleGet  # <- Attempt to specify a function.
    events:
      - http:
          path: skeleton/go
          method: get

  skeleton-go-post:
    name: skeleton-go-post
    runtime: go1.x
    handler: go-handler.HandlePost  # <- Attempt to specify a function.
    events:
      - http:
          path: skeleton/go
          method: post

问:如何在一个 Go 程序中包含多个 AWS Lambda 事件处理程序(使用无服务器框架)？

Answer 1

您可以对 get 和 post 使用相同的函数(和处理程序):

skeleton-go:
    name: skeleton-go
    runtime: go1.x
    handler: go-handler
    events:
      - http:
          path: skeleton/go
          method: get
      - http:
          path: skeleton/go
          method: post

使用 Go 的 built-in HTTP router或者使用第三方的，例如Gorilla Mux或Chi如下面的示例代码所示(因为那是我方便的)。本质上，您正在构建一个 Go HTTP 服务器，但是是在 Lambda 中。因此，请按照设置 Go Web 服务器的详细信息进行操作，并查看 AWS's API Gateway Proxy .

package main

import (
    "context"
    "net/http"

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

    "github.com/go-chi/chi"
    chiproxy "github.com/awslabs/aws-lambda-go-api-proxy/chi"
)

var adapter *chiproxy.ChiLambda

func GetSkeleton(w http.ResponseWriter, r *http.Request) {
    ...
}

func PostSkeletonToMom(w http.ResponseWriter, r *http.Request) {
    ...
}

func init() {
    r := chi.NewRouter()

    r.Get("/skeleton/go", GetSkeleton)
    r.Post("/skeleton/go", PostSkeletonToMom)

    adapter = chiproxy.New(r)
}

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

    return c, err
}

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