| /* |
| * Licensed to the Apache Software Foundation (ASF) under one or more |
| * contributor license agreements. See the NOTICE file distributed with |
| * this work for additional information regarding copyright ownership. |
| * The ASF licenses this file to You under the Apache License, Version 2.0 |
| * (the "License"); you may not use this file except in compliance with |
| * the License. You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| |
| package org.apache.ignite.internal.processors.cache.distributed.dht; |
| |
| import java.util.concurrent.Callable; |
| import java.util.concurrent.CountDownLatch; |
| import org.apache.ignite.Ignite; |
| import org.apache.ignite.cache.CacheMode; |
| import org.apache.ignite.cache.affinity.rendezvous.RendezvousAffinityFunction; |
| import org.apache.ignite.configuration.CacheConfiguration; |
| import org.apache.ignite.configuration.IgniteConfiguration; |
| import org.apache.ignite.events.Event; |
| import org.apache.ignite.events.EventType; |
| import org.apache.ignite.failure.NoOpFailureHandler; |
| import org.apache.ignite.internal.util.typedef.G; |
| import org.apache.ignite.lang.IgnitePredicate; |
| import org.apache.ignite.testframework.GridTestUtils; |
| import org.apache.ignite.testframework.junits.common.GridCommonAbstractTest; |
| import org.jetbrains.annotations.Nullable; |
| import org.junit.Test; |
| |
| /** |
| * MultiThreaded load test for DHT preloader. |
| */ |
| public class GridCacheDhtPreloadMultiThreadedSelfTest extends GridCommonAbstractTest { |
| /** |
| * Creates new test. |
| */ |
| public GridCacheDhtPreloadMultiThreadedSelfTest() { |
| super(false); |
| } |
| |
| /** |
| * @throws Exception If failed. |
| */ |
| @Test |
| public void testNodeLeaveBeforePreloadingComplete() throws Exception { |
| try { |
| final CountDownLatch startLatch = new CountDownLatch(1); |
| |
| final CountDownLatch stopLatch = new CountDownLatch(1); |
| |
| GridTestUtils.runMultiThreadedAsync( |
| new Callable<Object>() { |
| @Nullable @Override public Object call() throws Exception { |
| Ignite g = startGrid("first"); |
| |
| g.events().localListen( |
| new IgnitePredicate<Event>() { |
| @Override public boolean apply(Event evt) { |
| stopLatch.countDown(); |
| |
| return true; |
| } |
| }, |
| EventType.EVT_NODE_JOINED); |
| |
| startLatch.countDown(); |
| |
| stopLatch.await(); |
| |
| G.stop(g.name(), false); |
| |
| return null; |
| } |
| }, |
| 1, |
| "first" |
| ); |
| |
| GridTestUtils.runMultiThreaded( |
| new Callable<Object>() { |
| @Nullable @Override public Object call() throws Exception { |
| startLatch.await(); |
| |
| startGrid("second"); |
| |
| return null; |
| } |
| }, |
| 1, |
| "second" |
| ); |
| } |
| finally { |
| // Intentionally used this method. See startGrid(String, String). |
| G.stopAll(false); |
| } |
| } |
| |
| /** |
| * @throws Exception If failed. |
| */ |
| @Test |
| public void testConcurrentNodesStart() throws Exception { |
| try { |
| multithreadedAsync( |
| new Callable<Object>() { |
| @Nullable @Override public Object call() throws Exception { |
| IgniteConfiguration cfg = loadConfiguration("modules/core/src/test/config/spring-multicache.xml"); |
| |
| cfg.setGridLogger(getTestResources().getLogger()); |
| |
| startGrid(Thread.currentThread().getName(), cfg); |
| |
| return null; |
| } |
| }, |
| 4, |
| "starter" |
| ).get(); |
| } |
| finally { |
| G.stopAll(true); |
| } |
| } |
| |
| /** |
| * @throws Exception If failed. |
| */ |
| @Test |
| public void testConcurrentNodesStartStop() throws Exception { |
| try { |
| multithreadedAsync( |
| new Callable<Object>() { |
| @Nullable @Override public Object call() throws Exception { |
| String igniteInstanceName = "grid-" + Thread.currentThread().getName(); |
| |
| startGrid(igniteInstanceName, "modules/core/src/test/config/example-cache.xml"); |
| |
| // Immediately stop the grid. |
| stopGrid(igniteInstanceName); |
| |
| return null; |
| } |
| }, |
| 6, |
| "tester" |
| ).get(); |
| } |
| finally { |
| G.stopAll(true); |
| } |
| } |
| |
| /** {@inheritDoc} */ |
| @Override protected IgniteConfiguration getConfiguration(String igniteInstanceName) throws Exception { |
| IgniteConfiguration cfg = loadConfiguration("modules/core/src/test/config/spring-multicache.xml"); |
| |
| cfg.setGridLogger(getTestResources().getLogger()); |
| |
| cfg.setIgniteInstanceName(igniteInstanceName); |
| |
| cfg.setFailureHandler(new NoOpFailureHandler()); |
| |
| for (CacheConfiguration cCfg : cfg.getCacheConfiguration()) { |
| if (cCfg.getCacheMode() == CacheMode.PARTITIONED) { |
| cCfg.setAffinity(new RendezvousAffinityFunction(2048, null)); |
| cCfg.setBackups(1); |
| } |
| } |
| |
| return cfg; |
| } |
| } |