Examples with TxInfo com.arjuna.ats.internal.jta.xa.TxInfo used on opensource projects

Example 6 with TxInfo

use of com.arjuna.ats.internal.jta.xa.TxInfo in project narayana by jbosstm.

the class TransactionImple method endSuspendedRMs.

/**
 * If there are any suspended RMs then we should call end on them before the
 * transaction is terminated.
 */
protected boolean endSuspendedRMs() {
    boolean result = true;
    if (_suspendCount > 0) {
        Enumeration el = _resources.keys();
        if (el != null) {
            try {
                while (el.hasMoreElements()) {
                    /*
						 * Get the XAResource in case we have to call end on it.
						 */
                    XAResource xaRes = (XAResource) el.nextElement();
                    TxInfo info = (TxInfo) _resources.get(xaRes);
                    if (info.getState() == TxInfo.ASSOCIATION_SUSPENDED) {
                        if (XAUtils.mustEndSuspendedRMs(xaRes))
                            xaRes.start(info.xid(), XAResource.TMRESUME);
                        xaRes.end(info.xid(), XAResource.TMSUCCESS);
                        info.setState(TxInfo.NOT_ASSOCIATED);
                    }
                }
            } catch (XAException ex) {
                jtaLogger.i18NLogger.warn_transaction_arjunacore_xaenderror(ex);
                result = false;
            }
        }
        /*
			 * need to do the same for all duplicated resources
			 */
        el = _duplicateResources.keys();
        if (el != null) {
            try {
                while (el.hasMoreElements()) {
                    /*
						 * Get the XAResource in case we have to call end on it.
						 */
                    XAResource xaRes = (XAResource) el.nextElement();
                    TxInfo info = (TxInfo) _duplicateResources.get(xaRes);
                    if (info.getState() == TxInfo.ASSOCIATION_SUSPENDED) {
                        if (XAUtils.mustEndSuspendedRMs(xaRes))
                            xaRes.start(info.xid(), XAResource.TMRESUME);
                        xaRes.end(info.xid(), XAResource.TMSUCCESS);
                        info.setState(TxInfo.NOT_ASSOCIATED);
                    }
                }
            } catch (XAException ex) {
                jtaLogger.i18NLogger.warn_transaction_arjunacore_xaenderror(ex);
                result = false;
            }
        }
        _suspendCount = 0;
    }
    return result;
}

Example 7 with TxInfo

use of com.arjuna.ats.internal.jta.xa.TxInfo in project narayana by jbosstm.

the class TransactionImple method enlistResource.

