So erstellen Sie einen generischen Typ für Lambda-Middleware in Go

Frageninhalt

Ich verwende go in aws lambda und suche nach einer gemeinsamen Middleware-Lösung. Ich habe den folgenden Code:

func wshandler(ctx context.context, event events.apigatewaywebsocketproxyrequest) (events.apigatewayproxyresponse, error) {
}

type handlerfunc func(context.context, events.apigatewaywebsocketproxyrequest) (events.apigatewayproxyresponse, error)

func logmiddleware(next handlerfunc) handlerfunc {
    return handlerfunc(func(ctx context.context, event events.apigatewaywebsocketproxyrequest) (events.apigatewayproxyresponse, error) {

        return next(ctx, event)
    })
}

lambda.start(logmiddleware(wshandler))

Die Middleware-Funktion verfügt über einen Parameter events.apigatewaywebsocketproxyrequest 因为目标处理程序 wshandler, der diesen Typ verwendet.

Ich habe einen anderen Handler, der einen Parameter entgegennimmt event events.apigatewayproxyrequest wie unten gezeigt. Diese Middleware kann nicht verwendet werden, da die Parameter nicht übereinstimmen.

graphqlquerymutationhandler(ctx context.context, event events.apigatewayproxyrequest){
...
}

Ich habe versucht, das Middleware-Handle auf interface{} zu ändern, aber es hat nicht funktioniert. Go beschwert sich über diesen Typ.

type HandlerFunc func(context.Context, interface{}) (interface{}, error)

Gibt es eine Möglichkeit, Middleware für jeden Handlertyp funktionieren zu lassen?

Richtige Antwort

Lassen Sie mich eine funktionierende Lösung vorstellen, die ich auf meinem System replizieren konnte. Lassen Sie mich zunächst das von mir verwendete Projektlayout mit Ihnen teilen:

events/
  http_event.json
  sqs_event.json
hello-world/
  main.go
sqs/
  main.go
middlewares/
  middlewares.go

Konzentrieren wir uns nun auf den Code.

middlewares/middlewares.go

Der Code lautet wie folgt:

package middlewares

import (
    "context"
    "fmt"

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

type record struct {
    events.apigatewayproxyrequest `json:",omitempty"`
    events.sqsevent               `json:",omitempty"`
}

type event struct {
    records []record `json:"records"`
}

type handlerfunc func(ctx context.context, event event) (string, error)

func logmiddleware(ctx context.context, next handlerfunc) handlerfunc {
    return handlerfunc(func(ctx context.context, event event) (string, error) {
        fmt.println("log from middleware!")
        return next(ctx, event)
    })
}

Lassen Sie uns die Grundkonzepte zusammenfassen:

• Wir definieren einen Wrapper für die event 结构体，它将成为我们的通用事件。它是 record-Struktur.
• record 结构使用结构嵌入来嵌入我们要处理的所有事件（例如 event.apigatewayproxyrequest 和 sqsevent).
• Wir verlassen uns darauf, dass die Middleware-Signaturen so allgemein wie möglich sind.

events/http_event.json

