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}