go/echo
go/echo
1
0
Fork 0
mirror of https://github.com/labstack/echo.git synced 2025-03-21 21:27:04 +02:00
Code Issues Releases Activity

Fixed standard#WrapMiddleware

Browse Source 
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
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
engine/standard/server.go
View File

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