Examples with TestAffinityFunction org.apache.ignite.internal.processors.cache.metric.SqlViewExporterSpiTest.TestAffinityFunction used on opensource projects

Example 1 with TestAffinityFunction

use of org.apache.ignite.internal.processors.cache.metric.SqlViewExporterSpiTest.TestAffinityFunction in project ignite by apache.

the class SystemViewCommandTest method testPartitionStates.

/**
 */
@Test
public void testPartitionStates() throws Exception {
    String nodeName0 = getTestIgniteInstanceName(0);
    String nodeName1 = getTestIgniteInstanceName(1);
    String nodeName2 = getTestIgniteInstanceName(2);
    IgniteCache<Integer, Integer> cache1 = ignite0.createCache(new CacheConfiguration<Integer, Integer>().setName("cache1").setCacheMode(CacheMode.PARTITIONED).setAffinity(new TestAffinityFunction(new String[][] { { nodeName0, nodeName1 }, { nodeName1, nodeName2 }, { nodeName2, nodeName0 } })));
    IgniteCache<Integer, Integer> cache2 = ignite0.createCache(new CacheConfiguration<Integer, Integer>().setName("cache2").setCacheMode(CacheMode.PARTITIONED).setAffinity(new TestAffinityFunction(new String[][] { { nodeName0, nodeName1, nodeName2 }, { nodeName1 } })));
    for (int i = 0; i < 100; i++) {
        cache1.put(i, i);
        cache2.put(i, i);
    }
    try (IgniteEx ignite2 = startGrid(nodeName2)) {
        ignite2.rebalanceEnabled(false);
        ignite0.cluster().setBaselineTopology(ignite0.cluster().topologyVersion());
        String nodeId0 = ignite0.cluster().localNode().id().toString();
        String nodeId1 = ignite(1).cluster().localNode().id().toString();
        String nodeId2 = ignite2.cluster().localNode().id().toString();
        String cacheGrpId1 = Integer.toString(ignite0.cachex("cache1").context().groupId());
        String cacheGrpId2 = Integer.toString(ignite0.cachex("cache2").context().groupId());
        String owningState = GridDhtPartitionState.OWNING.name();
        String movingState = GridDhtPartitionState.MOVING.name();
        for (int i = 0; i < 3; i++) {
            List<List<String>> partStatesView = systemView(ignite(i), PART_STATES_VIEW);
            // Check partitions for cache1.
            checkPartitionStatesView(partStatesView, owningState, true, cacheGrpId1, nodeId0, 0);
            checkPartitionStatesView(partStatesView, owningState, false, cacheGrpId1, nodeId1, 0);
            checkPartitionStatesView(partStatesView, owningState, true, cacheGrpId1, nodeId1, 1);
            checkPartitionStatesView(partStatesView, movingState, false, cacheGrpId1, nodeId2, 1);
            checkPartitionStatesView(partStatesView, owningState, true, cacheGrpId1, nodeId0, 2);
            checkPartitionStatesView(partStatesView, movingState, false, cacheGrpId1, nodeId2, 2);
            checkPartitionStatesView(partStatesView, owningState, true, cacheGrpId2, nodeId0, 0);
            checkPartitionStatesView(partStatesView, owningState, false, cacheGrpId2, nodeId1, 0);
            checkPartitionStatesView(partStatesView, movingState, false, cacheGrpId2, nodeId2, 0);
            checkPartitionStatesView(partStatesView, owningState, true, cacheGrpId2, nodeId1, 1);
        }
        AffinityTopologyVersion topVer = ignite0.context().discovery().topologyVersionEx();
        ignite2.rebalanceEnabled(true);
        // Wait until rebalance complete.
        assertTrue(waitForCondition(() -> ignite0.context().discovery().topologyVersionEx().compareTo(topVer) > 0, getTestTimeout()));
        for (int i = 0; i < 3; i++) {
            List<List<String>> rows = systemView(ignite(i), PART_STATES_VIEW);
            assertEquals(10, rows.stream().filter(row -> (cacheGrpId1.equals(row.get(0)) || cacheGrpId2.equals(row.get(0))) && owningState.equals(row.get(3))).count());
            assertTrue(rows.stream().noneMatch(row -> (cacheGrpId1.equals(row.get(0)) || cacheGrpId2.equals(row.get(0))) && movingState.equals(row.get(3))));
        }
        // Check that assignment is now changed to ideal.
        for (int i = 0; i < 3; i++) {
            List<List<String>> rows = systemView(ignite(i), PART_STATES_VIEW);
            checkPartitionStatesView(rows, owningState, false, cacheGrpId1, nodeId0, 2);
            checkPartitionStatesView(rows, owningState, true, cacheGrpId1, nodeId2, 2);
        }
    } finally {
        ignite0.cluster().setBaselineTopology(ignite0.cluster().topologyVersion());
    }
}