All Downloads are FREE. Search and download functionalities are using the official Maven repository.

org.hibernate.engine.jdbc.internal.LogicalConnectionImpl Maven / Gradle / Ivy

There is a newer version: 7.0.0.Alpha1
Show newest version
/*
 * Hibernate, Relational Persistence for Idiomatic Java
 *
 * Copyright (c) 2010, Red Hat Inc. or third-party contributors as
 * indicated by the @author tags or express copyright attribution
 * statements applied by the authors.  All third-party contributions are
 * distributed under license by Red Hat Inc.
 *
 * 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, as published by the Free Software Foundation.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY 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
 * along with this distribution; if not, write to:
 * Free Software Foundation, Inc.
 * 51 Franklin Street, Fifth Floor
 * Boston, MA  02110-1301  USA
 */
package org.hibernate.engine.jdbc.internal;

import java.io.IOException;
import java.io.ObjectInputStream;
import java.io.ObjectOutputStream;
import java.sql.Connection;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.Iterator;
import java.util.List;

import org.hibernate.ConnectionReleaseMode;
import org.hibernate.HibernateException;
import org.hibernate.JDBCException;
import org.hibernate.engine.jdbc.spi.ConnectionObserver;
import org.hibernate.engine.jdbc.spi.JdbcConnectionAccess;
import org.hibernate.engine.jdbc.spi.JdbcServices;
import org.hibernate.engine.jdbc.spi.LogicalConnectionImplementor;
import org.hibernate.engine.jdbc.spi.NonDurableConnectionObserver;
import org.hibernate.engine.transaction.spi.TransactionContext;
import org.hibernate.internal.CoreMessageLogger;
import org.hibernate.internal.util.collections.CollectionHelper;

import org.jboss.logging.Logger;

