001/* 002 * HA-JDBC: High-Availability JDBC 003 * Copyright (C) 2012 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.pool; 019 020import java.lang.reflect.Method; 021import java.sql.SQLException; 022import java.util.Set; 023 024import javax.sql.ConnectionPoolDataSource; 025 026import net.sf.hajdbc.invocation.InvocationStrategies; 027import net.sf.hajdbc.invocation.InvocationStrategy; 028import net.sf.hajdbc.sql.CommonDataSourceInvocationHandler; 029import net.sf.hajdbc.sql.ProxyFactoryFactory; 030import net.sf.hajdbc.util.reflect.Methods; 031 032/** 033 * @author Paul Ferraro 034 * 035 */ 036@SuppressWarnings("nls") 037public class ConnectionPoolDataSourceInvocationHandler extends CommonDataSourceInvocationHandler<ConnectionPoolDataSource, ConnectionPoolDataSourceDatabase, ConnectionPoolDataSourceProxyFactory> 038{ 039 private static final Set<Method> getPooledConnectionMethodSet = Methods.findMethods(ConnectionPoolDataSource.class, "getPooledConnection"); 040 041 public ConnectionPoolDataSourceInvocationHandler(ConnectionPoolDataSourceProxyFactory factory) 042 { 043 super(ConnectionPoolDataSource.class, factory); 044 } 045 046 /** 047 * {@inheritDoc} 048 * @see net.sf.hajdbc.sql.CommonDataSourceInvocationHandler#getInvocationStrategy(javax.sql.CommonDataSource, java.lang.reflect.Method, java.lang.Object[]) 049 */ 050 @Override 051 protected InvocationStrategy getInvocationStrategy(ConnectionPoolDataSource dataSource, Method method, Object... parameters) throws SQLException 052 { 053 if (getPooledConnectionMethodSet.contains(method)) 054 { 055 return InvocationStrategies.TRANSACTION_INVOKE_ON_ALL; 056 } 057 return super.getInvocationStrategy(dataSource, method, parameters); 058 } 059 060 @Override 061 protected ProxyFactoryFactory<ConnectionPoolDataSource, ConnectionPoolDataSourceDatabase, ConnectionPoolDataSource, SQLException, ?, ? extends Exception> getProxyFactoryFactory(ConnectionPoolDataSource object, Method method, Object... parameters) throws SQLException 062 { 063 if (getPooledConnectionMethodSet.contains(method)) 064 { 065 return new PooledConnectionProxyFactoryFactory(); 066 } 067 068 return super.getProxyFactoryFactory(object, method, parameters); 069 } 070}