Question

使用julienschmidt / httprouter时连接中间件的最佳方法是什么？

就我用Google搜索而言，http.HandlerFunc只接受func (w ResponseWriter, r *Request)形式的函数，而httprouter.Handle函数的格式为func (w http.ResponseWriter, r *http.Request, ps httprouter.Params)。

如何在不将httprouter.Handle函数转换为http.HandlerFunc的情况下链接中间件？

例如：
我的routes.go是形式，

router  :=  httprouter.New()
router.POST("/api/user/create", middlewares.EscapeStringsMiddleware(User.CreateUser))
log.Fatal(http.ListenAndServe(":8000",  router))

如何为上述路线编写中间件功能？

已经尝试过的方法：

1

func EscapeStringsMiddleware(next http.Handler) httprouter.Handle {

    return func (response http.ResponseWriter, request *http.Request, ps httprouter.Params) {
        err := request.ParseForm()
        if err != nil {
            panic(err)
        }

        for key, values := range request.Form {
            for i, value := range values {
                value = template.HTMLEscapeString(value)
                value = template.JSEscapeString(value)
                request.Form[key][i] = value
            }
        }
        next.ServeHTTP(response, request)
    }
}

获得错误：

cannot use User.CreateUser (type func(http.ResponseWriter, *http.Request, httprouter.Params)) as type http.Handler in argument to middlewares.EscapeStringsMiddleware:
func(http.ResponseWriter, *http.Request, httprouter.Params) does not implement http.Handler (missing ServeHTTP method)

2

func EscapeStringsMiddleware(next httprouter.Handle) httprouter.Handle {

    return func (response http.ResponseWriter, request *http.Request, ps httprouter.Params) {
        err := request.ParseForm()
        if err != nil {
            panic(err)
        }

        for key, values := range request.Form {
            for i, value := range values {
                value = template.HTMLEscapeString(value)
                value = template.JSEscapeString(value)
                request.Form[key][i] = value
            }
        }
        next.ServeHTTP(response, request)
    }
}

获得错误：

next.ServeHTTP undefined (type httprouter.Handle has no field or method ServeHTTP)

另外，我如何链接多个中间件？

例如，

router.POST("/api/user/create", middlewares.VerifyCSRF(middlewares.EscapeStringsMiddleware(User.CreateUser)))

Answer 1

此问题不适用于您的中间件处理程序。由于User.CreateUser不属于http.Handler类型，因此您遇到了错误。

尝试这种模式：

重要的一点是返回http.Handler并使用func(w http.ResponseWriter, r *http.Request)包裹http.HandlerFunc。

func Handler() http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        // do stuff
    })
}

go source：

// The HandlerFunc type is an adapter to allow the use of
// ordinary functions as HTTP handlers. If f is a function
// with the appropriate signature, HandlerFunc(f) is a
// Handler that calls f.
type HandlerFunc func(ResponseWriter, *Request)

// ServeHTTP calls f(w, r).
func (f HandlerFunc) ServeHTTP(w ResponseWriter, r *Request) {
    f(w, r)
}

根据您的反馈：

httprouter.Handle未实施ServeHTTP。它是直接调用的。例如：next(w, r, ps)

您将在下面找到中间件处理程序的示例。

// Middleware without "github.com/julienschmidt/httprouter"
func StdToStdMiddleware(next http.Handler) http.Handler {

    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        // do stuff
        next.ServeHTTP(w, r)
    })
}

// Middleware for a standard handler returning a "github.com/julienschmidt/httprouter" Handle
func StdToJulienMiddleware(next http.Handler) httprouter.Handle {

    return func(w http.ResponseWriter, r *http.Request, ps httprouter.Params) {
        // do stuff
        next.ServeHTTP(w, r)
    }
}

// Pure "github.com/julienschmidt/httprouter" middleware
func JulienToJulienMiddleware(next httprouter.Handle) httprouter.Handle {

    return func(w http.ResponseWriter, r *http.Request, ps httprouter.Params) {
        // do stuff
        next(w, r, ps)
    }
}

func JulienHandler() httprouter.Handle {
    return func(w http.ResponseWriter, r *http.Request, ps httprouter.Params) {
        // do stuff
    }
}

func StdHandler() http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        // do stuff
    })
}

func main() {
    router := httprouter.New()
    router.POST("/api/user/create", StdToJulienMiddleware(StdHandler()))
    router.GET("/api/user/create", JulienToJulienMiddleware(JulienHandler()))
    log.Fatal(http.ListenAndServe(":8000", router))
}

Answer 2

从https://github.com/cnvrtly/adaptr lib中，还将适配器添加到链请求中间件函数中。

func compatibleHandler(h http.Handler, httprParamsCtxKey interface{}) httprouter.Handle {
return toHttpRouterHandle(h, httprParamsCtxKey)
}
func toHttpRouterHandle(h http.Handler, httprParamsCtxKey interface{}) func(w http.ResponseWriter, r *http.Request, p httprouter.Params) {
   return func(w http.ResponseWriter, r *http.Request, p httprouter.Params) {
    if httprParamsCtxKey != nil {
        r = SetCtxValue(r, httprParamsCtxKey, p)
    }
    h.ServeHTTP(w, r)
   }
}

如果您正在寻找示例/想法如何定义路线：https://github.com/cnvrtly/golang-appengine-seed

在goLang中使用julienschmidt / httprouter时如何使用中间件？

2 个答案: