web-dev-qa-db-ja.com

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/muxを使用していない場合、ハンドラーをcontext.ClearHandlerでラップする必要があります。そうしないと、メモリリークが発生します。これを行う簡単な方法は、http.ListenAndServeを呼び出すときに最上位のmuxをラップすることです。

このシナリオを防ぐにはどうすればよいですか？

gogorilla
27
zero_coding

ラッパーを作成するだけで、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))
}
60
OneOfOne

@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)
})}
6
Orr

ミドルウェアチェーンをルーターまたはサブルーターのすべてのルートに適用する場合は、Gorilla muxのフォークを使用できます https://github.com/bezrukovspb/mux

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

negroni などのミドルウェアパッケージを検討することもできます。

1
kpublik
func Middleware(h http.Handler) http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        // do stuff
        h.ServeHTTP(w, r)
    })
}

func main () {
    router := mux.NewRouter()
    router.Use(Middleware)
    // Add more middleware if you need
    // router.Use(Middleware2) 
    // router.Use(Middleware3)

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