Examples with LocalNodeMasterListener org.elasticsearch.cluster.LocalNodeMasterListener used on opensource projects

Example 1 with LocalNodeMasterListener

use of org.elasticsearch.cluster.LocalNodeMasterListener in project elasticsearch by elastic.

the class ClusterServiceTests method testLocalNodeMasterListenerCallbacks.

public void testLocalNodeMasterListenerCallbacks() throws Exception {
    TimedClusterService timedClusterService = createTimedClusterService(false);
    AtomicBoolean isMaster = new AtomicBoolean();
    timedClusterService.addLocalNodeMasterListener(new LocalNodeMasterListener() {

        @Override
        public void onMaster() {
            isMaster.set(true);
        }

        @Override
        public void offMaster() {
            isMaster.set(false);
        }

        @Override
        public String executorName() {
            return ThreadPool.Names.SAME;
        }
    });
    ClusterState state = timedClusterService.state();
    DiscoveryNodes nodes = state.nodes();
    DiscoveryNodes.Builder nodesBuilder = DiscoveryNodes.builder(nodes).masterNodeId(nodes.getLocalNodeId());
    state = ClusterState.builder(state).blocks(ClusterBlocks.EMPTY_CLUSTER_BLOCK).nodes(nodesBuilder).build();
    setState(timedClusterService, state);
    assertThat(isMaster.get(), is(true));
    nodes = state.nodes();
    nodesBuilder = DiscoveryNodes.builder(nodes).masterNodeId(null);
    state = ClusterState.builder(state).blocks(ClusterBlocks.builder().addGlobalBlock(DiscoverySettings.NO_MASTER_BLOCK_WRITES)).nodes(nodesBuilder).build();
    setState(timedClusterService, state);
    assertThat(isMaster.get(), is(false));
    nodesBuilder = DiscoveryNodes.builder(nodes).masterNodeId(nodes.getLocalNodeId());
    state = ClusterState.builder(state).blocks(ClusterBlocks.EMPTY_CLUSTER_BLOCK).nodes(nodesBuilder).build();
    setState(timedClusterService, state);
    assertThat(isMaster.get(), is(true));
    timedClusterService.close();
}