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.invocation;
019
020import java.util.Map;
021import java.util.Map.Entry;
022import java.util.SortedMap;
023import java.util.TreeMap;
024import java.util.AbstractMap;
025
026import net.sf.hajdbc.Database;
027import net.sf.hajdbc.sql.ProxyFactory;
028
029/**
030 * @author Paul Ferraro
031 *
032 */
033public class ExistingResultsCollector implements InvokeOnManyInvocationStrategy.ResultsCollector
034{
035        @Override
036        public <Z, D extends Database<Z>, T, R, E extends Exception> Entry<SortedMap<D, R>, SortedMap<D, E>> collectResults(ProxyFactory<Z, D, T, E> factory, Invoker<Z, D, T, R, E> invoker)
037        {
038                SortedMap<D, R> resultMap = new TreeMap<D, R>();
039                SortedMap<D, E> exceptionMap = new TreeMap<D, E>();
040
041                for (Map.Entry<D, T> entry: factory.entries())
042                {
043                        D database = entry.getKey();
044                        
045                        try
046                        {
047                                resultMap.put(database, invoker.invoke(database, entry.getValue()));
048                        }
049                        catch (Exception e)
050                        {
051                                // If this database was concurrently deactivated, just ignore the failure
052                                if (factory.getDatabaseCluster().getBalancer().contains(database))
053                                {
054                                        exceptionMap.put(database, factory.getExceptionFactory().createException(e));
055                                }
056                        }
057                }
058                
059                return new AbstractMap.SimpleImmutableEntry<SortedMap<D, R>, SortedMap<D, E>>(resultMap, exceptionMap);
060        }
061}