1
0
mirror of https://github.com/labstack/echo.git synced 2025-06-08 23:56:20 +02:00

Fixed standard#WrapMiddleware

Signed-off-by: Vishal Rana <vr@labstack.com>
This commit is contained in:
Vishal Rana 2016-03-10 13:29:23 -08:00
parent 8ee864ae08
commit 76cfd2f666

View File

@ -136,11 +136,13 @@ func WrapHandler(h http.Handler) echo.HandlerFunc {
func WrapMiddleware(m func(http.Handler) http.Handler) echo.MiddlewareFunc { func WrapMiddleware(m func(http.Handler) http.Handler) echo.MiddlewareFunc {
return func(next echo.Handler) echo.Handler { return func(next echo.Handler) echo.Handler {
return echo.HandlerFunc(func(c echo.Context) (err error) { return echo.HandlerFunc(func(c echo.Context) (err error) {
w := c.Response().(*Response).ResponseWriter req := c.Request().(*Request)
r := c.Request().(*Request).Request res := c.Response().(*Response)
m(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { m(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
res.ResponseWriter = w
req.Request = r
err = next.Handle(c) err = next.Handle(c)
})).ServeHTTP(w, r) })).ServeHTTP(res.ResponseWriter, req.Request)
return return
}) })
} }