go - 如何通过 yaag 中间件包装 gorilla mux func 处理程序

Question

我正在学习本教程。 http://thenewstack.io/make-a-restful-json-api-go/

router := mux.NewRouter().StrictSlash(true)
for _, route := range routes {
    router.
        Methods(route.Method).
        Path(route.Pattern).
        Name(route.Name).
        Handler(route.HandlerFunc)
}

我需要使用 yaag 中间件包装端点函数。

r.HandleFunc("/", middleware.HandleFunc(handler))

如何实现？

编辑:我正在环绕 Logger 并返回 haddler。记录器采用第一个参数，作为 http.Handle。所以包装 route.HandlerFunc 是行不通的。你能帮我一下吗？

handler := Logger(route.HandlerFunc, route.Name)

    router.
        Methods(route.Method).
        Path(route.Pattern).
        Name(route.Name).
        Handler(handler)

Answer 1

您所要做的就是将 .Handler() 替换为 .HandlerFunc() 并用中间件包装您的处理程序函数，因此每个端点将首先传递给 yaag 中间件，然后传递给您的处理程序函数，像这样:

router := mux.NewRouter().StrictSlash(true)
for _, route := range routes {
    router.
        Methods(route.Method).
        Path(route.Pattern).
        Name(route.Name).
        HandlerFunc(middleware.HandleFunc(route.HandlerFunc)) // change here
}