WrapMiddleware for echo.Handler

Signed-off-by: Vishal Rana <[email protected]>
firevenus · Mar 7, 2016 · b6deec4 · b6deec4
1 parent 4f08fed
commit b6deec4
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/echo.go b/echo.go
@@ -471,6 +471,18 @@ func (binder) Bind(i interface{}, c Context) (err error) {
 	return
 }
 
+// WrapMiddleware wrap `echo.Handler` into `echo.MiddlewareFunc`.
+func WrapMiddleware(h Handler) MiddlewareFunc {
+	return func(next Handler) Handler {
+		return HandlerFunc(func(c Context) error {
+			if !c.Response().Committed() {
+				h.Handle(c)
+			}
+			return next.Handle(c)
+		})
+	}
+}
+
 func handlerName(h Handler) string {
 	t := reflect.ValueOf(h).Type()
 	if t.Kind() == reflect.Func {