2

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

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

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

有没有更优雅的方法，所以我可以以某种方式使用可变参数类型函数来做r.GET("/", applyMiddleware(HomeIndex, m1, m2, m3, m4)？我现在正在尝试，但我觉得有更好的方法来做到这一点。

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

谢谢！

4

1 回答 1

3

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

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

链接到代码（虽然你不能运行它play.golang.org）： https: //play.golang.org/p/BOCt97xcoY

于 2017-11-07T07:20:18.273 回答