micro-network-transport-http/options.go
Vasiliy Tolstov 5a602b5818 use own fork
Signed-off-by: Vasiliy Tolstov <v.tolstov@unistack.org>
2020-09-20 17:07:36 +03:00

24 lines
570 B
Go

package http
import (
"context"
"net/http"
"github.com/unistack-org/micro/v3/network/transport"
)
// Handle registers the handler for the given pattern.
func Handle(pattern string, handler http.Handler) transport.Option {
return func(o *transport.Options) {
if o.Context == nil {
o.Context = context.Background()
}
handlers, ok := o.Context.Value("http_handlers").(map[string]http.Handler)
if !ok {
handlers = make(map[string]http.Handler)
}
handlers[pattern] = handler
o.Context = context.WithValue(o.Context, "http_handlers", handlers)
}
}