Java源码示例:org.apache.flink.runtime.leaderelection.TestingListener
示例1
@Test
public void testNotifyListenerLater() throws Exception {
final String localhost = "localhost";
settableLeaderRetrievalService.notifyListener(localhost, HighAvailabilityServices.DEFAULT_LEADER_ID);
final TestingListener listener = new TestingListener();
settableLeaderRetrievalService.start(listener);
assertThat(listener.getAddress(), equalTo(localhost));
assertThat(listener.getLeaderSessionID(), equalTo(HighAvailabilityServices.DEFAULT_LEADER_ID));
}
示例2
@Test
public void testNotifyListenerImmediately() throws Exception {
final TestingListener listener = new TestingListener();
settableLeaderRetrievalService.start(listener);
final String localhost = "localhost";
settableLeaderRetrievalService.notifyListener(localhost, HighAvailabilityServices.DEFAULT_LEADER_ID);
assertThat(listener.getAddress(), equalTo(localhost));
assertThat(listener.getLeaderSessionID(), equalTo(HighAvailabilityServices.DEFAULT_LEADER_ID));
}
示例3
private void runCleanupTest(
Configuration configuration,
TestingBlobStoreService blobStoreService,
ThrowingConsumer<ZooKeeperHaServices, Exception> zooKeeperHaServicesConsumer) throws Exception {
try (ZooKeeperHaServices zooKeeperHaServices = new ZooKeeperHaServices(
ZooKeeperUtils.startCuratorFramework(configuration),
Executors.directExecutor(),
configuration,
blobStoreService)) {
// create some Zk services to trigger the generation of paths
final LeaderRetrievalService resourceManagerLeaderRetriever = zooKeeperHaServices.getResourceManagerLeaderRetriever();
final LeaderElectionService resourceManagerLeaderElectionService = zooKeeperHaServices.getResourceManagerLeaderElectionService();
final RunningJobsRegistry runningJobsRegistry = zooKeeperHaServices.getRunningJobsRegistry();
final TestingListener listener = new TestingListener();
resourceManagerLeaderRetriever.start(listener);
resourceManagerLeaderElectionService.start(new TestingContender("foobar", resourceManagerLeaderElectionService));
final JobID jobId = new JobID();
runningJobsRegistry.setJobRunning(jobId);
listener.waitForNewLeader(2000L);
resourceManagerLeaderRetriever.stop();
resourceManagerLeaderElectionService.stop();
runningJobsRegistry.clearJob(jobId);
zooKeeperHaServicesConsumer.accept(zooKeeperHaServices);
}
}
示例4
@Test
public void testNotifyListenerLater() throws Exception {
final String localhost = "localhost";
settableLeaderRetrievalService.notifyListener(localhost, HighAvailabilityServices.DEFAULT_LEADER_ID);
final TestingListener listener = new TestingListener();
settableLeaderRetrievalService.start(listener);
assertThat(listener.getAddress(), equalTo(localhost));
assertThat(listener.getLeaderSessionID(), equalTo(HighAvailabilityServices.DEFAULT_LEADER_ID));
}
示例5
@Test
public void testNotifyListenerImmediately() throws Exception {
final TestingListener listener = new TestingListener();
settableLeaderRetrievalService.start(listener);
final String localhost = "localhost";
settableLeaderRetrievalService.notifyListener(localhost, HighAvailabilityServices.DEFAULT_LEADER_ID);
assertThat(listener.getAddress(), equalTo(localhost));
assertThat(listener.getLeaderSessionID(), equalTo(HighAvailabilityServices.DEFAULT_LEADER_ID));
}
示例6
private void runCleanupTest(
Configuration configuration,
TestingBlobStoreService blobStoreService,
ThrowingConsumer<ZooKeeperHaServices, Exception> zooKeeperHaServicesConsumer) throws Exception {
try (ZooKeeperHaServices zooKeeperHaServices = new ZooKeeperHaServices(
ZooKeeperUtils.startCuratorFramework(configuration),
Executors.directExecutor(),
configuration,
blobStoreService)) {
// create some Zk services to trigger the generation of paths
final LeaderRetrievalService resourceManagerLeaderRetriever = zooKeeperHaServices.getResourceManagerLeaderRetriever();
final LeaderElectionService resourceManagerLeaderElectionService = zooKeeperHaServices.getResourceManagerLeaderElectionService();
final RunningJobsRegistry runningJobsRegistry = zooKeeperHaServices.getRunningJobsRegistry();
final TestingListener listener = new TestingListener();
resourceManagerLeaderRetriever.start(listener);
resourceManagerLeaderElectionService.start(new TestingContender("foobar", resourceManagerLeaderElectionService));
final JobID jobId = new JobID();
runningJobsRegistry.setJobRunning(jobId);
listener.waitForNewLeader(2000L);
resourceManagerLeaderRetriever.stop();
resourceManagerLeaderElectionService.stop();
runningJobsRegistry.clearJob(jobId);
zooKeeperHaServicesConsumer.accept(zooKeeperHaServices);
}
}
示例7
@Test
public void testNotifyListenerLater() throws Exception {
final String localhost = "localhost";
settableLeaderRetrievalService.notifyListener(localhost, HighAvailabilityServices.DEFAULT_LEADER_ID);
final TestingListener listener = new TestingListener();
settableLeaderRetrievalService.start(listener);
assertThat(listener.getAddress(), equalTo(localhost));
assertThat(listener.getLeaderSessionID(), equalTo(HighAvailabilityServices.DEFAULT_LEADER_ID));
}
示例8
@Test
public void testNotifyListenerImmediately() throws Exception {
final TestingListener listener = new TestingListener();
settableLeaderRetrievalService.start(listener);
final String localhost = "localhost";
settableLeaderRetrievalService.notifyListener(localhost, HighAvailabilityServices.DEFAULT_LEADER_ID);
assertThat(listener.getAddress(), equalTo(localhost));
assertThat(listener.getLeaderSessionID(), equalTo(HighAvailabilityServices.DEFAULT_LEADER_ID));
}
示例9
private void runCleanupTest(
Configuration configuration,
TestingBlobStoreService blobStoreService,
ThrowingConsumer<ZooKeeperHaServices, Exception> zooKeeperHaServicesConsumer) throws Exception {
try (ZooKeeperHaServices zooKeeperHaServices = new ZooKeeperHaServices(
ZooKeeperUtils.startCuratorFramework(configuration),
Executors.directExecutor(),
configuration,
blobStoreService)) {
// create some Zk services to trigger the generation of paths
final LeaderRetrievalService resourceManagerLeaderRetriever = zooKeeperHaServices.getResourceManagerLeaderRetriever();
final LeaderElectionService resourceManagerLeaderElectionService = zooKeeperHaServices.getResourceManagerLeaderElectionService();
final RunningJobsRegistry runningJobsRegistry = zooKeeperHaServices.getRunningJobsRegistry();
final TestingListener listener = new TestingListener();
resourceManagerLeaderRetriever.start(listener);
resourceManagerLeaderElectionService.start(new TestingContender("foobar", resourceManagerLeaderElectionService));
final JobID jobId = new JobID();
runningJobsRegistry.setJobRunning(jobId);
listener.waitForNewLeader(2000L);
resourceManagerLeaderRetriever.stop();
resourceManagerLeaderElectionService.stop();
runningJobsRegistry.clearJob(jobId);
zooKeeperHaServicesConsumer.accept(zooKeeperHaServices);
}
}