[#383] rpc/client: Export URI-parsing function

Signed-off-by: Evgenii Stratonikov <evgeniy@nspcc.ru>
This commit is contained in:
Evgenii Stratonikov 2022-03-09 12:56:02 +03:00 committed by Alex Vanin
parent fadd47f4fb
commit 50382114f4
2 changed files with 45 additions and 25 deletions

View file

@ -2,7 +2,6 @@ package client
import (
"crypto/tls"
"net/url"
"time"
"google.golang.org/grpc"
@ -58,35 +57,23 @@ func WithNetworkAddress(v string) Option {
//
// Ignored if WithGRPCConn is provided.
func WithNetworkURIAddress(addr string, tlsCfg *tls.Config) []Option {
uri, err := url.ParseRequestURI(addr)
host, isTLS, err := ParseURI(addr)
if err != nil {
return []Option{WithNetworkAddress(addr)}
}
// check if passed string was parsed correctly
// URIs that do not start with a slash after the scheme are interpreted as:
// `scheme:opaque` => if `opaque` is not empty, then it is supposed that URI
// is in `host:port` format
if uri.Opaque != "" {
return []Option{WithNetworkAddress(addr)}
}
switch uri.Scheme {
case grpcScheme:
tlsCfg = nil
case grpcTLSScheme:
if tlsCfg == nil {
tlsCfg = &tls.Config{}
}
default:
// not supported scheme
return nil
}
return []Option{
WithNetworkAddress(uri.Host),
WithTLSCfg(tlsCfg),
opts := make([]Option, 2)
opts[0] = WithNetworkAddress(host)
if isTLS {
if tlsCfg == nil {
tlsCfg = &tls.Config{}
}
opts[1] = WithTLSCfg(tlsCfg)
} else {
opts[1] = WithTLSCfg(nil)
}
return opts
}
// WithDialTimeout returns option to specify