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}