models: split auth with strings.SplitN

hiqsociety · Sep 18, 2017 · f8bc4e3 · f8bc4e3
1 parent 4169d1e
commit f8bc4e3
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/pkg/models/etcd/etcdclient.go b/pkg/models/etcd/etcdclient.go
@@ -51,7 +51,7 @@ func New(addrlist string, auth string, timeout time.Duration) (*Client, error) {
 	}
 
 	if auth != "" {
-		split := strings.Split(auth, ":")
+		split := strings.SplitN(auth, ":", 2)
 		if len(split) != 2 || split[0] == "" {
 			return nil, errors.Errorf("invalid auth")
 		}

diff --git a/pkg/models/zk/zkclient.go b/pkg/models/zk/zkclient.go
@@ -60,7 +60,7 @@ func NewWithLogfunc(addrlist string, auth string, timeout time.Duration, logfunc
 		logger: &zkLogger{logfunc},
 	}
 	if auth != "" {
-		split := strings.Split(auth, ":")
+		split := strings.SplitN(auth, ":", 2)
 		if len(split) != 2 || split[0] == "" {
 			return nil, errors.Errorf("invalid auth")
 		}