基于Zookeeper实现服务Master选举（哨兵机制）

一、master选举使用场景及结构

        现在很多时候我们的服务需要7*24小时工作，假如一台机器挂了，我们希望能有其它机器顶替它继续工作。此类问题现在多采用master-salve模式，也就是常说的主从模式，正常情况下主机提供服务，备机负责监听主机状态，当主机异常时，可以自动切换到备机继续提供服务(这里有点儿类似于数据库主库跟备库，备机正常情况下只监听，不工作)，这个切换过程中选出下一个主机的过程就是master选举。

        对于以上提到的场景，传统的解决方式是采用一个备用节点，这个备用节点定期给当前主节点发送ping包，主节点收到ping包后会向备用节点发送应答ack，当备用节点收到应答，就认为主节点还活着，让它继续提供服务，否则就认为主节点挂掉了，自己将开始行使主节点职责。如图1所示：

 

二、代码实现

maven依赖

<parent>
        <groupId>org.springframework.boot</groupId>
        <artifactId>spring-boot-starter-parent</artifactId>
        <version>2.0.0.RELEASE</version>
    </parent>
    <dependencies>
        <dependency>
            <groupId>com.101tec</groupId>
        <artifactId>zkclient</artifactId>
        <version>0.10</version>
        <exclusions>
            <exclusion>
                <artifactId>slf4j-api</artifactId>
                <groupId>org.slf4j</groupId>
            </exclusion>
            <exclusion>
                <artifactId>log4j</artifactId>
                <groupId>log4j</groupId>
            </exclusion>
            <exclusion>
                <artifactId>slf4j-log4j12</artifactId>
                <groupId>org.slf4j</groupId>
            </exclusion>
        </exclusions>
    </dependency>
    <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-web</artifactId>
        </dependency>
    </dependencies>

MyApplicationRunner.java

@Component
public class MyApplicationRunner implements ApplicationRunner {
    //创建zk连接
    ZkClient zkClient = new ZkClient("127.0.0.1:2181");
    private String path = "/election";

    @Value("${server.port}")
    private String serverPort;

    //项目启动时候执行
    @Override
    public void run(ApplicationArguments args) throws Exception {
        System.out.println("项目启动成功...");
        createEphemeral();
        //创建事件监听
        zkClient.subscribeDataChanges(path, new IZkDataListener() {
            @Override
            public void handleDataChange(String s, Object o) throws Exception {

            }

            @Override
            public void handleDataDeleted(String s) throws Exception {
                System.out.println("开启重新选举");
                createEphemeral();
            }
        });
    }

    private void createEphemeral(){
        try {
            zkClient.createEphemeral(path);
            ElectionMaster.isSurvival = true;
            System.out.println("serverPort：" + serverPort + ",选举成功....");
        } catch (Exception e) {
            ElectionMaster.isSurvival = false;
        }

    }
}

ElectionMaster.java

public class ElectionMaster {

	// 服务器info信息 是否存活
	public static boolean isSurvival;

}

IndexController.java

@RestController
public class IndexController {
    //1.项目启动时会在zk上创建一个相同临时节点
    //2.谁能够创建成功谁就是主服务器
    //3.使用服务器监听节点是否被删除，如果节点被删除的话重新开始选举（重新创建）
    @Value("${server.port}")
    private String serverPort;

    @RequestMapping("/getServerInfo")
    public String getServerInfo() {
        return "serverPort:" + serverPort + (ElectionMaster.isSurvival ? "选举为主服务器" : "选举为从服务器");
    }
}

application.yml

server:
  port: 8081

启动类

@SpringBootApplication
public class AppMaster {

    public static void main(String[] args) {
        SpringApplication.run(AppMaster.class,args);
    }
}

运行结果

启动项目，访问http://localhost:8081/getServerInfo

把端口号改为8082，启动项目，访问http://localhost:8082/getServerInfo

当把8081停止时，进行重新选举，8082就会成为主服务器，访问http://localhost:8082/getServerInfo