Small bug fix. Can't close a connection that doesn't exist

This commit is contained in:
Renan DelValle 2016-08-22 12:43:53 -07:00
parent 073faabc21
commit b740c1a61a

4
zk.go
View file

@ -39,13 +39,15 @@ type ServiceInstance struct {
func LeaderFromZK(cluster Cluster) (string, error) {
endpoints := strings.Split(cluster.ZK, ",")
//TODO (rdelvalle): When enabling debugging, change logger here
c, _, err := zk.Connect(endpoints, time.Second*10, zk.WithoutLogger())
defer c.Close()
if err != nil {
return "", errors.Wrap(err, "Failed to connect to Zookeeper at "+cluster.ZK)
}
defer c.Close()
children, _, _, err := c.ChildrenW(cluster.SchedZKPath)
if err != nil {
return "", errors.Wrapf(err, "Path %s doesn't exist on Zookeeper ", cluster.SchedZKPath)