go - fasthttp + fasthttprouter，尝试写中间件

Question

我目前正在尝试编写一些中间件来使用 fasthttp 和 fasthttprouter。我被卡住了。

func jwt(h fasthttprouter.Handle) fasthttprouter.Handle {
myfunc := func(ctx *fasthttp.RequestCtx, _ fasthttprouter.Params) {
    fmt.Println(string(ctx.Request.Header.Cookie("Authorization")))
}

return myfunc
}

我现在如何运行实际的处理程序？我觉得我错过了一些非常简单的东西。

我已通读这篇博文:Middleware in Golang .但是我迷路了。

有什么想法吗？

问候

Answer 1

例如，让我们创建一个中间件函数来处理 CORS，使用:

github.com/buaazp/fasthttprouter 和 github.com/valyala/fasthttp

var (
    corsAllowHeaders     = "authorization"
    corsAllowMethods     = "HEAD,GET,POST,PUT,DELETE,OPTIONS"
    corsAllowOrigin      = "*"
    corsAllowCredentials = "true"
)

func CORS(next fasthttp.RequestHandler) fasthttp.RequestHandler {
    return func(ctx *fasthttp.RequestCtx) {

        ctx.Response.Header.Set("Access-Control-Allow-Credentials", corsAllowCredentials)
        ctx.Response.Header.Set("Access-Control-Allow-Headers", corsAllowHeaders)
        ctx.Response.Header.Set("Access-Control-Allow-Methods", corsAllowMethods)
        ctx.Response.Header.Set("Access-Control-Allow-Origin", corsAllowOrigin)

        next(ctx)
    }
}

现在我们将这个中间件函数链接到我们的索引处理程序上，并将其注册到路由器上。

func Index(ctx *fasthttp.RequestCtx) {
    fmt.Fprint(ctx, "some-api")
}

func main() {
    router := fasthttprouter.New()
    router.GET("/", Index)

    if err := fasthttp.ListenAndServe(":8181", CORS(router.Handler)); err != nil {
        log.Fatalf("Error in ListenAndServe: %s", err)
    }
}