Examples with MigrationListener com.hazelcast.core.MigrationListener used on opensource projects

Example 1 with MigrationListener

use of com.hazelcast.core.MigrationListener in project hazelcast by hazelcast.

the class PartitionMigrationListenerTest method testAddMigrationListener_whenListenerRegisteredTwice.

@Test
public void testAddMigrationListener_whenListenerRegisteredTwice() {
    TestHazelcastInstanceFactory factory = createHazelcastInstanceFactory(2);
    HazelcastInstance hz1 = factory.newHazelcastInstance();
    PartitionService partitionService = hz1.getPartitionService();
    MigrationListener listener = mock(MigrationListener.class);
    String id1 = partitionService.addMigrationListener(listener);
    String id2 = partitionService.addMigrationListener(listener);
    // first we check if the registration id's are different
    assertNotEquals(id1, id2);
}

Example 2 with MigrationListener

use of com.hazelcast.core.MigrationListener in project hazelcast by hazelcast.

the class PartitionMigrationListenerTest method testRemoveMigrationListener.

@Test
public void testRemoveMigrationListener() {
    TestHazelcastInstanceFactory factory = createHazelcastInstanceFactory(2);
    HazelcastInstance hz1 = factory.newHazelcastInstance();
    PartitionService partitionService = hz1.getPartitionService();
    MigrationListener listener = mock(MigrationListener.class);
    String id = partitionService.addMigrationListener(listener);
    boolean removed = partitionService.removeMigrationListener(id);
    assertTrue(removed);
    // now we add a member
    HazelcastInstance hz2 = factory.newHazelcastInstance();
    warmUpPartitions(hz1, hz2);
    // and verify that the listener isn't called.
    verify(listener, never()).migrationStarted(any(MigrationEvent.class));
}

Example 3 with MigrationListener

use of com.hazelcast.core.MigrationListener in project hazelcast by hazelcast.

the class Node method initializeListeners.

@SuppressWarnings({ "checkstyle:npathcomplexity", "checkstyle:cyclomaticcomplexity" })
private void initializeListeners(Config config) {
    for (final ListenerConfig listenerCfg : config.getListenerConfigs()) {
        Object listener = listenerCfg.getImplementation();
        if (listener == null) {
            try {
                listener = ClassLoaderUtil.newInstance(configClassLoader, listenerCfg.getClassName());
            } catch (Exception e) {
                logger.severe(e);
            }
        }
        if (listener instanceof HazelcastInstanceAware) {
            ((HazelcastInstanceAware) listener).setHazelcastInstance(hazelcastInstance);
        }
        boolean known = false;
        if (listener instanceof DistributedObjectListener) {
            final ProxyServiceImpl proxyService = (ProxyServiceImpl) nodeEngine.getProxyService();
            proxyService.addProxyListener((DistributedObjectListener) listener);
            known = true;
        }
        if (listener instanceof MembershipListener) {
            clusterService.addMembershipListener((MembershipListener) listener);
            known = true;
        }
        if (listener instanceof MigrationListener) {
            partitionService.addMigrationListener((MigrationListener) listener);
            known = true;
        }
        if (listener instanceof PartitionLostListener) {
            partitionService.addPartitionLostListener((PartitionLostListener) listener);
            known = true;
        }
        if (listener instanceof LifecycleListener) {
            hazelcastInstance.lifecycleService.addLifecycleListener((LifecycleListener) listener);
            known = true;
        }
        if (listener instanceof ClientListener) {
            String serviceName = ClientEngineImpl.SERVICE_NAME;
            nodeEngine.getEventService().registerLocalListener(serviceName, serviceName, listener);
            known = true;
        }
        if (listener instanceof InternalMigrationListener) {
            final InternalPartitionServiceImpl partitionService = (InternalPartitionServiceImpl) nodeEngine.getPartitionService();
            partitionService.setInternalMigrationListener((InternalMigrationListener) listener);
            known = true;
        }
        if (nodeExtension.registerListener(listener)) {
            known = true;
        }
        if (listener != null && !known) {
            final String error = "Unknown listener type: " + listener.getClass();
            Throwable t = new IllegalArgumentException(error);
            logger.warning(error, t);
        }
    }
}