add http handler option for broker

This commit is contained in:
Asim Aslam
2018-11-18 20:40:43 +00:00
parent d9fc2c922d
commit 1cb40831a4
3 changed files with 37 additions and 0 deletions

View File

@@ -126,7 +126,19 @@ func newHttpBroker(opts ...Option) Broker {
mux: http.NewServeMux(),
}
// specify the message handler
h.mux.Handle(DefaultSubPath, h)
// get optional handlers
if h.opts.Context != nil {
handlers, ok := h.opts.Context.Value("http_handlers").(map[string]http.Handler)
if ok {
for pattern, handler := range handlers {
h.mux.Handle(pattern, handler)
}
}
}
return h
}