Flink基于zookeeper的高可用实现源码分析
Flink中JobMaster、ResourceManager、Dispatcher、WebMonitorEndpoint提供了基于zookeeper高可用，涉及到leader选举与监听， leader选举基于zookeeper开源客户端CuratorFramework 的LeaderLatch方式实现，监听则通过NodeCache实现。基于此Flink提供了zookeeper高可用ZooKeeperHaServices, 通过该工具类可以创建LeaderElectionService与LeaderRetrievalService,包含了对应与zookeeper的znode节点分别是
/resource_manager_lock
/dispatcher_lock
/job_manager_lock
/rest_server_lockLeaderElectionService用于leader选举服务，基于zk的实现类是ZooKeeperLeaderElectionService，LeaderRetrievalService用于监听leader恢复服务，基于zk的实现类是ZooKeeperLeaderRetrievalService。
涉及到两个重要的角色：
LeaderContender接口，在leader选举中使用，代表了参与leader竞争的角色，其实现类有JobManagerRunner、ResourceManager、Dispatcher、WebMonitorEndpoint，该接口中包含了两个重要的方法：
1. grantLeadership,表示leader竞选成功的回调方法
2. revokeLeadership,表示由leader变为非leader的回调方法
LeaderRetrievalListener接口，在监听leader恢复中使用，代表的是一个监听者，其实现类有ResourceManagerLeaderListener、JobManagerLeaderListener等，该接口包含一个重要的方法：
1. notifyLeaderAddress,leader发生变化时回调方法
以JobMaster与TaskExecutor之间的交互为例，JobMaster需要通过LeaderElectionService完成leader的选举，而TaskExecutor则需要监听jobMaster leader的变化，通过LeaderRetrievalService完成。
JobMaster选举
JobManagerRunner负责JobMaster的选举与启动，内部包含了LeaderElectionService对象属性leaderElectionService，也就是ZooKeeperLeaderElectionService的对象，在调用JobManagerRunner的start方法时会调用leaderElectionService的start方法，传入参数this也就是当前的JobManagerRunner对象，
public void start() throws Exception {
try {
leaderElectionService.start(this);
} catch (Exception e) {
log.error("Could not start the JobManager because the leader election service did not start.", e);
throw new Exception("Could not start the leader election service.", e);
}
}在ZooKeeperLeaderElectionService 的start方法里面
public void start(LeaderContender contender) throws Exception {
Preconditions.checkNotNull(contender, "Contender must not be null.");
Preconditions.checkState(leaderContender == null, "Contender was already set.");
LOG.info("Starting ZooKeeperLeaderElectionService {}.", this);
synchronized (lock) {
client.getUnhandledErrorListenable().addListener(this);
leaderContender = contender;
leaderLatch.addListener(this);
leaderLatch.start();
cache.getListenable().addListener(this);
cache.start();
client.getConnectionStateListenable().addListener(listener);
running = true;
}
}leaderLatch 表示的是CuratorFramework里面参与leader选举的LeaderLatch对象，添加了一个this的listener, 也就是LeaderLatchListener，选举成功会调用isLeader方法，由leader变为非leader调用notLeader方法；并且还要通过NodeCache方式添加了监控当前节点变化的listener，也就是NodeCacheListener，当监听的节点发生变化则调用nodeChanged方法。
若参与者竞选leader成功就会调用isLeader方法
public void isLeader() {
synchronized (lock) {
if (running) {
issuedLeaderSessionID = UUID.randomUUID();
confirmedLeaderSessionID = null;
if (LOG.isDebugEnabled()) {
LOG.debug(
"Grant leadership to contender {} with session ID {}.",
leaderContender.getAddress(),
issuedLeaderSessionID);
}
leaderContender.grantLeadership(issuedLeaderSessionID);
} else {
LOG.debug("Ignoring the grant leadership notification since the service has " +
"already been stopped.");
}
}
}在里面会调用leaderContender的grantLeadership方法，也就是JobManagerRunner的grantLeadership，就会执行JobMaster的启动并且执行任务的调度。
在检测到选择leader所用的path发生变化就会调用nodeChanged方法，在nodeChanged方法里面判断当前角色是leader就将leader信息写入到zookeeper路径里面去。
TaskExecutor监听
在TaskExecutor中包含了一个JobLeaderService的对象属性jobLeaderService，在jobLeaderService中维护了所有的jobmaster leader,并且监听它，监听正是通过ZooKeeperLeaderRetrievalService完成。在申请slot时会调用TaskExecutor的requestSlot方法，在该方法里面调用jobLeaderService.add方法，
public void addJob(final JobID jobId, final String defaultTargetAddress) throws Exception {
Preconditions.checkState(JobLeaderService.State.STARTED == state, "The service is currently not running.");
LOG.info("Add job {} for job leader monitoring.", jobId);
final LeaderRetrievalService leaderRetrievalService = highAvailabilityServices.getJobManagerLeaderRetriever(
jobId,
defaultTargetAddress);
JobLeaderService.JobManagerLeaderListener jobManagerLeaderListener = new JobManagerLeaderListener(jobId);
final Tuple2<LeaderRetrievalService, JobManagerLeaderListener> oldEntry = jobLeaderServices.put(jobId, Tuple2.of(leaderRetrievalService, jobManagerLeaderListener));
if (oldEntry != null) {
oldEntry.f0.stop();
oldEntry.f1.stop();
}
leaderRetrievalService.start(jobManagerLeaderListener);
}在该方法里面开始对jobmaster leader的监听，创建一个LeaderRetrievalService对象，并且调用其start方法，传入参数是JobManagerLeaderListener对象，也就是LeaderRetrievalListener对象，在ZooKeeperLeaderRetrievalService的start方法里面，通过NodeCache方式添加了监控当前节点变化的listener，也就是NodeCacheListener，当监听的节点发生变化则调用nodeChanged方法，而在该方法里面会调用LeaderRetrievalListener的notifyLeaderAddress方法，也就是JobManagerLeaderListener的notifyLeaderAddress方法，会重新与新的leader建立连接。
腾讯云开发者
