001/*
002 * HA-JDBC: High-Availability JDBC
003 * Copyright (C) 2013  Paul Ferraro
004 *
005 * This program is free software: you can redistribute it and/or modify
006 * it under the terms of the GNU Lesser General Public License as published by
007 * the Free Software Foundation, either version 3 of the License, or
008 * (at your option) any later version.
009 *
010 * This program is distributed in the hope that it will be useful,
011 * but WITHOUT ANY WARRANTY; without even the implied warranty of
012 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
013 * GNU Lesser General Public License for more details.
014 *
015 * You should have received a copy of the GNU Lesser General Public License
016 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
017 */
018package net.sf.hajdbc.sql;
019
020import java.sql.Connection;
021import java.sql.SQLException;
022import java.util.Map;
023
024import net.sf.hajdbc.Database;
025import net.sf.hajdbc.invocation.Invoker;
026import net.sf.hajdbc.logging.Level;
027import net.sf.hajdbc.util.Resources;
028import net.sf.hajdbc.util.reflect.Proxies;
029
030/**
031 * 
032 * @author Paul Ferraro
033 */
034public class ConnectionProxyFactory<Z, D extends Database<Z>, P> extends AbstractSQLProxyFactory<Z, D, P, Connection>
035{
036        public ConnectionProxyFactory(P parentProxy, ProxyFactory<Z, D, P, SQLException> parent, Invoker<Z, D, P, Connection, SQLException> invoker, Map<D, Connection> map, TransactionContext<Z, D> context)
037        {
038                super(parentProxy, parent, invoker, map, context);
039        }
040
041        @Override
042        public Connection getConnection(D database)
043        {
044                return this.get(database);
045        }
046
047        @Override
048        public Connection createProxy()
049        {
050                return Proxies.createProxy(Connection.class, new ConnectionInvocationHandler<Z, D, P>(this));
051        }
052
053        @Override
054        public void close(D database, Connection connection)
055        {
056                try
057                {
058                        if (!connection.isClosed())
059                        {
060                                // Ensure that we rollback any current transaction to release any locks before closing
061                                // This is necessary if the connection is pooled
062                                if (!connection.getAutoCommit())
063                                {
064                                        if (!connection.isReadOnly() || connection.getTransactionIsolation() >= Connection.TRANSACTION_REPEATABLE_READ)
065                                        {
066                                                connection.rollback();
067                                        }
068                                }
069                        }
070                }
071                catch (SQLException e)
072                {
073                        this.logger.log(Level.WARN, e);
074                }
075
076                Resources.close(connection);
077        }
078}