http - 中间件如何在 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用于路由。

我不明白的是 WriteToConsole 中的那个(下一个 http.Handler)参数指的是什么？

它是指我们的 mux 路由器吗？

此外，当我在函数 writetoconsole 的 next.ServeHTTP 行下注释时，html 未呈现或其他任何内容？谁能解释一下 next http.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 处理程序，即执行 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