httprouter 传入了很多中间件函数

Question

我来自 node express，我能够传入尽可能多的中间件，例如：routes.use('/*', ensureAuth, logImportant, ... n);

如何在使用 r.GET("/", HomeIndex) 时做类似的事情？

我是否被迫做类似 EnsureAuth(HomeIndex) 的事情？因为我可以让它工作。不幸的是，我不确定在不将函数链接在一起的情况下添加任意数量的中间件的好方法是什么。

是否有更优雅的方法，以便我可以以某种方式使用可变类型函数来完成 r.GET("/", applyMiddleware(HomeIndex, m1, m2, m3, m4)？我现在正在尝试，但我觉得有更好的方法。

我查看了 httprouter 问题页面，找不到任何内容:(

谢谢！

Answer 1

这是我如何做的一个例子：

package main

import (
    "context"
    "fmt"
    "log"
    "net/http"

    "github.com/julienschmidt/httprouter"
    "github.com/justinas/alice"
)

// m1 is middleware 1
func m1(next http.Handler) http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        //do something with m1
        log.Println("m1 start here")
        next.ServeHTTP(w, r)
        log.Println("m1 end here")
    })
}

// m2 is middleware 2
func m2(next http.Handler) http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        //do something with m2
        log.Println("m2 start here")
        next.ServeHTTP(w, r)
        log.Println("m2 end here")
    })
}

func index(w http.ResponseWriter, r *http.Request) {
    // get httprouter.Params from request context
    ps := r.Context().Value("params").(httprouter.Params)
    fmt.Fprintf(w, "Hello, %s", ps.ByName("name"))
}

// wrapper wraps http.Handler and returns httprouter.Handle
func wrapper(next http.Handler) httprouter.Handle {
    return func(w http.ResponseWriter, r *http.Request, ps httprouter.Params) {
        //pass httprouter.Params to request context
        ctx := context.WithValue(r.Context(), "params", ps)
        //call next middleware with new context
        next.ServeHTTP(w, r.WithContext(ctx))
    }
}

func main() {
    router := httprouter.New()

    chain := alice.New(m1, m2)

    //need to wrap http.Handler to be compatible with httprouter.Handle
    router.GET("/user/:name", wrapper(chain.ThenFunc(index)))

    log.Fatal(http.ListenAndServe(":9000", router))
}

Link 编码（你不能运行它来自 play.golang.org）：https://play.golang.org/p/BOCt97xcoY

httprouter 传入了很多中间件函数

httprouter pass in many middleware functions

middleware

go

httprouter