build middleware methods

Author: 2hmad

middleware.go

@@ -0,0 +1,27 @@
+package routing
+
+type MiddlewareFunc func(handler Handler) Handler
+
+type Middleware interface {
+	Middleware(handler Handler) Handler
+	Handle(ctx *Context, next Handler) error
+}
+
+func (m MiddlewareFunc) Middleware(handler Handler) Handler {
+	return m(handler)
+}
+
+func (r *Router) Use(method string, middlewares ...interface{}) {
+	for _, middleware := range middlewares {
+		if middlewareFunc, ok := middleware.(MiddlewareFunc); ok {
+			r.middlewares[method] = append(r.middlewares[method], middlewareFunc)
+		} else if middleware, ok := middleware.(Middleware); ok {
+			r.middlewares[method] = append(r.middlewares[method], middleware)
+		}
+	}
+}
+
+func (m MiddlewareFunc) Handle(ctx *Context, next Handler) error {
+	h := m(next)
+	return h(ctx)
+}