etcd/machines.go

package main

// machineNum returns the number of machines in the cluster
func machineNum() int {
	response, _ := etcdStore.RawGet("_etcd/machines")

	return len(response)
}

// getMachines gets the current machines in the cluster
func getMachines(etcd bool) []string {

	peers := r.Peers()

	machines := make([]string, len(peers)+1)

	var toURL func(string) (string, bool)

	if etcd {
		toURL = nameToEtcdURL
	} else {
		toURL = nameToRaftURL
	}

	leader, ok := toURL(r.Leader())
	self := e.url
	i := 1

	if ok {
		machines[0] = leader
		if leader != self {
			machines[1] = self
			i = 2
		}
	} else {
		machines[0] = self
	}

	// Add all peers to the slice
	for peerName, _ := range peers {
		if machine, ok := toURL(peerName); ok {
			// do not add leader twice
			if machine != leader {
				machines[i] = machine
				i++
			}
		}
	}
	return machines
}