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;
022
023import javax.sql.XAConnection;
024import javax.sql.XADataSource;
025
026import net.sf.hajdbc.sql.ProxyFactoryFactory;
027import net.sf.hajdbc.sql.pool.AbstractPooledConnectionInvocationHandler;
028import net.sf.hajdbc.util.reflect.Methods;
029
030/**
031 * @author Paul Ferraro
032 */
033public class XAConnectionInvocationHandler extends AbstractPooledConnectionInvocationHandler<XADataSource, XADataSourceDatabase, XAConnection, XAConnectionProxyFactory>
034{
035        private static final Method getXAResource = Methods.getMethod(XAConnection.class, "getXAResource");
036        
037        public XAConnectionInvocationHandler(XAConnectionProxyFactory proxyFactory)
038        {
039                super(XAConnection.class, proxyFactory);
040        }
041
042        @Override
043        protected ProxyFactoryFactory<XADataSource, XADataSourceDatabase, XAConnection, SQLException, ?, ? extends Exception> getProxyFactoryFactory(XAConnection object, Method method, Object... parameters) throws SQLException
044        {
045                if (method.equals(getXAResource))
046                {
047                        return new XAResourceProxyFactoryFactory();
048                }
049                
050                return super.getProxyFactoryFactory(object, method, parameters);
051        }
052}