001/*
002 * HA-JDBC: High-Availability JDBC
003 * Copyright (c) 2004-2007 Paul Ferraro
004 * 
005 * This library is free software; you can redistribute it and/or modify it 
006 * under the terms of the GNU Lesser General Public License as published by the 
007 * Free Software Foundation; either version 2.1 of the License, or (at your 
008 * option) any later version.
009 * 
010 * This library is distributed in the hope that it will be useful, but WITHOUT
011 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or 
012 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License 
013 * for more details.
014 * 
015 * You should have received a copy of the GNU Lesser General Public License
016 * along with this library; if not, write to the Free Software Foundation, 
017 * Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
018 * 
019 * Contact: ferraro@users.sourceforge.net
020 */
021package net.sf.hajdbc.sql;
022
023import java.lang.reflect.Method;
024import java.util.Set;
025
026import javax.sql.DataSource;
027
028import net.sf.hajdbc.DatabaseCluster;
029import net.sf.hajdbc.util.reflect.Methods;
030
031/**
032 * @author Paul Ferraro
033 *
034 */
035@SuppressWarnings("nls")
036public class DataSourceInvocationHandler extends CommonDataSourceInvocationHandler<DataSource>
037{
038        private static final Set<Method> getConnectionMethods = Methods.findMethods(DataSource.class, "getConnection");
039        
040        /**
041         * @param databaseCluster
042         */
043        public DataSourceInvocationHandler(DatabaseCluster<DataSource> databaseCluster)
044        {
045                super(databaseCluster, DataSource.class);
046        }
047
048        /**
049         * @see net.sf.hajdbc.sql.AbstractChildInvocationHandler#getInvocationStrategy(java.lang.Object, java.lang.reflect.Method, java.lang.Object[])
050         */
051        @Override
052        protected InvocationStrategy<DataSource, DataSource, ?> getInvocationStrategy(DataSource dataSource, Method method, Object[] parameters) throws Exception
053        {
054                if (getConnectionMethods.contains(method))
055                {
056                        TransactionContext<DataSource> context = new LocalTransactionContext<DataSource>(this.cluster);
057                        
058                        return new ConnectionInvocationStrategy<DataSource, DataSource>(this.cluster, dataSource, context);
059                }
060                
061                return super.getInvocationStrategy(dataSource, method, parameters);
062        }
063}