public boolean enlistResource(XAResource xaRes, Object[] params) throws RollbackException, IllegalStateException, javax.transaction.SystemException {
    if (jtaLogger.logger.isTraceEnabled()) {
        jtaLogger.logger.trace("TransactionImple.enlistResource ( " + xaRes + " )");
    }
    if (xaRes == null)
        throw new javax.transaction.SystemException("TransactionImple.enlistResource - " + jtaLogger.i18NLogger.get_transaction_arjunacore_nullres());
    int status = getStatus();
    switch(status) {
        case javax.transaction.Status.STATUS_MARKED_ROLLBACK:
            throw new RollbackException("TransactionImple.enlistResource - " + jtaLogger.i18NLogger.get_transaction_arjunacore_invalidstate());
        case javax.transaction.Status.STATUS_ACTIVE:
            break;
        default:
            throw new IllegalStateException(jtaLogger.i18NLogger.get_transaction_arjunacore_inactive());
    }
    XAModifier theModifier = null;
    if (params != null) {
        if (params.length >= XAMODIFIER + 1) {
            if (params[XAMODIFIER] instanceof XAModifier) {
                theModifier = (XAModifier) params[XAMODIFIER];
            }
        }
    }
    try {
        /*
			 * For each transaction we maintain a list of resources registered
			 * with it. Each element on this list also contains a list of
			 * threads which have registered this resource, and what their XID
			 * was for that registration.
			 */
        TxInfo info = null;
        try {
            synchronized (this) {
                info = (TxInfo) _resources.get(xaRes);
                if (info == null) {
                    /*
						 * Null info means it's not in the main resources list,
						 * but may be in the duplicates.
						 */
                    info = (TxInfo) _duplicateResources.get(xaRes);
                }
            }
            if (info != null) {
                switch(info.getState()) {
                    case TxInfo.ASSOCIATION_SUSPENDED:
                        {
                            /*
						 * Have seen resource before, so do a resume. The
						 * Resource instance will still be registered with the
						 * transaction though.
						 */
                            int xaStartResume = ((theModifier == null) ? XAResource.TMRESUME : theModifier.xaStartParameters(XAResource.TMRESUME));
                            xaRes.start(info.xid(), xaStartResume);
                            info.setState(TxInfo.ASSOCIATED);
                            synchronized (this) {
                                _suspendCount--;
                            }
                            // already registered resource with this
                            return true;
                        // transaction!
                        }
                    case TxInfo.ASSOCIATED:
                        {
                            return true;
                        }
                    case TxInfo.NOT_ASSOCIATED:
                        {
                            /*
						 * Resource was associated, but was presumably delisted.
						 */
                            int xaStartJoin = ((theModifier == null) ? XAResource.TMJOIN : theModifier.xaStartParameters(XAResource.TMJOIN));
                            xaRes.start(info.xid(), xaStartJoin);
                            info.setState(TxInfo.ASSOCIATED);
                            return true;
                        }
                    default:
                        {
                            throw new IllegalStateException("TransactionImple.enlistResource - " + jtaLogger.i18NLogger.get_transaction_arjunacore_illresstate() + ":" + info.getState());
                        }
                }
            }
        } catch (IllegalStateException ex) {
            // we threw it in the first place
            throw ex;
        } catch (XAException exp) {
            if (info != null)
                info.setState(TxInfo.FAILED);
            jtaLogger.i18NLogger.warn_transaction_arjunacore_enlisterror("TransactionImple.enlistResource", XAHelper.printXAErrorCode(exp));
            return false;
        }
        // if (threadIsActive(xaRes))
        // return true; // this thread has already registered a resource for
        // this db
        /*
			 * We definitely haven't seen this specific resource instance
			 * before, but that doesn't mean that we haven't seen the RM it is
			 * connected to.
			 */
        Xid xid = null;
        TxInfo existingRM = isNewRM(xaRes);
        if (existingRM == null) {
            /*
				 * New RM, so create xid with new branch.
				 */
            boolean branchRequired = true;
            synchronized (this) {
                if (// first ever, so no need for
                _resources.size() == 0) // branch
                {
                    // branchRequired = false;
                    branchRequired = true;
                }
            }
            xid = createXid(branchRequired, theModifier, xaRes);
            boolean associatedWork = false;
            int retry = 20;
            while (!associatedWork) {
                try {
                    if (_xaTransactionTimeoutEnabled) {
                        int timeout = _theTransaction.getTimeout();
                        if (timeout > 0) {
                            try {
                                xaRes.setTransactionTimeout(timeout);
                            } catch (XAException te) {
                                jtaLogger.i18NLogger.warn_transaction_arjunacore_timeouterror("TransactionImple.enlistResource", XAHelper.xidToString(xid), XAHelper.printXAErrorCode(te), te);
                            }
                        }
                    }
                    int xaStartNormal = ((theModifier == null) ? XAResource.TMNOFLAGS : theModifier.xaStartParameters(XAResource.TMNOFLAGS));
                    // Pay attention now, this bit is hairy. We need to add a new AbstractRecord (XAResourceRecord)
                    // to the BasicAction, which will thereafter drive its completion. However, the transaction
                    // core is not directly XA aware, so it's our job to start the XAResource. Problem is, if
                    // adding the record fails, BasicAction will never end the resource via the XAResourceRecord,
                    // so we must do so directly.  start may fail due to dupl xid or other reason, and transactions
                    // may rollback async, for which reasons we can't call add before start.
                    // The xid will change on each pass of the loop, so we need to create a new record on each pass.
                    // The add will fail in the case of multiple last resources being disallowed
                    // see JBTM-362 and JBTM-363
                    AbstractRecord abstractRecord = createRecord(xaRes, params, xid);
                    if (abstractRecord != null) {
                        xaRes.start(xid, xaStartNormal);
                        if (_theTransaction.add(abstractRecord) == AddOutcome.AR_ADDED) {
                            _resources.put(xaRes, new TxInfo(xid));
                            // dive out, no need to set associatedWork = true;
                            return true;
                        } else {
                            // we called start on the resource, but _theTransaction did not accept it.
                            // we therefore have a mess which we must now clean up by ensuring the start is undone:
                            abstractRecord.topLevelAbort();
                        }
                    }
                    // if we get to here, something other than a failure of xaRes.start probably went wrong.
                    // so we don't loop and retry, we just give up.
                    markRollbackOnly();
                    return false;
                } catch (XAException e) {
                    if ((e.errorCode == XAException.XAER_DUPID) || (e.errorCode == XAException.XAER_RMERR)) {
                        if (retry > 0)
                            xid = createXid(true, theModifier, xaRes);
                        retry--;
                    } else {
                        /*
							 * Can't do start, so set transaction to rollback
							 * only.
							 */
                        jtaLogger.i18NLogger.warn_transaction_arjunacore_enliststarterror("TransactionImple.enlistResource", XAHelper.xidToString(xid), XAHelper.printXAErrorCode(e), e);
                        markRollbackOnly();
                        throw e;
                    }
                    if (retry < 0) {
                        jtaLogger.i18NLogger.warn_transaction_arjunacore_enliststarterror("TransactionImple.enlistResource", XAHelper.xidToString(xid), XAHelper.printXAErrorCode(e), e);
                        markRollbackOnly();
                        throw new javax.transaction.SystemException("TransactionImple.enlistResource - XAResource.start " + jtaLogger.i18NLogger.get_transaction_arjunacore_couldnotregister() + ": " + xid);
                    }
                }
            }
        } else {
            /*
				 * Have seen this RM before, so ignore this instance. The first
				 * registered RM instance will be used to drive the transaction
				 * completion. We add it to the duplicateResource list so we can
				 * delist it correctly later though.
				 */
            /*
				 * Re-create xid.
				 */
            xid = existingRM.xid();
            try {
                int xaStartJoin = ((theModifier == null) ? XAResource.TMJOIN : theModifier.xaStartParameters(XAResource.TMJOIN));
                xaRes.start(xid, xaStartJoin);
            } catch (XAException ex) {
                jtaLogger.i18NLogger.warn_transaction_arjunacore_xastart("TransactionImple.enlistResource - xa_start ", XAHelper.xidToString(xid), XAHelper.printXAErrorCode(ex), ex);
                markRollbackOnly();
                throw ex;
            }
            /*
				 * Add to duplicate resources list so we can keep track of it
				 * (particularly if we later have to delist).
				 */
            _duplicateResources.put(xaRes, new TxInfo(xid));
            return true;
        }
        return false;
    } catch (Exception e) {
        e.printStackTrace();
        /*
			 * Some exceptional condition arose and we probably could not enlist
			 * the resouce. So, for safety mark the transaction as rollback
			 * only.
			 */
        markRollbackOnly();
        return false;
    }
}

