大猩猩mux自定义中间件

Question

我正在使用gorilla mux来管理路由。我缺少的是在每个请求之间集成一个中间件。

例如：

package main

import (
    "fmt"
    "github.com/gorilla/mux"
    "log"
    "net/http"
    "strconv"
)

func HomeHandler(response http.ResponseWriter, request *http.Request) {

    fmt.Fprintf(response, "Hello home")
}

func main() {

    port := 3000
    portstring := strconv.Itoa(port)

    r := mux.NewRouter()
    r.HandleFunc("/", HomeHandler)
    http.Handle("/", r)

    log.Print("Listening on port " + portstring + " ... ")
    err := http.ListenAndServe(":"+portstring, nil)
    if err != nil {
        log.Fatal("ListenAndServe error: ", err)
    }
}

每个进入的请求都应该经过中间件。我如何集成一个中间件？

更新

我将与gorilla/sessions一起使用，他们说：

重要提示：如果您不使用gorilla/mux，则需要使用context.ClearHandler包装处理程序，否则会泄漏内存！在调用http.ListenAndServe时包装顶级mux是一种简单的方法：

我该如何避免这种情况？

- softshipper

---

针对那些来到这里的人：gorilla/mux 现在内置了中间件处理功能：https://github.com/gorilla/mux#middleware - kaustavdm

Answer 1

只需在 Go 中创建一个包装器，这非常容易：

func HomeHandler(response http.ResponseWriter, request *http.Request) {

    fmt.Fprintf(response, "Hello home")
}

func Middleware(h http.Handler) http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        log.Println("middleware", r.URL)
        h.ServeHTTP(w, r)
    })
}
func main() {
    r := mux.NewRouter()
    r.HandleFunc("/", HomeHandler)
    http.Handle("/", Middleware(r))
}

- OneOfOne

---

谢谢，h.ServeHTTP(w, r) 这部分是我所缺失的。他们没有很清楚地说明如何直接调用处理程序。 - tkiethanom

您也可以使用 r.Use(Middleware) 来注册中间件。 - DBadura

Answer 2

Mux有一种官方的方式来做到这一点，看看这个例子

// a regular middleware
func Middleware(h http.Handler) http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        // do stuff before the handlers
        h.ServeHTTP(w, r)
        // do stuff after the hadlers

    })
}

// if you want to pass data into the middleware 
func Middleware2(s string) mux.MiddlewareFunc {
    return func(h http.Handler) http.Handler {
        return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
            // do stuff
            fmt.Println(s)
            h.ServeHTTP(w, r)
        })
    }
}

func main() {
    router := mux.NewRouter()


    router.Use(Middleware)
    //you can apply it to a sub-router too
    subRouter := router.PathPrefix("/sub_router/").Subrouter()
    subRouter.Use(Middleware2("somePrams"))
    // Add more middleware if you need, call router.Use Again
    router.Use(Middleware3, Middleware4, Middleware5)

    _ = http.ListenAndServe(":80", router)
}

mux官网上的官方文档

- Isaac Weingarten

Answer 3

我不确定为什么@OneOfOne选择将路由器链接到中间件，我认为这是稍微更好的方法：

func main() {
    r.Handle("/",Middleware(http.HandlerFunc(homeHandler)))
    http.Handle("/", r)
}

func Middleware(h http.Handler) http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
    h.ServeHTTP(w, r)
})}

- Orr

---

这种方法正是我一直在寻找的——一个带有中间件的嵌套Handle。谢谢！ - James Taylor

这基本上与@OneOfOne的答案相同。 - Alexander Mills

Answer 4

如果你想对一个路由或子路由应用一个中间件链，你可以使用 Gorilla mux 的分支 https://github.com/bezrukovspb/mux

subRouter := router.PathPrefix("/use-a-b").Subrouter().Use(middlewareA, middlewareB)
subRouter.Path("/hello").HandlerFunc(requestHandlerFunc)

- Anton Bezrukov

---

谢谢，伙计！这个 express.js 中间件风格正是我所寻找的。 - alsdkjasdlkja

中间件现在已经内置于Gorilla Mux中的使用。 - Ozan

@ozy 你是指使用 router.Use()，但最好使用子路由器，这样中间件可以应用于不同的位置。 - Alexander Mills

Answer 5

您可以考虑使用中间件包，例如negroni。

- kpublik

---

当我使用gorilla/sessions时，不会发生内存泄漏吗？ - softshipper

我无法编辑，因为“建议编辑队列已满”。该链接不再存在。我想用一个新的链接negroni来替换它。 - Carson