{
    "records": [
        {
            "body": "{\"message\": \"hello world\"}",
            "resource": "/hello",
            "path": "/hello",
            "httpmethod": "get",
            "isbase64encoded": false,
            "querystringparameters": {
                "foo": "bar"
            },
            "pathparameters": {
                "proxy": "/path/to/resource"
            },
            "stagevariables": {
                "baz": "qux"
            },
            "headers": {
                "accept": "text/html,application/xhtml+xml,application/xml;q=0.9,image/webp,*/*;q=0.8",
                "accept-encoding": "gzip, deflate, sdch",
                "accept-language": "en-us,en;q=0.8",
                "cache-control": "max-age=0",
                "cloudfront-forwarded-proto": "https",
                "cloudfront-is-desktop-viewer": "true",
                "cloudfront-is-mobile-viewer": "false",
                "cloudfront-is-smarttv-viewer": "false",
                "cloudfront-is-tablet-viewer": "false",
                "cloudfront-viewer-country": "us",
                "host": "1234567890.execute-api.us-east-1.amazonaws.com",
                "upgrade-insecure-requests": "1",
                "user-agent": "custom user agent string",
                "via": "1.1 08f323deadbeefa7af34d5feb414ce27.cloudfront.net (cloudfront)",
                "x-amz-cf-id": "cdehvqoznx43vyqb9j2-nvch-9z396uhbp027y2jvkcpnlmgjhqlaa==",
                "x-forwarded-for": "127.0.0.1, 127.0.0.2",
                "x-forwarded-port": "443",
                "x-forwarded-proto": "https"
            },
            "requestcontext": {
                "accountid": "123456789012",
                "resourceid": "123456",
                "stage": "prod",
                "requestid": "c6af9ac6-7b61-11e6-9a41-93e8deadbeef",
                "requesttime": "09/apr/2015:12:34:56 +0000",
                "requesttimeepoch": 1428582896000,
                "identity": {
                    "cognitoidentitypoolid": null,
                    "accountid": null,
                    "cognitoidentityid": null,
                    "caller": null,
                    "accesskey": null,
                    "sourceip": "127.0.0.1",
                    "cognitoauthenticationtype": null,
                    "cognitoauthenticationprovider": null,
                    "userarn": null,
                    "useragent": "custom user agent string",
                    "user": null
                },
                "path": "/prod/hello",
                "resourcepath": "/hello",
                "httpmethod": "get",
                "apiid": "1234567890",
                "protocol": "http/1.1"
            }
        }
    ]
}

Hier gibt es nichts zu sagen.

events/sqs_event.json

{
    "records": [
        {
            "records": [
                {
                    "messageid": "19dd0b57-b21e-4ac1-bd88-01bbb068cb78",
                    "receipthandle": "messagereceipthandle",
                    "body": "my own event payload!",
                    "attributes": {
                        "approximatereceivecount": "1",
                        "senttimestamp": "1523232000000",
                        "senderid": "123456789012",
                        "approximatefirstreceivetimestamp": "1523232000001"
                    },
                    "messageattributes": {},
                    "md5ofbody": "4d1d0024b51659ad8c3725f9ba7e2471",
                    "eventsource": "aws:sqs",
                    "eventsourcearn": "arn:aws:sqs:us-east-1:123456789012:myqueue",
                    "awsregion": "us-east-1"
                }
            ]
        }
    ]
}

Das Gleiche gilt auch hier.

hello-world/main.go

package main

import (
    "context"
    "fmt"

    "httplambda/middlewares"

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

func lambdahandler(ctx context.context, event middlewares.event) (string, error) {
    _ = ctx
    fmt.println("path:", event.records[0].apigatewayproxyrequest.path)

    fmt.println("hi from http-triggered lambda!")

    return "", nil
}

func main() {
    // start the lambda handler
    lambda.start(middlewares.logmiddleware(context.background(), lambdahandler))
}

Bitte beachten Sie, wie wir Veranstaltungsinformationen erhalten.

sqs/main.go

package main

import (
    "context"
    "fmt"

    "httplambda/middlewares"

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

func lambdaHandler(ctx context.Context, event middlewares.Event) (string, error) {
    _ = ctx
    fmt.Println("Queue name:", event.Records[0].SQSEvent.Records[0].EventSourceARN)
    fmt.Println("Hi from SQS-triggered lambda!")
    return "", nil
}

func main() {
    lambda.Start(middlewares.LogMiddleware(context.Background(), lambdaHandler))
}

Finale

Es gibt ein paar Dinge zu beachten:

1. Bevor ich dieser Lösung gefolgt bin, habe ich erfolglos versucht, Typparameter zu verwenden. Sie scheinen in der Signatur der Middleware nicht erlaubt zu sein.
2. Der Code ist zu stark vereinfacht und nicht produktionsbereit.

Wenn das hilft oder Sie noch etwas brauchen, lassen Sie es mich bitte wissen, danke!

Das obige ist der detaillierte Inhalt vonSo erstellen Sie einen generischen Typ für Lambda-Middleware in Go. Für weitere Informationen folgen Sie bitte anderen verwandten Artikeln auf der PHP chinesischen Website!