Example usage for org.springframework.jdbc.datasource ConnectionHolder ConnectionHolder

List of usage examples for org.springframework.jdbc.datasource ConnectionHolder ConnectionHolder

Introduction

In this page you can find the example usage for org.springframework.jdbc.datasource ConnectionHolder ConnectionHolder.

Prototype

public ConnectionHolder(Connection connection) 

Source Link

Document

Create a new ConnectionHolder for the given JDBC Connection, wrapping it with a SimpleConnectionHandle , assuming that there is no ongoing transaction.

Usage

From source file:org.springframework.jdbc.datasource.DataSourceUtils.java

/**
 * Actually obtain a JDBC Connection from the given DataSource.
 * Same as {@link #getConnection}, but throwing the original SQLException.
 * <p>Is aware of a corresponding Connection bound to the current thread, for example
 * when using {@link DataSourceTransactionManager}. Will bind a Connection to the thread
 * if transaction synchronization is active (e.g. if in a JTA transaction).
 * <p>Directly accessed by {@link TransactionAwareDataSourceProxy}.
 * @param dataSource the DataSource to obtain Connections from
 * @return a JDBC Connection from the given DataSource
 * @throws SQLException if thrown by JDBC methods
 * @see #doReleaseConnection/*  w w w .  j  a va  2s. c  om*/
 */
public static Connection doGetConnection(DataSource dataSource) throws SQLException {
    Assert.notNull(dataSource, "No DataSource specified");

    ConnectionHolder conHolder = (ConnectionHolder) TransactionSynchronizationManager.getResource(dataSource);
    if (conHolder != null && (conHolder.hasConnection() || conHolder.isSynchronizedWithTransaction())) {
        conHolder.requested();
        if (!conHolder.hasConnection()) {
            logger.debug("Fetching resumed JDBC Connection from DataSource");
            conHolder.setConnection(fetchConnection(dataSource));
        }
        return conHolder.getConnection();
    }
    // Else we either got no holder or an empty thread-bound holder here.

    logger.debug("Fetching JDBC Connection from DataSource");
    Connection con = fetchConnection(dataSource);

    if (TransactionSynchronizationManager.isSynchronizationActive()) {
        logger.debug("Registering transaction synchronization for JDBC Connection");
        // Use same Connection for further JDBC actions within the transaction.
        // Thread-bound object will get removed by synchronization at transaction completion.
        ConnectionHolder holderToUse = conHolder;
        if (holderToUse == null) {
            holderToUse = new ConnectionHolder(con);
        } else {
            holderToUse.setConnection(con);
        }
        holderToUse.requested();
        TransactionSynchronizationManager
                .registerSynchronization(new ConnectionSynchronization(holderToUse, dataSource));
        holderToUse.setSynchronizedWithTransaction(true);
        if (holderToUse != conHolder) {
            TransactionSynchronizationManager.bindResource(dataSource, holderToUse);
        }
    }

    return con;
}

From source file:org.springframework.jdbc.object.StoredProcedureTests.java

public void testAddInvoicesWithinTransaction() throws Exception {
    mockCallable.setObject(1, new Integer(1106), Types.INTEGER);
    ctrlCallable.setVoidCallable();//from ww w.j  a v  a2s. c  om
    mockCallable.setObject(2, new Integer(3), Types.INTEGER);
    ctrlCallable.setVoidCallable();
    mockCallable.registerOutParameter(3, Types.INTEGER);
    ctrlCallable.setVoidCallable();
    mockCallable.execute();
    ctrlCallable.setReturnValue(false);
    mockCallable.getUpdateCount();
    ctrlCallable.setReturnValue(-1);
    mockCallable.getObject(3);
    ctrlCallable.setReturnValue(new Integer(4));
    if (debugEnabled) {
        mockCallable.getWarnings();
        ctrlCallable.setReturnValue(null);
    }
    mockCallable.close();
    ctrlCallable.setVoidCallable();

    mockConnection.prepareCall("{call " + AddInvoice.SQL + "(?, ?, ?)}");
    ctrlConnection.setReturnValue(mockCallable);

    replay();

    TransactionSynchronizationManager.bindResource(mockDataSource, new ConnectionHolder(mockConnection));

    try {
        testAddInvoice(1106, 3);
    } finally {
        TransactionSynchronizationManager.unbindResource(mockDataSource);
    }
}