/*
 * JBoss, Home of Professional Open Source
 * Copyright 2007, Red Hat Middleware LLC, and individual contributors
 * as indicated by the @author tags.
 * 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.
 *
 * (C) 2005-2006,
 * @author JBoss Inc.
 */
package org.jboss.jbossts.qa.ArjunaCore.LockManager.impl;

import com.arjuna.ats.arjuna.AtomicAction;
import com.arjuna.ats.arjuna.ObjectType;
import com.arjuna.ats.arjuna.common.Uid;
import com.arjuna.ats.arjuna.state.InputObjectState;
import com.arjuna.ats.arjuna.state.OutputObjectState;
import com.arjuna.ats.txoj.Lock;
import com.arjuna.ats.txoj.LockMode;
import com.arjuna.ats.txoj.LockResult;
import org.jboss.jbossts.qa.ArjunaCore.Utils.qautil;
import org.jboss.jbossts.qa.Utils.JVMStats;

/**
 * Simple record used to test AtomicAction
 */
public class TXBasicLockRecord2 extends TXBasicLockRecord
{
	public TXBasicLockRecord2()
	{
		super(ObjectType.ANDPERSISTENT);
	}

	public TXBasicLockRecord2(int id)
	{
		super(ObjectType.ANDPERSISTENT);
		mId = id;
	}

	public TXBasicLockRecord2(Uid oldId)
	{
		super(oldId, ObjectType.ANDPERSISTENT);
	}

	public int increase()
	{
		return increase(0);
	}

	public int increase(int retry)
	{
		return increase(retry, 0);
	}

	/**
	 * We will start a subtrancastion during the increase to see what effet this has.
	 *
	 * @return +1 if tx was committed, 0 if tx was not committed
	 */
	public int increase(int retry, int wait_time)
	{
		int returnValue = 0;
		AtomicAction a = new AtomicAction();
		a.begin();
		try
		{
			int locking_result = LockResult.REFUSED;
			int locking_attempt_count = 0;
			Lock lck = new Lock(LockMode.WRITE);
			do
			{
				locking_result = setlock(lck, retry, wait_time);

				if (locking_result == LockResult.GRANTED)
				{
					mValue++;
				}
				else
				{
					locking_attempt_count++;
				}
			}
			while ((locking_result != LockResult.GRANTED) && (locking_attempt_count < mLimit));

			if (locking_result != LockResult.GRANTED)
			{
				qautil.qadebug("trying to get lock for " + mLimit + "th time");
				a.abort();
			}
			else
			{
				a.commit();
				returnValue = 1;
			}
		}
		catch (Exception e)
		{
			a.abort();
			qautil.debug("exception in increase method ", e);
		}
		return returnValue;
	}

	/**
	 * Call to getValue with default value of 10 for retry time.
	 */
	public int getValue()
	{
		return getValue(1);
	}

	public int getValue(int retry)
	{
		return getValue(retry, 10);
	}

	public int getValue(int retry, int wait_time)
	{
		int return_value = 0;

		AtomicAction a = new AtomicAction();
		a.begin();
		try
		{
			int locking_result = LockResult.REFUSED;
			int locking_attempt_count = 0;
			Lock lck = new Lock(LockMode.READ);
			do
			{
				locking_result = setlock(lck, retry, wait_time);

				if (locking_result == LockResult.GRANTED)
				{
					return_value = mValue;
				}
				else
				{
					locking_attempt_count++;
				}
			}
			while ((locking_result != LockResult.GRANTED) && (locking_attempt_count < mLimit));

			if (locking_result != LockResult.GRANTED)
			{
				qautil.qadebug("trying to get lock for " + mLimit + "th time");
			}

			a.commit();
		}
		catch (Exception e)
		{
			a.abort();
			qautil.debug("exception in get method ", e);
		}

		return return_value;
	}

	/**
	 * Method used to check for any memory leaks.
	 */
	public int getMemory()
	{
		return (int) JVMStats.getMemory();
	}

	public boolean save_state(OutputObjectState objectState, int objectType)
	{
		super.save_state(objectState, objectType);
		try
		{
			objectState.packInt(mValue);
			return true;
		}
		catch (Exception exception)
		{
			qautil.debug("TXBasicLockRecord.save_state: ", exception);
			return false;
		}
	}

	public boolean restore_state(InputObjectState objectState, int objectType)
	{
		qautil.qadebug("restore state of " + this + " = " + mValue);
		super.restore_state(objectState, objectType);
		try
		{
			mValue = objectState.unpackInt();
			return true;
		}
		catch (Exception exception)
		{
			qautil.debug("TXBasicLockRecord.restore_state: ", exception);
			return false;
		}
	}

	public String type()
	{
		return "/StateManager/LockManager/TXBasicLockRecord";
	}

	private int mValue = 0;
	private int mLimit = 1000;
	private int mId = 0;
}