Example 8 with TxInfo

use of com.arjuna.ats.internal.jta.xa.TxInfo in project narayana by jbosstm.

the class TransactionImple method delistResource.

/*
      * Do we have to unregister resources? Assume not as it would not make much
      * sense otherwise!
      */
public boolean delistResource(XAResource xaRes, int flags) throws IllegalStateException, javax.transaction.SystemException {
    if (jtaLogger.logger.isTraceEnabled()) {
        jtaLogger.logger.trace("TransactionImple.delistResource ( " + xaRes + ", " + flags + " )");
    }
    if (xaRes == null)
        throw new javax.transaction.SystemException("TransactionImple.delistResource - " + jtaLogger.i18NLogger.get_transaction_arjunacore_nullres());
    int status = getStatus();
    switch(status) {
        case javax.transaction.Status.STATUS_ACTIVE:
            break;
        case javax.transaction.Status.STATUS_MARKED_ROLLBACK:
            break;
        default:
            throw new IllegalStateException(jtaLogger.i18NLogger.get_transaction_arjunacore_inactive());
    }
    TxInfo info = null;
    try {
        synchronized (this) {
            info = (TxInfo) _resources.get(xaRes);
            if (info == null)
                info = (TxInfo) _duplicateResources.get(xaRes);
        }
        if (info == null) {
            jtaLogger.i18NLogger.warn_transaction_arjunacore_unknownresource("TransactionImple.delistResource");
            return false;
        } else {
            boolean optimizedRollback = false;
            try {
                if (status == javax.transaction.Status.STATUS_MARKED_ROLLBACK) {
                    if (XAUtils.canOptimizeDelist(xaRes)) {
                        xaRes.end(info.xid(), XAResource.TMFAIL);
                        xaRes.rollback(info.xid());
                        info.setState(TxInfo.OPTIMIZED_ROLLBACK);
                        optimizedRollback = true;
                    }
                }
            } catch (Exception e) {
            // failed, so try again when transaction does rollback
            }
            switch(info.getState()) {
                case TxInfo.ASSOCIATED:
                    {
                        if ((flags & XAResource.TMSUCCESS) != 0) {
                            xaRes.end(info.xid(), XAResource.TMSUCCESS);
                            info.setState(TxInfo.NOT_ASSOCIATED);
                        } else {
                            if ((flags & XAResource.TMSUSPEND) != 0) {
                                xaRes.end(info.xid(), XAResource.TMSUSPEND);
                                info.setState(TxInfo.ASSOCIATION_SUSPENDED);
                                synchronized (this) {
                                    _suspendCount++;
                                }
                            } else {
                                xaRes.end(info.xid(), XAResource.TMFAIL);
                                info.setState(TxInfo.FAILED);
                            }
                        }
                    }
                    break;
                case TxInfo.ASSOCIATION_SUSPENDED:
                    {
                        if ((flags & XAResource.TMSUCCESS) != 0) {
                            if (XAUtils.mustEndSuspendedRMs(xaRes))
                                xaRes.start(info.xid(), XAResource.TMRESUME);
                            xaRes.end(info.xid(), XAResource.TMSUCCESS);
                            info.setState(TxInfo.NOT_ASSOCIATED);
                            synchronized (this) {
                                _suspendCount--;
                            }
                        } else {
                            if ((flags & XAResource.TMSUSPEND) != 0) {
                                throw new IllegalStateException("TransactionImple.delistResource - " + jtaLogger.i18NLogger.get_transaction_arjunacore_ressuspended());
                            } else {
                                xaRes.end(info.xid(), XAResource.TMFAIL);
                                info.setState(TxInfo.FAILED);
                                synchronized (this) {
                                    _suspendCount--;
                                }
                            }
                        }
                    }
                    break;
                default:
                    {
                        if (!optimizedRollback)
                            throw new IllegalStateException("TransactionImple.delistResource - " + jtaLogger.i18NLogger.get_transaction_arjunacore_illresstate() + ":" + info.getState());
                    }
            }
            info = null;
            return true;
        }
    } catch (IllegalStateException ex) {
        throw ex;
    } catch (XAException exp) {
        if (info != null)
            info.setState(TxInfo.FAILED);
        /*
			 * For safety mark the transaction as rollback only.
			 */
        markRollbackOnly();
        jtaLogger.i18NLogger.warn_transaction_arjunacore_delistresource("TransactionImple.delistResource", XAHelper.printXAErrorCode(exp), exp);
        return false;
    } catch (Exception e) {
        jtaLogger.i18NLogger.warn_transaction_arjunacore_delistgeneral("TransactionImple.delistResource", e);
        /*
			 * Some exception occurred and we probably could not delist the
			 * resource. So, for safety mark the transaction as rollback only.
			 */
        markRollbackOnly();
        return false;
    }
}

