/** * @param cacheName Cache name. * @param expNodes Expected number of nodes per partition. */ private void checkCache(String cacheName, int expNodes) { log.info("Test cache: " + cacheName); Affinity<Object> aff0 = ignite(0).affinity(cacheName); Ignite client = ignite(NODE_CNT - 1); assertTrue(client.configuration().isClientMode()); ClusterNode clientNode = client.cluster().localNode(); for (int i = 0; i < NODE_CNT; i++) { Ignite ignite = ignite(i); Affinity<Object> aff = ignite.affinity(cacheName); for (int part = 0; part < aff.partitions(); part++) { Collection<ClusterNode> nodes = aff.mapPartitionToPrimaryAndBackups(part); assertEquals(expNodes, nodes.size()); assertEquals(aff0.mapPartitionToPrimaryAndBackups(part), nodes); assertFalse(nodes.contains(clientNode)); assertEquals(aff0.partition(part), aff.partition(part)); TestKey key = new TestKey(part, part + 1); assertEquals(aff0.partition(key), aff.partition(key)); assertEquals(aff0.mapKeyToPrimaryAndBackups(key), aff.mapKeyToPrimaryAndBackups(key)); } } }
/** * @throws Exception If failed. */ @Test public void testClientNodeNotInAffinity() throws Exception { checkCache(CACHE1, 2); checkCache(CACHE2, 2); checkCache(CACHE4, 3); checkCache(CACHE5, 2); Ignite client = ignite(NODE_CNT - 1); CacheConfiguration ccfg = new CacheConfiguration(DEFAULT_CACHE_NAME); ccfg.setBackups(0); ccfg.setNodeFilter(new TestNodesFilter()); IgniteCache<Integer, Integer> cache = client.createCache(ccfg); try { checkCache(DEFAULT_CACHE_NAME, 1); } finally { cache.destroy(); } cache = client.createCache(ccfg, new NearCacheConfiguration()); try { checkCache(DEFAULT_CACHE_NAME, 1); } finally { cache.destroy(); } }