diff options
Diffstat (limited to 'plumbing/transport/client/client.go')
-rw-r--r-- | plumbing/transport/client/client.go | 9 |
1 files changed, 9 insertions, 0 deletions
diff --git a/plumbing/transport/client/client.go b/plumbing/transport/client/client.go index f749875..ea1e034 100644 --- a/plumbing/transport/client/client.go +++ b/plumbing/transport/client/client.go @@ -23,6 +23,11 @@ var Protocols = map[string]transport.Transport{ // InstallProtocol adds or modifies an existing protocol. func InstallProtocol(scheme string, c transport.Transport) { + if c == nil { + delete(Protocols, scheme) + return + } + Protocols[scheme] = c } @@ -35,5 +40,9 @@ func NewClient(endpoint transport.Endpoint) (transport.Transport, error) { return nil, fmt.Errorf("unsupported scheme %q", endpoint.Scheme) } + if f == nil { + return nil, fmt.Errorf("malformed client for scheme %q, client is defined as nil", endpoint.Scheme) + } + return f, nil } |