/*
* Copyright 2011 Red Hat, Inc. and/or its affiliates.
*
* 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 library; if not, write to the Free Software
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
* 02110-1301 USA
*/
package org.infinispan.interceptors;
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;
import org.infinispan.interceptors.base.CommandInterceptor;
import org.infinispan.util.logging.Log;
import org.infinispan.util.logging.LogFactory;
import org.testng.annotations.Test;
/**
* Tests {@link InterceptorChain} logic
*
* @author Galder ZamarreƱo
* @author Sanne Grinovero
* @since 5.1
*/
@Test(groups = "functional", testName = "interceptors.InterceptorChainTest")
public class InterceptorChainTest {
private static final Log log = LogFactory.getLog(InterceptorChainTest.class);
public void testConcurrentAddRemove() throws Exception {
InterceptorChain ic = new InterceptorChain(new CallInterceptor());
ic.addInterceptor(new ActivationInterceptor(), 1);
CyclicBarrier barrier = new CyclicBarrier(4);
List<Future<Void>> futures = new ArrayList<Future<Void>>(2);
ExecutorService executorService = Executors.newFixedThreadPool(3);
try {
// We do test concurrent add/remove of different types per thread,
// so that the final result is predictable (testable) and that we
// can not possibly fail because of the InterceptorChain checking
// that no interceptor is ever added twice.
futures.add(executorService.submit(new InterceptorChainUpdater(ic, barrier, new CacheMgmtInterceptor())));
futures.add(executorService.submit(new InterceptorChainUpdater(ic, barrier, new DistCacheStoreInterceptor())));
futures.add(executorService.submit(new InterceptorChainUpdater(ic, barrier, new InvalidationInterceptor())));
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();
} finally {
executorService.shutdownNow();
}
assert ic.containsInterceptorType(CallInterceptor.class);
assert ic.containsInterceptorType(ActivationInterceptor.class);
assert ic.containsInterceptorType(CacheMgmtInterceptor.class);
assert ic.containsInterceptorType(DistCacheStoreInterceptor.class);
assert ic.containsInterceptorType(InvalidationInterceptor.class);
assert ic.asList().size() == 5 : "Resulting interceptor chain was actually " + ic.asList();
}
private static class InterceptorChainUpdater implements Callable<Void> {
private final InterceptorChain ic;
private final CyclicBarrier barrier;
private final CommandInterceptor commandToExcercise;
InterceptorChainUpdater(InterceptorChain ic, CyclicBarrier barrier, CommandInterceptor commandToExcercise) {
this.ic = ic;
this.barrier = barrier;
this.commandToExcercise = commandToExcercise;
}
@Override
public Void call() throws Exception {
final Class<? extends CommandInterceptor> commandClass = commandToExcercise.getClass();
try {
log.debug("Wait for all executions paths to be ready to perform calls");
barrier.await();
// test in a loop as the barrier is otherwise not enough to make sure
// the different testing threads actually do make changes concurrently
// 2000 is still almost nothing in terms of testsuite time.
for (int i = 0; i < 2000; i++) {
ic.removeInterceptor(commandClass);
ic.addInterceptor(commandToExcercise, 1);
}
return null;
} finally {
log.debug("Wait for all execution paths to finish");
barrier.await();
}
}
}
}