/*
* JBoss, Home of Professional Open Source
* Copyright 2006, 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 com.hp.mwtests.ts.jta.tools;
import javax.transaction.xa.XAException;
import javax.transaction.xa.XAResource;
import javax.transaction.xa.Xid;
import java.io.Serializable;
import java.util.HashMap;
import java.util.Map;
public class FailureXAResource implements XAResource, Serializable
{
public enum FailLocation { none, prepare, commit, rollback, end, prepare_and_rollback };
public enum FailType { normal, timeout, heurcom, nota, inval, proto, rmfail, rollback, XA_RBCOMMFAIL };
public FailureXAResource()
{
this(FailLocation.none, FailType.normal);
}
public FailureXAResource(FailLocation loc)
{
this(loc, FailType.normal);
}
public FailureXAResource(FailLocation loc, FailType type)
{
_locale = loc;
_type = type;
}
public void commit(Xid id, boolean onePhase) throws XAException
{
if (_locale == FailLocation.commit)
{
if (_type == FailType.normal)
throw new XAException(XAException.XA_HEURMIX);
if (_type == FailType.heurcom)
throw new XAException(XAException.XA_HEURCOM);
if (_type == FailType.rollback)
throw new XAException(XAException.XA_HEURRB);
if (_type == FailType.nota)
throw new XAException(XAException.XAER_NOTA);
if (_type == FailType.inval)
throw new XAException(XAException.XAER_INVAL);
if (_type == FailType.proto)
throw new XAException(XAException.XAER_PROTO);
if (_type == FailType.rmfail)
throw new XAException(XAException.XAER_RMFAIL);
throw new XAException(XAException.XA_RBTIMEOUT);
}
}
public void end(Xid xid, int flags) throws XAException
{
if (_locale == FailLocation.end)
{
if (_type == FailType.normal)
throw new XAException(XAException.XA_HEURRB);
if (_type == FailType.timeout)
throw new XAException(XAException.XA_RBTIMEOUT);
if (_type == FailType.XA_RBCOMMFAIL)
throw new XAException(XAException.XA_RBCOMMFAIL);
}
}
public void forget(Xid xid) throws XAException
{
XidInfo info = getXidInfo(xid);
info.forgetCount += 1;
if (info.refuseForget)
throw new XAException(XAException.XAER_RMERR);
}
public int getTransactionTimeout() throws XAException
{
return 0;
}
public boolean isSameRM(XAResource xares) throws XAException
{
return false;
}
public int prepare(Xid xid) throws XAException
{
if ((_locale == FailLocation.prepare) || (_locale == FailLocation.prepare_and_rollback))
throw new XAException(XAException.XAER_INVAL);
return XA_OK;
}
public Xid[] recover(int flag) throws XAException
{
return null;
}
public void rollback(Xid xid) throws XAException
{
if ((_locale == FailLocation.rollback) || (_locale == FailLocation.prepare_and_rollback))
{
if (_type == FailType.normal)
throw new XAException(XAException.XA_HEURMIX);
if (_type == FailType.heurcom)
throw new XAException(XAException.XA_HEURCOM);
if (_type == FailType.rollback)
throw new XAException(XAException.XA_HEURRB);
if (_type == FailType.nota)
throw new XAException(XAException.XAER_NOTA);
if (_type == FailType.inval)
throw new XAException(XAException.XAER_INVAL);
if (_type == FailType.proto)
throw new XAException(XAException.XAER_PROTO);
if (_type == FailType.rmfail)
throw new XAException(XAException.XAER_RMFAIL);
throw new XAException(XAException.XA_HEURHAZ);
}
}
public boolean setTransactionTimeout(int seconds) throws XAException
{
return true;
}
public void start(Xid xid, int flags) throws XAException
{
this._xid = xid;
}
public Xid getXid() {
return _xid;
}
public int getForgetCount(Xid xid) {
return getXidInfo(xid).forgetCount;
}
public void setRefuseForget(Xid xid, boolean refuseForget) {
getXidInfo(xid).refuseForget = refuseForget;
}
public static void resetForgetCounts() {
forgetCounts.clear();
}
private XidInfo getXidInfo(Xid xid) {
XidInfo info = forgetCounts.get(xid);
if (info == null) {
info = new XidInfo();
forgetCounts.put(xid, info);
}
return info;
}
private FailLocation _locale;
private FailType _type;
private Xid _xid;
static private Map<Xid, XidInfo> forgetCounts = new HashMap<>();
static private class XidInfo {
int forgetCount;
boolean refuseForget;
XidInfo() {
this(0, false);
}
XidInfo(int forgetCount, boolean refuseForget) {
this.forgetCount = forgetCount;
this.refuseForget = refuseForget;
}
}
}