Example 9 with TxInfo

use of com.arjuna.ats.internal.jta.xa.TxInfo in project narayana by jbosstm.

the class TransactionImple method threadIsActive.

/*
	 * If this thread has already registered a resource for the same db then
	 * don't use this copy. For some databases it would actually be ok for us to
	 * use the resource (at least to do an xa_start equivalent on it), but for
	 * Oracle 8.1.6 it causes their JDBC driver to crash!
	 */
private final boolean threadIsActive(XAResource xaRes) {
    Thread t = Thread.currentThread();
    try {
        Enumeration el = _resources.keys();
        if (el != null) {
            while (el.hasMoreElements()) {
                XAResource x = (XAResource) el.nextElement();
                if (x.isSameRM(xaRes)) {
                    TxInfo info = (TxInfo) _resources.get(x);
                    if (info.thread() == t)
                        return true;
                }
            }
        }
        el = _duplicateResources.keys();
        if (el != null) {
            while (el.hasMoreElements()) {
                XAResource x = (XAResource) el.nextElement();
                if (x.isSameRM(xaRes)) {
                    TxInfo info = (TxInfo) _resources.get(x);
                    if (info.thread() == t)
                        return true;
                }
            }
        }
    } catch (Exception e) {
        jtaLogger.i18NLogger.warn_transaction_arjunacore_threadexception(e);
        throw new com.arjuna.ats.arjuna.exceptions.FatalError(e.toString(), e);
    }
    return false;
}

Example 10 with TxInfo

use of com.arjuna.ats.internal.jta.xa.TxInfo in project narayana by jbosstm.

the class TransactionImple method isNewRM.

/**
 * isNewRM returns an existing TxInfo for the same RM, if present. Null
 * otherwise.
 */
private final TxInfo isNewRM(XAResource xaRes) {
    try {
        synchronized (this) {
            Enumeration el = _resources.keys();
            if (el != null) {
                while (el.hasMoreElements()) {
                    XAResource x = (XAResource) el.nextElement();
                    if (x.isSameRM(xaRes)) {
                        return (TxInfo) _resources.get(x);
                    }
                }
            }
            el = _duplicateResources.keys();
            if (el != null) {
                while (el.hasMoreElements()) {
                    XAResource x = (XAResource) el.nextElement();
                    if (x.isSameRM(xaRes)) {
                        return (TxInfo) _duplicateResources.get(x);
                    }
                }
            }
        }
    } catch (XAException ex) {
        jtaxLogger.i18NLogger.warn_jtax_transaction_jts_xaerror("TransactionImple.isNewRM", XAHelper.printXAErrorCode(ex), ex);
        throw new com.arjuna.ats.arjuna.exceptions.FatalError(ex.toString(), ex);
    } catch (Exception e) {
        jtaxLogger.i18NLogger.warn_jtax_transaction_jts_rmerror(e);
        throw new com.arjuna.ats.arjuna.exceptions.FatalError(e.toString(), e);
    }
    return null;
}