/*
* 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.container.versioning;
import org.infinispan.Cache;
import org.infinispan.commands.VisitableCommand;
import org.infinispan.commands.tx.CommitCommand;
import org.infinispan.configuration.cache.CacheMode;
import org.infinispan.context.Flag;
import org.infinispan.context.InvocationContext;
import org.infinispan.context.impl.TxInvocationContext;
import org.infinispan.distribution.DistributionTestHelper;
import org.infinispan.distribution.MagicKey;
import org.infinispan.interceptors.InvocationContextInterceptor;
import org.infinispan.interceptors.base.CommandInterceptor;
import org.infinispan.test.fwk.CleanupAfterMethod;
import org.testng.annotations.Test;
import javax.transaction.RollbackException;
import javax.transaction.Transaction;
@Test(testName = "container.versioning.DistWriteSkewTest", groups = "functional")
@CleanupAfterMethod
public class DistWriteSkewTest extends AbstractClusteredWriteSkewTest {
@Override
protected CacheMode getCacheMode() {
return CacheMode.DIST_SYNC;
}
@Override
protected int clusterSize() {
return 4;
}
public void testWriteSkew() throws Exception {
Cache<Object, Object> cache0 = cache(0);
Cache<Object, Object> cache1 = cache(1);
Cache<Object, Object> cache2 = cache(2);
Cache<Object, Object> cache3 = cache(3);
MagicKey hello = new MagicKey(cache(2), "hello");
// Auto-commit is true
cache1.put(hello, "world 1");
tm(1).begin();
assert "world 1".equals(cache1.get(hello));
Transaction t = tm(1).suspend();
// Induce a write skew
cache3.put(hello, "world 3");
assert cache0.get(hello).equals("world 3");
assert cache1.get(hello).equals("world 3");
assert cache2.get(hello).equals("world 3");
assert cache3.get(hello).equals("world 3");
tm(1).resume(t);
cache1.put(hello, "world 2");
try {
tm(1).commit();
assert false : "Transaction should roll back";
} catch (RollbackException re) {
// expected
}
assert "world 3".equals(cache0.get(hello));
assert "world 3".equals(cache1.get(hello));
assert "world 3".equals(cache2.get(hello));
assert "world 3".equals(cache3.get(hello));
}
public void testWriteSkewOnNonOwner() throws Exception {
Cache<Object, Object> cache0 = cache(0);
Cache<Object, Object> cache1 = cache(1);
Cache<Object, Object> cache2 = cache(2);
Cache<Object, Object> cache3 = cache(3);
MagicKey hello = new MagicKey(cache(0), "hello"); // Owned by cache0 and cache1
int owners[] = {0, 0};
int nonOwners[] = {0, 0};
int j=0, k = 0;
for (int i=0; i<4; i++) {
if (DistributionTestHelper.isOwner(cache(i), hello))
owners[j++] = i;
else
nonOwners[k++] = i;
}
// Auto-commit is true
cache(owners[1]).put(hello, "world 1");
tm(nonOwners[0]).begin();
assert "world 1".equals(cache(nonOwners[0]).get(hello));
Transaction t = tm(nonOwners[0]).suspend();
// Induce a write skew
cache(nonOwners[1]).put(hello, "world 3");
assert cache0.get(hello).equals("world 3");
assert cache1.get(hello).equals("world 3");
assert cache2.get(hello).equals("world 3");
assert cache3.get(hello).equals("world 3");
tm(nonOwners[0]).resume(t);
cache(nonOwners[0]).put(hello, "world 2");
try {
tm(nonOwners[0]).commit();
assert false : "Transaction should roll back";
} catch (RollbackException re) {
// expected
}
assert "world 3".equals(cache0.get(hello));
assert "world 3".equals(cache1.get(hello));
assert "world 3".equals(cache2.get(hello));
assert "world 3".equals(cache3.get(hello));
}
public void testWriteSkewMultiEntries() throws Exception {
Cache<Object, Object> cache0 = cache(0);
Cache<Object, Object> cache1 = cache(1);
Cache<Object, Object> cache2 = cache(2);
Cache<Object, Object> cache3 = cache(3);
MagicKey hello = new MagicKey(cache(2), "hello");
MagicKey hello2 = new MagicKey(cache(3), "hello2");
MagicKey hello3 = new MagicKey(cache(0), "hello3");
tm(1).begin();
cache1.put(hello, "world 1");
cache1.put(hello2, "world 1");
cache1.put(hello3, "world 1");
tm(1).commit();
tm(1).begin();
cache1.put(hello2, "world 2");
cache1.put(hello3, "world 2");
assert "world 1".equals(cache1.get(hello));
assert "world 2".equals(cache1.get(hello2));
assert "world 2".equals(cache1.get(hello3));
Transaction t = tm(1).suspend();
// Induce a write skew
// Auto-commit is true
cache3.put(hello, "world 3");
for (Cache<Object, Object> c : caches()) {
assert "world 3".equals(c.get(hello));
assert "world 1".equals(c.get(hello2));
assert "world 1".equals(c.get(hello3));
}
tm(1).resume(t);
cache1.put(hello, "world 2");
try {
tm(1).commit();
assert false : "Transaction should roll back";
} catch (RollbackException re) {
// expected
}
for (Cache<Object, Object> c : caches()) {
assert "world 3".equals(c.get(hello));
assert "world 1".equals(c.get(hello2));
assert "world 1".equals(c.get(hello3));
}
}
public void testNullEntries() throws Exception {
Cache<Object, Object> cache0 = cache(0);
Cache<Object, Object> cache1 = cache(1);
Cache<Object, Object> cache2 = cache(2);
Cache<Object, Object> cache3 = cache(3);
MagicKey hello = new MagicKey(cache(2), "hello");
// Auto-commit is true
cache0.put(hello, "world");
tm(0).begin();
assert "world".equals(cache0.get(hello));
Transaction t = tm(0).suspend();
cache1.remove(hello);
assert null == cache0.get(hello);
assert null == cache1.get(hello);
assert null == cache2.get(hello);
assert null == cache3.get(hello);
tm(0).resume(t);
cache0.put(hello, "world2");
try {
tm(0).commit();
assert false : "This transaction should roll back";
} catch (RollbackException expected) {
// expected
}
assert null == cache0.get(hello);
assert null == cache1.get(hello);
assert null == cache2.get(hello);
assert null == cache3.get(hello);
}
public void testResendPrepare() throws Exception {
Cache<Object, Object> cache0 = cache(0);
Cache<Object, Object> cache1 = cache(1);
Cache<Object, Object> cache2 = cache(2);
Cache<Object, Object> cache3 = cache(3);
MagicKey hello = new MagicKey(cache(2), "hello");
// Auto-commit is true
cache0.put(hello, "world");
// create a write skew
tm(2).begin();
assert "world".equals(cache2.get(hello));
Transaction t = tm(2).suspend();
// Set up cache-3 to force the prepare to retry
cache(3).getAdvancedCache().addInterceptorAfter(new CommandInterceptor() {
boolean used = false;
@Override
public Object visitCommitCommand(TxInvocationContext ctx, CommitCommand c) throws Throwable {
if (!used) {
used = true;
return CommitCommand.RESEND_PREPARE;
} else {
return invokeNextInterceptor(ctx, c);
}
}
@Override
protected Object handleDefault(InvocationContext ctx, VisitableCommand command) throws Throwable {
return super.handleDefault(ctx, command);
}
}, InvocationContextInterceptor.class);
// Implicit tx. Prepare should be retried.
cache(0).put(hello, "world 2");
assert cache0.get(hello).equals("world 2");
assert cache1.get(hello).equals("world 2");
assert cache2.get(hello).equals("world 2");
assert cache3.get(hello).equals("world 2");
tm(2).resume(t);
cache2.put(hello, "world 3");
try {
tm(2).commit();
assert false : "This transaction should roll back";
} catch (RollbackException expected) {
// expected
}
assert cache0.get(hello).equals("world 2");
assert cache1.get(hello).equals("world 2");
assert cache2.get(hello).equals("world 2");
assert cache3.get(hello).equals("world 2");
}
public void testLocalOnlyPut() {
localOnlyPut(this.<Integer, String>cache(0), 1, "v1");
localOnlyPut(this.<Integer, String>cache(1), 2, "v2");
localOnlyPut(this.<Integer, String>cache(2), 3, "v3");
localOnlyPut(this.<Integer, String>cache(3), 4, "v4");
}
private void localOnlyPut(Cache<Integer, String> cache, Integer k, String v) {
cache.getAdvancedCache().withFlags(Flag.CACHE_MODE_LOCAL).put(k, v);
}
}