golang - Gorilla mux customware

gorilla mux example

Question

Estoy usando gorilla mux para administrar enrutamiento. Lo que me falta es integrar un middleware entre cada solicitud.

Por ejemplo

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

Cada solicitud entrante debe pasar a través del middleware. ¿Cómo puedo integrar aquí un midleware?

Actualizar

Lo usaré en combinación con gorila / sesiones, y dicen:

Nota importante: si no está usando gorilla / mux, necesita envolver sus manejadores con context.ClearHandler ya que de lo contrario perderá memoria. Una forma fácil de hacer esto es envolver el mux de nivel superior al llamar a http.ListenAndServe:

¿Cómo puedo prevenir este escenario?

Answer 1

No estoy seguro de por qué @OneOfOne eligió encadenar el enrutador al Middleware, creo que este es un enfoque ligeramente mejor:

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 2

Puede considerar un paquete de middleware como negroni .

Answer 3

Si desea aplicar una cadena de middleware a todas las rutas de un enrutador o un subruteador, puede usar un fork de Gorilla mux https://github.com/bezrukovspb/mux

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

Answer 4

Simplemente crea una envoltura, es bastante fácil en 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))
}