From 76cfd2f6669d51f16f40313e04d53d2514e8e8d1 Mon Sep 17 00:00:00 2001 From: Vishal Rana Date: Thu, 10 Mar 2016 13:29:23 -0800 Subject: [PATCH] Fixed standard#WrapMiddleware Signed-off-by: Vishal Rana --- engine/standard/server.go | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/engine/standard/server.go b/engine/standard/server.go index 3f61a28a..b34eaf87 100644 --- a/engine/standard/server.go +++ b/engine/standard/server.go @@ -136,11 +136,13 @@ func WrapHandler(h http.Handler) echo.HandlerFunc { func WrapMiddleware(m func(http.Handler) http.Handler) echo.MiddlewareFunc { return func(next echo.Handler) echo.Handler { return echo.HandlerFunc(func(c echo.Context) (err error) { - w := c.Response().(*Response).ResponseWriter - r := c.Request().(*Request).Request + req := c.Request().(*Request) + res := c.Response().(*Response) m(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { + res.ResponseWriter = w + req.Request = r err = next.Handle(c) - })).ServeHTTP(w, r) + })).ServeHTTP(res.ResponseWriter, req.Request) return }) }