中间件在 Go 中的 chi 路由中如何工作以及 http.Handler 参数在中间件中指的是什么？

Question

-- routes.go --
package main

import (
    "hotelsystem/pkg/config"
    "hotelsystem/pkg/handlers"
    "net/http"

    "github.com/go-chi/chi/v5"
    "github.com/go-chi/chi/v5/middleware"
)

func routes(app *config.AppConfig) http.Handler {
    mux := chi.NewRouter()
    mux.Use(middleware.Recoverer)
    mux.Use(WriteToConsole)
    mux.Get("/", handlers.Repo.Home)
    mux.Get("/about", handlers.Repo.About)
    return mux

}
-- middleware.go --
package main

import (
    "fmt"
    "net/http"
)

func WriteToConsole(next http.Handler) http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        fmt.Println("Hit the page")
        next.ServeHTTP(w, r)
    })
}
-- main.go --
package main

import (
    "hotelsystem/pkg/config"
    "hotelsystem/pkg/handlers"
    "hotelsystem/pkg/render"
    "log"
    "net/http"
)

const portNumber = ":3000"

func main() {
    var app config.AppConfig
    tc, err := render.CreateTemplateCache()
    if err != nil {
        log.Fatal("Can't create templatecache", err)
    }
    app.TemplateCache = tc
    app.UseCache = false
    repo := handlers.NewRepo(&app)
    handlers.NewHandlers(repo)
    render.NewTemplate(&app)
    // http.HandleFunc("/", handlers.Repo.Home)
    // http.HandleFunc("/about", handlers.Repo.About)
    // http.ListenAndServe(portNumber, nil)

    srv := &http.Server{
        Addr:    portNumber,
        Handler: routes(&app),
    }
    err = srv.ListenAndServe()
    if err != nil {
        log.Fatal(err)
    }
}

我很难理解中间件。

我正在使用chi进行路由。

我不明白的是（下一个http.Handler）参数指WriteToConsole的是什么？

它指的是我们的多路复用器路由器吗？

另外，当我注释掉next.ServeHTTPwritetoconsole 函数行时，html 不会呈现或什么？有人能解释一下 nexthttp.Handler指的是什么吗next.serveHTTP？

Answer 1

next是“处理程序链”中的下一个处理程序。

当你这样做时：

mux.Use(middleware.Recoverer)
mux.Use(WriteToConsole)
mux.Get("/", handlers.Repo.Home)
mux.Get("/about", handlers.Repo.About)

您本质上是在注册两个“处理程序链”：

mux.Get("/", middleware.Recoverer(WriteToConsole(handlers.Repo.Home)))
mux.Get("/about", middleware.Recoverer(WriteToConsole(handlers.Repo.About)))

中间件函数返回的每个处理程序都必须调用next为其提供的处理程序，即 do next.ServeHTTP(w, r)，如果它不调用，next则链将被破坏，并且该链中的其余处理程序将被忽略。

一个简化的代码示例可以更好地说明链接：

type handler func()

// your handler
func f() { fmt.Println("f") }

// one middleware
func g(next handler) handler {
    return func() {
        fmt.Print("g.")
        next()
    }
}

// another middleware
func h(next handler) handler {
    return func() {
        fmt.Print("h.")
        next()
    }
}

有了上面的内容，您就可以执行以下操作：

func main() {
    h1 := h(g(f))
    h1()

    h2 := g(h(f))
    h2()

    // And you can chain as many of these as you like
    // and in any order you like.
    h3 := h(g(h(h(h(g(g(h(f))))))))
    h3()
}

https://play.golang.org/p/4NXquYsaljr