> 作者：ReganYue
>
> 來源：[恆生LIGHT雲社群](https://developer.hs.net/thread/1738)

## **超細！細說Zookeeper選舉的一個案例**

**今天我們來帶著大家實現用Zookeeper實現選舉的案例，幫助大家更好的學習Zookeeper。**

### **六、判斷是否連線Zookeeper**

```
func (electionManager *ElectionManager) isConnected() bool {
    if electionManager.ZKClientConn == nil {
        return false
    } else if electionManager.ZKClientConn.State() != zk.StateConnected {
        return false
    }
    return true
}
```

**在初始化Zookeeper連線時需要判斷是否連線了Zookeeper，其實我們可以看連線是否為nil值來判斷有沒有連線，如果想判斷連線是否有問題的話，我們最好用連線的State()方法，如果是zk.StateConnected，就表示連線成功了，如果是其他就表示連線異常，下面列舉連線異常和成功的值。**

```
StateUnknown           State = -1
StateDisconnected      State = 0
StateConnecting        State = 1
StateAuthFailed        State = 4
StateConnectedReadOnly State = 5
StateSaslAuthenticated State = 6
StateExpired           State = -112

StateConnected  = State(100)
StateHasSession = State(101)
```

**可以看到，zk.StateConnected也就是100，所以這裡可能可以將zk.StateConnected替換為100，我也不知道是否可行，你可以試一試~ StateUnknown也就是-1表示狀態是未知的，StateDisconnected也就是0表示狀態是未連線，其它我就不細說了。**

### **七、選舉的邏輯**

```
func (electionManager *ElectionManager) Run() {
    err := electionManager.electMaster()
    if err != nil {
        fmt.Println(err)
    }
    electionManager.watchMaster()
}
```

**是不是很眼熟，這就是main函式中開闢協程執行的東西——選舉。**

**很簡單，直接進行選舉選出master，然後對master節點進行監聽。**

### **八、選舉邏輯**

```
func (electionManager *ElectionManager) electMaster() error {
    err := electionManager.initConnection()
    if err != nil {
        return err
    }
    isExist, _, err := electionManager.ZKClientConn.Exists(electionManager.ZKConfig.RootPath)
    if err != nil {
        return err
    }
    if !isExist {
        path, err := electionManager.ZKClientConn.Create(electionManager.ZKConfig.RootPath,
            nil, 0, zk.WorldACL(zk.PermAll))
        if err != nil {
            return err
        }
        if electionManager.ZKConfig.RootPath != path {
            return errors.New("建立的" + electionManager.ZKConfig.RootPath + " !=" + path)
        }
    }

masterPath := electionManager.ZKConfig.RootPath + electionManager.ZKConfig.MasterPath
    
    path, err := electionManager.ZKClientConn.Create(masterPath, nil, zk.FlagEphemeral, zk.WorldACL(zk.PermAll))
    if err == nil {
        if path == masterPath {
            fmt.Println("選舉master成功")
            electionManager.IsMaster

超細！細說Zookeeper選舉的一個案例（下）

相關文章