Gorilla mux custom middleware
I'm not sure why @OneOfOne chose to chain router into the Middleware, I think this is slight better approach:
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)
})}
Mux has an official way of doing it look on this example
// 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)
}
the official doc on the mux website
Just create a wrapper, it's rather easy in 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))
}