allow to use http middlewares

Signed-off-by: Vasiliy Tolstov <v.tolstov@unistack.org>
This commit is contained in:
2021-02-07 13:27:28 +03:00
parent 1e4d56d059
commit dc1e05bb18
2 changed files with 23 additions and 2 deletions

12
http.go
View File

@@ -375,7 +375,17 @@ func (h *httpServer) Start() error {
}
}
go http.Serve(ts, handler)
fn := handler
if h.opts.Context != nil {
if mwf, ok := h.opts.Context.Value(middlewareKey{}).([]func(http.Handler) http.Handler); ok && len(mwf) > 0 {
// wrap the handler func
for i := len(mwf); i > 0; i-- {
fn = mwf[i-1](fn)
}
}
}
go http.Serve(ts, fn)
go func() {
t := new(time.Ticker)