/*
* JBoss, Home of Professional Open Source
* Copyright 2010 Red Hat Inc. and/or its affiliates and other
* contributors as indicated by the @author tags. All rights reserved.
* See the copyright.txt in the distribution for a full listing of
* individual contributors.
*
* This is free software; you can redistribute it and/or modify it
* under the terms of the GNU Lesser General Public License as
* published by the Free Software Foundation; either version 2.1 of
* the License, or (at your option) any later version.
*
* This software is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public
* License along with this software; if not, write to the Free
* Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
* 02110-1301 USA, or see the FSF site: http://www.fsf.org.
*/
package org.infinispan.manager;
import org.infinispan.test.AbstractCacheTest;
import org.infinispan.test.TestingUtil;
import org.infinispan.test.fwk.TestCacheManagerFactory;
import org.testng.annotations.AfterMethod;
import org.testng.annotations.BeforeMethod;
import org.testng.annotations.Test;
import java.util.ArrayList;
import java.util.List;
import java.util.concurrent.Callable;
import java.util.concurrent.CyclicBarrier;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.concurrent.Future;
/**
* Test that verifies that CacheContainer.getCache() can be called concurrently.
*
* @author Galder Zamarreño
* @since 4.2
*/
@Test(groups = "functional", testName = "ConcurrentCacheManagerTest")
public class ConcurrentCacheManagerTest extends AbstractCacheTest {
static final int NUM_CACHES = 4;
static final int NUM_THREADS = 25;
private EmbeddedCacheManager cacheManager;
@BeforeMethod
protected void setup() throws Exception {
EmbeddedCacheManager manager = TestCacheManagerFactory.createCacheManager();
for (int i = 0; i < NUM_CACHES; i++) {
manager.defineConfiguration("cache" + i, TestCacheManagerFactory.getDefaultConfiguration(true));
}
cacheManager = manager;
}
@AfterMethod(alwaysRun = true)
protected void teardown() {
TestingUtil.killCacheManagers(cacheManager);
}
public void testConcurrentGetCacheCalls() throws Exception {
final CyclicBarrier barrier = new CyclicBarrier(NUM_THREADS + 1);
List<Future<Void>> futures = new ArrayList<Future<Void>>(NUM_THREADS);
ExecutorService executorService = Executors.newFixedThreadPool(NUM_THREADS);
for (int i = 0; i < NUM_THREADS; i++) {
log.debug("Schedule execution");
final String name = "cache" + (i % NUM_CACHES);
Future<Void> future = executorService.submit(new Callable<Void>(){
@Override
public Void call() throws Exception {
try {
barrier.await();
cacheManager.getCache(name).put("a", "b");
return null;
} catch (Throwable t) {
log.error("Got", t);
throw new RuntimeException(t);
} finally {
log.debug("Wait for all execution paths to finish");
barrier.await();
}
}
});
futures.add(future);
}
barrier.await(); // wait for all threads to be ready
barrier.await(); // wait for all threads to finish
log.debug("All threads finished, let's shutdown the executor and check whether any exceptions were reported");
for (Future<Void> future : futures) future.get();
executorService.shutdownNow();
}
}