001/*
002 * HA-JDBC: High-Availability JDBC
003 * Copyright (c) 2004-2008 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.pool;
022
023import javax.sql.ConnectionPoolDataSource;
024import javax.sql.PooledConnection;
025
026import net.sf.hajdbc.DatabaseCluster;
027import net.sf.hajdbc.sql.DatabaseWriteInvocationStrategy;
028import net.sf.hajdbc.sql.Invoker;
029import net.sf.hajdbc.sql.SQLProxy;
030import net.sf.hajdbc.util.reflect.ProxyFactory;
031
032/**
033 * @author Paul Ferraro
034 */
035public class PooledConnectionInvocationStrategy extends DatabaseWriteInvocationStrategy<ConnectionPoolDataSource, ConnectionPoolDataSource, PooledConnection>
036{
037        private ConnectionPoolDataSource dataSource;
038        
039        /**
040         * @param cluster 
041         * @param dataSource
042         */
043        public PooledConnectionInvocationStrategy(DatabaseCluster<ConnectionPoolDataSource> cluster, ConnectionPoolDataSource dataSource)
044        {
045                super(cluster.getNonTransactionalExecutor());
046                
047                this.dataSource = dataSource;
048        }
049        
050        @Override
051        public PooledConnection invoke(SQLProxy<ConnectionPoolDataSource, ConnectionPoolDataSource> proxy, Invoker<ConnectionPoolDataSource, ConnectionPoolDataSource, PooledConnection> invoker) throws Exception
052        {
053                return ProxyFactory.createProxy(PooledConnection.class, new PooledConnectionInvocationHandler(this.dataSource, proxy, invoker, this.invokeAll(proxy, invoker)));
054        }
055}