mirror of
https://github.com/cubixle/tuu.git
synced 2026-04-30 18:08:39 +01:00
quick implementation of middleware
This commit is contained in:
@@ -0,0 +1,28 @@
|
||||
package tuu
|
||||
|
||||
type Middleware func(Handler) Handler
|
||||
|
||||
type MiddlewareStack struct {
|
||||
stack []Middleware
|
||||
}
|
||||
|
||||
func (m *MiddlewareStack) Use(mw ...Middleware) {
|
||||
m.stack = append(m.stack, mw...)
|
||||
}
|
||||
|
||||
func (m *MiddlewareStack) Get() []Middleware {
|
||||
return m.stack
|
||||
}
|
||||
|
||||
func (m *MiddlewareStack) handler(r *Route) Handler {
|
||||
handler := r.Handler
|
||||
if len(m.stack) == 0 {
|
||||
return handler
|
||||
}
|
||||
|
||||
for _, mw := range m.stack {
|
||||
handler = mw(handler)
|
||||
}
|
||||
|
||||
return handler
|
||||
}
|
||||
Reference in New Issue
Block a user