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}