/**
 * Standard Hibernate {@link org.hibernate.engine.jdbc.spi.LogicalConnection} implementation
 * 

* IMPL NOTE : Custom serialization handling! * * @author Steve Ebersole * @author Brett Meyer */ public class LogicalConnectionImpl implements LogicalConnectionImplementor { private static final CoreMessageLogger LOG = Logger.getMessageLogger( CoreMessageLogger.class, LogicalConnectionImpl.class.getName() ); private transient Connection physicalConnection; private final transient ConnectionReleaseMode connectionReleaseMode; private final transient JdbcServices jdbcServices; private final transient JdbcConnectionAccess jdbcConnectionAccess; private final transient List observers; private final boolean isUserSuppliedConnection; private boolean isClosed; /** * Constructs a LogicalConnectionImpl * * @param userSuppliedConnection The user-supplied connection * @param connectionReleaseMode The connection release mode to use * @param jdbcServices JdbcServices * @param jdbcConnectionAccess JDBC Connection access */ public LogicalConnectionImpl( Connection userSuppliedConnection, ConnectionReleaseMode connectionReleaseMode, JdbcServices jdbcServices, JdbcConnectionAccess jdbcConnectionAccess) { this( connectionReleaseMode, jdbcServices, jdbcConnectionAccess, (userSuppliedConnection != null), false, new ArrayList() ); this.physicalConnection = userSuppliedConnection; } /** * Constructs a LogicalConnectionImpl. This for used from deserialization */ private LogicalConnectionImpl( ConnectionReleaseMode connectionReleaseMode, JdbcServices jdbcServices, JdbcConnectionAccess jdbcConnectionAccess, boolean isUserSuppliedConnection, boolean isClosed, List observers) { this.connectionReleaseMode = determineConnectionReleaseMode( jdbcConnectionAccess, isUserSuppliedConnection, connectionReleaseMode ); this.jdbcServices = jdbcServices; this.jdbcConnectionAccess = jdbcConnectionAccess; this.observers = observers; this.isUserSuppliedConnection = isUserSuppliedConnection; this.isClosed = isClosed; } private static ConnectionReleaseMode determineConnectionReleaseMode( JdbcConnectionAccess jdbcConnectionAccess, boolean isUserSuppliedConnection, ConnectionReleaseMode connectionReleaseMode) { if ( isUserSuppliedConnection ) { return ConnectionReleaseMode.ON_CLOSE; } else if ( connectionReleaseMode == ConnectionReleaseMode.AFTER_STATEMENT && ! jdbcConnectionAccess.supportsAggressiveRelease() ) { LOG.debug( "Connection provider reports to not support aggressive release; overriding" ); return ConnectionReleaseMode.AFTER_TRANSACTION; } else { return connectionReleaseMode; } } @Override public JdbcServices getJdbcServices() { return jdbcServices; } @Override public void addObserver(ConnectionObserver observer) { observers.add( observer ); } @Override public void removeObserver(ConnectionObserver connectionObserver) { observers.remove( connectionObserver ); } @Override public boolean isOpen() { return !isClosed; } @Override public boolean isPhysicallyConnected() { return physicalConnection != null; } @Override public Connection getConnection() throws HibernateException { if ( isClosed ) { throw new HibernateException( "Logical connection is closed" ); } if ( physicalConnection == null ) { if ( isUserSuppliedConnection ) { // should never happen throw new HibernateException( "User-supplied connection was null" ); } obtainConnection(); } return physicalConnection; } @Override public Connection close() { LOG.trace( "Closing logical connection" ); final Connection c = isUserSuppliedConnection ? physicalConnection : null; try { if ( !isUserSuppliedConnection && physicalConnection != null ) { releaseConnection(); } return c; } finally { // no matter what physicalConnection = null; isClosed = true; LOG.trace( "Logical connection closed" ); for ( ConnectionObserver observer : observers ) { observer.logicalConnectionClosed(); } observers.clear(); } } @Override public ConnectionReleaseMode getConnectionReleaseMode() { return connectionReleaseMode; } /** * Force aggressive release of the underlying connection. */ @Override public void aggressiveRelease() { if ( isUserSuppliedConnection ) { LOG.debug( "Cannot aggressively release user-supplied connection; skipping" ); } else { LOG.debug( "Aggressively releasing JDBC connection" ); if ( physicalConnection != null ) { releaseConnection(); } } } /** * Physically opens a JDBC Connection. * * @throws org.hibernate.JDBCException Indicates problem opening a connection */ private void obtainConnection() throws JDBCException { LOG.debug( "Obtaining JDBC connection" ); try { physicalConnection = jdbcConnectionAccess.obtainConnection(); for ( ConnectionObserver observer : observers ) { observer.physicalConnectionObtained( physicalConnection ); } LOG.debug( "Obtained JDBC connection" ); } catch ( SQLException sqle) { throw getJdbcServices().getSqlExceptionHelper().convert( sqle, "Could not open connection" ); } } /** * Physically closes the JDBC Connection. * * @throws JDBCException Indicates problem closing a connection */ @Override public void releaseConnection() throws JDBCException { LOG.debug( "Releasing JDBC connection" ); if ( physicalConnection == null ) { return; } try { if ( !physicalConnection.isClosed() ) { getJdbcServices().getSqlExceptionHelper().logAndClearWarnings( physicalConnection ); } if ( !isUserSuppliedConnection ) { jdbcConnectionAccess.releaseConnection( physicalConnection ); } } catch (SQLException e) { throw getJdbcServices().getSqlExceptionHelper().convert( e, "Could not close connection" ); } finally { physicalConnection = null; } LOG.debug( "Released JDBC connection" ); for ( ConnectionObserver observer : observers ) { observer.physicalConnectionReleased(); } releaseNonDurableObservers(); } private void releaseNonDurableObservers() { final Iterator observers = this.observers.iterator(); while ( observers.hasNext() ) { if ( NonDurableConnectionObserver.class.isInstance( observers.next() ) ) { observers.remove(); } } } @Override public Connection manualDisconnect() { if ( isClosed ) { throw new IllegalStateException( "cannot manually disconnect because logical connection is already closed" ); } final Connection c = physicalConnection; releaseConnection(); return c; } @Override public void manualReconnect(Connection suppliedConnection) { if ( isClosed ) { throw new IllegalStateException( "cannot manually reconnect because logical connection is already closed" ); } if ( !isUserSuppliedConnection ) { throw new IllegalStateException( "cannot manually reconnect unless Connection was originally supplied" ); } else { if ( suppliedConnection == null ) { throw new IllegalArgumentException( "cannot reconnect a null user-supplied connection" ); } else if ( suppliedConnection == physicalConnection ) { LOG.debug( "reconnecting the same connection that is already connected; should this connection have been disconnected?" ); } else if ( physicalConnection != null ) { throw new IllegalArgumentException( "cannot reconnect to a new user-supplied connection because currently connected; must disconnect before reconnecting." ); } physicalConnection = suppliedConnection; LOG.debug( "Reconnected JDBC connection" ); } } @Override public boolean isAutoCommit() { if ( !isOpen() || ! isPhysicallyConnected() ) { return true; } try { return getConnection().getAutoCommit(); } catch (SQLException e) { throw jdbcServices.getSqlExceptionHelper().convert( e, "could not inspect JDBC autocommit mode" ); } } @Override public boolean isUserSuppliedConnection() { return isUserSuppliedConnection; } @Override public void notifyObserversStatementPrepared() { for ( ConnectionObserver observer : observers ) { observer.statementPrepared(); } } /** * Serialization hook * * @param oos The stream to write out state to * * @throws IOException Problem accessing stream */ public void serialize(ObjectOutputStream oos) throws IOException { oos.writeBoolean( isUserSuppliedConnection ); oos.writeBoolean( isClosed ); final List durableConnectionObservers = new ArrayList(); for ( ConnectionObserver observer : observers ) { if ( ! NonDurableConnectionObserver.class.isInstance( observer ) ) { durableConnectionObservers.add( observer ); } } oos.writeInt( durableConnectionObservers.size() ); for ( ConnectionObserver observer : durableConnectionObservers ) { oos.writeObject( observer ); } } /** * Deserialization hook * * @param ois The stream to read our state from * @param transactionContext The transactionContext which owns this logical connection * * @return The deserialized LogicalConnectionImpl * * @throws IOException Trouble accessing the stream * @throws ClassNotFoundException Trouble reading the stream */ public static LogicalConnectionImpl deserialize( ObjectInputStream ois, TransactionContext transactionContext) throws IOException, ClassNotFoundException { final boolean isUserSuppliedConnection = ois.readBoolean(); final boolean isClosed = ois.readBoolean(); final int observerCount = ois.readInt(); final List observers = CollectionHelper.arrayList( observerCount ); for ( int i = 0; i < observerCount; i++ ) { observers.add( (ConnectionObserver) ois.readObject() ); } return new LogicalConnectionImpl( transactionContext.getConnectionReleaseMode(), transactionContext.getTransactionEnvironment().getJdbcServices(), transactionContext.getJdbcConnectionAccess(), isUserSuppliedConnection, isClosed, observers ); } }





© 2015 - 2024 Weber Informatics LLC | Privacy Policy