Examples with ScopedConnectionWrapper org.apache.aries.tx.control.jdbc.common.impl.ScopedConnectionWrapper used on opensource projects

Example 1 with ScopedConnectionWrapper

use of org.apache.aries.tx.control.jdbc.common.impl.ScopedConnectionWrapper in project aries by apache.

the class TxContextBindingConnection method getDelegate.

@Override
protected final Connection getDelegate() {
    TransactionContext txContext = txControl.getCurrentContext();
    if (txContext == null) {
        throw new TransactionException("The resource " + provider + " cannot be accessed outside of an active Transaction Context");
    }
    Connection existing = (Connection) txContext.getScopedValue(resourceId);
    if (existing != null) {
        return existing;
    }
    Connection toReturn;
    Connection toClose;
    try {
        if (txContext.getTransactionStatus() == NO_TRANSACTION) {
            toClose = provider.getConnection();
            toReturn = new ScopedConnectionWrapper(toClose);
        } else if (txContext.supportsLocal()) {
            toClose = provider.getConnection();
            toReturn = new TxConnectionWrapper(toClose);
            txContext.registerLocalResource(getLocalResource(toClose));
        } else {
            throw new TransactionException("There is a transaction active, but it does not support local participants");
        }
    } catch (Exception sqle) {
        throw new TransactionException("There was a problem getting hold of a database connection", sqle);
    }
    txContext.postCompletion(x -> {
        try {
            toClose.close();
        } catch (SQLException sqle) {
        }
    });
    txContext.putScopedValue(resourceId, toReturn);
    return toReturn;
}

Example 2 with ScopedConnectionWrapper

use of org.apache.aries.tx.control.jdbc.common.impl.ScopedConnectionWrapper in project aries by apache.

the class XAEnabledTxContextBindingConnection method getDelegate.

@Override
protected final Connection getDelegate() {
    TransactionContext txContext = txControl.getCurrentContext();
    if (txContext == null) {
        throw new TransactionException("The resource " + provider + " cannot be accessed outside of an active Transaction Context");
    }
    Connection existing = (Connection) txContext.getScopedValue(resourceId);
    if (existing != null) {
        return existing;
    }
    Connection toReturn;
    Connection toClose;
    try {
        if (txContext.getTransactionStatus() == NO_TRANSACTION) {
            toClose = provider.getConnection();
            toReturn = new ScopedConnectionWrapper(toClose);
        } else if (txContext.supportsXA() && xaEnabled) {
            toClose = provider.getConnection();
            toReturn = new TxConnectionWrapper(toClose);
            txContext.registerXAResource(getXAResource(toClose), recoveryIdentifier);
        } else if (txContext.supportsLocal() && localEnabled) {
            toClose = provider.getConnection();
            toReturn = new TxConnectionWrapper(toClose);
            txContext.registerLocalResource(getLocalResource(toClose));
        } else {
            throw new TransactionException("There is a transaction active, but it does not support local participants");
        }
    } catch (Exception sqle) {
        throw new TransactionException("There was a problem getting hold of a database connection", sqle);
    }
    txContext.postCompletion(x -> {
        try {
            toClose.close();
        } catch (SQLException sqle) {
        }
    });
    txContext.putScopedValue(resourceId, toReturn);
    return toReturn;
}