/*
* JBoss, Home of Professional Open Source
* Copyright 2006, JBoss Inc., and individual contributors as indicated
* by the @authors tag. 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.jboss.stm;
import java.util.Random;
import org.jboss.stm.annotations.Transactional;
import org.jboss.stm.annotations.ReadLock;
import org.jboss.stm.annotations.State;
import org.jboss.stm.annotations.WriteLock;
import org.jboss.stm.internal.PersistentContainer;
import org.jboss.stm.internal.RecoverableContainer;
import com.arjuna.ats.arjuna.AtomicAction;
import com.arjuna.ats.arjuna.ObjectModel;
import junit.framework.TestCase;
/**
* Hammer equivalent test.
*
* @author Mark Little
*/
public class ExtendedHammerUnitTest extends TestCase
{
private static final int NUMBER_OF_CLIENTS = 2;
private static final int ITERS = 20;
@Transactional
public interface Sample
{
public void increment ();
public void decrement ();
public int value ();
}
public class SampleLockable implements Sample
{
public SampleLockable (int init)
{
_isState = init;
}
@ReadLock
public int value ()
{
return _isState;
}
@WriteLock
public void increment ()
{
_isState++;
}
@WriteLock
public void decrement ()
{
_isState--;
}
@State
private int _isState;
}
public class Worker extends Thread
{
public Worker (Sample obj1, Sample obj2)
{
_obj1 = obj1;
_obj2 = obj2;
}
public void run ()
{
Random rand = new Random();
for (int i = 0; i < ITERS; i++)
{
AtomicAction A = new AtomicAction();
boolean doCommit = true;
Thread.yield();
A.begin();
try
{
// always keep the two objects in sync.
_obj1.increment();
_obj2.decrement();
}
catch (final Throwable ex)
{
doCommit = false;
}
if (rand.nextInt() % 2 == 0)
doCommit = false;
if (doCommit)
A.commit();
else
A.abort();
Thread.yield();
try
{
Thread.sleep((int) rand.nextFloat()*1000);
}
catch (InterruptedException e)
{
}
}
}
private Sample _obj1;
private Sample _obj2;
}
public void testRecoverableHammer ()
{
RecoverableContainer<Sample> theContainer = new RecoverableContainer<Sample>();
Sample obj1 = theContainer.enlist(new SampleLockable(10));
Sample obj2 = theContainer.enlist(new SampleLockable(10));
Worker[] workers = new Worker[NUMBER_OF_CLIENTS];
for (int i = 0; i < NUMBER_OF_CLIENTS; i++)
workers[i] = new Worker(obj1, obj2);
for (int j = 0; j < NUMBER_OF_CLIENTS; j++)
workers[j].start();
try
{
for (int k = 0; k < NUMBER_OF_CLIENTS; k++)
workers[k].join();
}
catch (final Throwable ex)
{
}
assertEquals(obj1.value()+obj2.value(), 20);
}
public void testPersistentHammer ()
{
PersistentContainer<Sample> theContainer = new PersistentContainer<Sample>();
Sample obj1 = theContainer.enlist(new SampleLockable(10));
Sample obj2 = theContainer.enlist(new SampleLockable(10));
Worker[] workers = new Worker[NUMBER_OF_CLIENTS];
for (int i = 0; i < NUMBER_OF_CLIENTS; i++)
workers[i] = new Worker(obj1, obj2);
for (int j = 0; j < NUMBER_OF_CLIENTS; j++)
workers[j].start();
try
{
for (int k = 0; k < NUMBER_OF_CLIENTS; k++)
workers[k].join();
}
catch (final Throwable ex)
{
}
assertEquals(obj1.value()+obj2.value(), 20);
}
public void testPersistentHammerMULTIPLE ()
{
PersistentContainer<Sample> theContainer = new PersistentContainer<Sample>(ObjectModel.MULTIPLE);
Sample obj1 = theContainer.enlist(new SampleLockable(10));
Sample obj2 = theContainer.enlist(new SampleLockable(10));
Worker[] workers = new Worker[NUMBER_OF_CLIENTS];
for (int i = 0; i < NUMBER_OF_CLIENTS; i++)
workers[i] = new Worker(obj1, obj2);
for (int j = 0; j < NUMBER_OF_CLIENTS; j++)
workers[j].start();
try
{
for (int k = 0; k < NUMBER_OF_CLIENTS; k++)
workers[k].join();
}
catch (final Throwable ex)
{
}
assertEquals(obj1.value()+obj2.value(), 20);
}
}