renamed site directory to server
This commit is contained in:
@@ -0,0 +1,34 @@
|
||||
package response
|
||||
|
||||
import "net/http"
|
||||
|
||||
type (
|
||||
Mux struct {
|
||||
Mux *http.ServeMux
|
||||
middleware []Middleware
|
||||
}
|
||||
|
||||
Middleware = func(HandlerFunc) HandlerFunc
|
||||
)
|
||||
|
||||
func NewMux(ms ...Middleware) *Mux {
|
||||
return &Mux{
|
||||
Mux: http.NewServeMux(),
|
||||
middleware: ms,
|
||||
}
|
||||
}
|
||||
|
||||
func (m *Mux) Handle(pattern string, fn HandlerFunc) {
|
||||
for _, mw := range m.middleware {
|
||||
prev := fn
|
||||
fn = mw(func(r *http.Request) (Response, error) {
|
||||
return prev(r)
|
||||
})
|
||||
}
|
||||
|
||||
m.Mux.Handle(pattern, Handler(fn))
|
||||
}
|
||||
|
||||
func (m *Mux) ServeHTTP(w http.ResponseWriter, r *http.Request) {
|
||||
m.Mux.ServeHTTP(w, r)
|
||||
}
|
||||
Reference in New Issue
Block a user