public XATransaction(TransactionStore transactionStore,
    XATransactionId xid,
    TransactionBroker broker) 
{
        this.transactionStore = transactionStore;
        this.xid = xid;
        this.broker = broker;
        if (LOG.isDebugEnabled()) {
            LOG.debug("XA Transaction new/begin : " + xid);
        }
}

 public  void commit(boolean onePhase) throws XAException, IOException 
{
        if (LOG.isDebugEnabled()) {
            LOG.debug("XA Transaction commit: " + xid);
        }
        switch (getState()) {
        case START_STATE:
            // 1 phase commit, no work done.
            checkForPreparedState(onePhase);
            setStateFinished();
            break;
        case IN_USE_STATE:
            // 1 phase commit, work done.
            checkForPreparedState(onePhase);
            doPrePrepare();
            setStateFinished();
            transactionStore.commit(getTransactionId(), false);
            doPostCommit();
            break;
        case PREPARED_STATE:
            // 2 phase commit, work done.
            // We would record commit here.
            setStateFinished();
            transactionStore.commit(getTransactionId(), true);
            doPostCommit();
            break;
        default:
            illegalStateTransition("commit");
        }
}

 public TransactionId getTransactionId() 
{
        return xid;
}

 public int prepare() throws XAException, IOException 
{
        if (LOG.isDebugEnabled()) {
            LOG.debug("XA Transaction prepare: " + xid);
        }
        switch (getState()) {
        case START_STATE:
            // No work done.. no commit/rollback needed.
            setStateFinished();
            return XAResource.XA_RDONLY;
        case IN_USE_STATE:
            // We would record prepare here.
            doPrePrepare();
            setState(Transaction.PREPARED_STATE);
            transactionStore.prepare(getTransactionId());
            return XAResource.XA_OK;
        default:
            illegalStateTransition("prepare");
            return XAResource.XA_RDONLY;
        }
}

 public  void rollback() throws XAException, IOException 
{
        if (LOG.isDebugEnabled()) {
            LOG.debug("XA Transaction rollback: " + xid);
        }
        switch (getState()) {
        case START_STATE:
            // 1 phase rollback no work done.
            setStateFinished();
            break;
        case IN_USE_STATE:
            // 1 phase rollback work done.
            setStateFinished();
            transactionStore.rollback(getTransactionId());
            doPostRollback();
            break;
        case PREPARED_STATE:
            // 2 phase rollback work done.
            setStateFinished();
            transactionStore.rollback(getTransactionId());
            doPostRollback();
            break;
        default:
            throw new XAException("Invalid state");
        }
}