From 7725d5ed31c0d1600d07c46eb97dc00f207251a6 Mon Sep 17 00:00:00 2001 From: Alexey Khit Date: Fri, 4 Nov 2022 06:24:39 +0300 Subject: [PATCH] Rewrite get handlers code --- cmd/streams/handlers.go | 13 ++++++++----- 1 file changed, 8 insertions(+), 5 deletions(-) diff --git a/cmd/streams/handlers.go b/cmd/streams/handlers.go index c9fa2eeb..7879f3e2 100644 --- a/cmd/streams/handlers.go +++ b/cmd/streams/handlers.go @@ -17,17 +17,20 @@ func HandleFunc(scheme string, handler Handler) { handlers[scheme] = handler } -func HasProducer(url string) bool { +func getHandler(url string) Handler { i := strings.IndexByte(url, ':') if i <= 0 { // TODO: i < 4 ? - return false + return nil } - return handlers[url[:i]] != nil + return handlers[url[:i]] +} + +func HasProducer(url string) bool { + return getHandler(url) != nil } func GetProducer(url string) (streamer.Producer, error) { - i := strings.IndexByte(url, ':') - handler := handlers[url[:i]] + handler := getHandler(url) if handler == nil { return nil, fmt.Errorf("unsupported scheme: %s", url) }