/*
* JBoss, Home of Professional Open Source
* Copyright 2013 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 copyrighted material is made available to anyone wishing to use,
* modify, copy, or redistribute it subject to the terms and conditions
* of the GNU Lesser General Public License, v. 2.1.
* This program is distributed in the hope that it will be useful, but WITHOUT A
* 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,
* v.2.1 along with this distribution; if not, write to the Free Software
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
* MA 02110-1301, USA.
*/
package org.infinispan.distribution.rehash;
import org.infinispan.Cache;
import org.infinispan.configuration.cache.CacheMode;
import org.infinispan.configuration.cache.ConfigurationBuilder;
import org.infinispan.test.MultipleCacheManagersTest;
import org.infinispan.test.fwk.CleanupAfterMethod;
import org.infinispan.transaction.TransactionMode;
import org.testng.annotations.Test;
import java.util.concurrent.Callable;
import java.util.concurrent.Future;
import java.util.concurrent.TimeUnit;
import static org.testng.AssertJUnit.assertEquals;
/**
* Tests data loss during state transfer when the originator of a put operation becomes the primary owner of the
* modified key. See https://issues.jboss.org/browse/ISPN-3357
*
* @author Dan Berindei
*/
@Test(groups = "stress", testName = "distribution.rehash.NonTxPutIfAbsentDuringJoinStressTest")
@CleanupAfterMethod
public class NonTxPutIfAbsentDuringJoinStressTest extends MultipleCacheManagersTest {
private static final int NUM_WRITERS = 4;
private static final int NUM_ORIGINATORS = 2;
private static final int NUM_KEYS = 1000;
@Override
protected void createCacheManagers() throws Throwable {
ConfigurationBuilder c = getConfigurationBuilder();
addClusterEnabledCacheManager(c);
addClusterEnabledCacheManager(c);
waitForClusterToForm();
}
private ConfigurationBuilder getConfigurationBuilder() {
ConfigurationBuilder c = new ConfigurationBuilder();
c.clustering().cacheMode(CacheMode.DIST_SYNC);
c.transaction().transactionMode(TransactionMode.NON_TRANSACTIONAL);
return c;
}
public void testNodeJoiningDuringPutIfAbsent() throws Exception {
Future[] futures = new Future[NUM_WRITERS];
for (int i = 0; i < NUM_WRITERS; i++) {
final int finalI = i;
futures[i] = fork(new Callable() {
@Override
public Object call() throws Exception {
for (int j = 0; j < NUM_KEYS; j++) {
Cache<Object, Object> cache = cache(finalI % NUM_ORIGINATORS);
cache.putIfAbsent("key_" + finalI + "_" + j, "value_" + finalI + "_" + j);
}
return null;
}
});
}
addClusterEnabledCacheManager(getConfigurationBuilder());
waitForClusterToForm();
addClusterEnabledCacheManager(getConfigurationBuilder());
waitForClusterToForm();
TimeUnit.MILLISECONDS.sleep(NUM_KEYS * NUM_WRITERS);
for (int i = 0; i < NUM_WRITERS; i++) {
futures[i].get(10, TimeUnit.SECONDS);
for (int j = 0; j < NUM_KEYS; j++) {
for (int k = 0; k < caches().size(); k++) {
assertEquals(cache(k).get("key_" + i + "_" + j), "value_" + i + "_" + j);
}
}
}
}
}