大猩猩多路複用器定製中間件

Question

我正在使用大猩猩多路複用器進行管理路由。我所缺少的是在每個請求之間集成一箇中間件。

例如

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) 
    } 
} 

每個傳入請求應該穿過中間件。我怎樣才能在這裏整合一個midleware？

更新

我會用它與大猩猩/會話組合，他們說：

重要提示：如果您使用的不是大猩猩/多路複用器，你需要用你的 與context.ClearHandler作爲處理程序，否則你會泄漏內存！一個 簡單的方法來做到這一點是包裝頂級MUX調用 http.ListenAndServe時：

我怎樣才能避免這種情況？

Answer 1

只需創建一個包裝，這是相當容易去：

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)) 
} 

Answer 2

你可能會考慮一箇中間件包，如negroni。

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) 
})} 

Answer 4

如果你想中間件鏈應用到路由器的所有路由或subrouter可以使用大猩猩MUX的叉https://github.com/bezrukovspb/mux

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