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.xa; 019 020import java.lang.reflect.Method; 021import java.sql.SQLException; 022import java.util.Set; 023 024import javax.sql.XADataSource; 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 */ 036public class XADataSourceInvocationHandler extends CommonDataSourceInvocationHandler<XADataSource, XADataSourceDatabase, XADataSourceProxyFactory> 037{ 038 private static final Set<Method> getXAConnectionMethodSet = Methods.findMethods(XADataSource.class, "getXAConnection"); 039 040 public XADataSourceInvocationHandler(XADataSourceProxyFactory factory) 041 { 042 super(XADataSource.class, factory); 043 } 044 045 /** 046 * {@inheritDoc} 047 * @see net.sf.hajdbc.sql.CommonDataSourceInvocationHandler#getInvocationStrategy(javax.sql.CommonDataSource, java.lang.reflect.Method, java.lang.Object[]) 048 */ 049 @Override 050 protected InvocationStrategy getInvocationStrategy(XADataSource dataSource, Method method, Object... parameters) throws SQLException 051 { 052 if (getXAConnectionMethodSet.contains(method)) 053 { 054 return InvocationStrategies.TRANSACTION_INVOKE_ON_ALL; 055 } 056 return super.getInvocationStrategy(dataSource, method, parameters); 057 } 058 059 @Override 060 protected ProxyFactoryFactory<XADataSource, XADataSourceDatabase, XADataSource, SQLException, ?, ? extends Exception> getProxyFactoryFactory(XADataSource object, Method method, Object... parameters) throws SQLException 061 { 062 if (getXAConnectionMethodSet.contains(method)) 063 { 064 return new XAConnectionProxyFactoryFactory(); 065 } 066 067 return super.getProxyFactoryFactory(object, method, parameters); 068 } 069}