/*
*
* The DbUnit Database Testing Framework
* Copyright (C)2002-2004, DbUnit.org
*
* This library is free software; you can redistribute it and/or
* modify it under the terms of the GNU Lesser General Public
* License as published by the Free Software Foundation; either
* version 2.1 of the License, or (at your option) any later version.
*
* This library is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public
* License along with this library; if not, write to the Free Software
* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
*
*/
package org.dbunit.dataset;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.dbunit.database.AmbiguousTableNameException;
import org.dbunit.dataset.filter.ITableFilter;
import org.dbunit.dataset.filter.SequenceTableFilter;
/**
* Decorates a dataset and exposes only some tables from it. Can be used with
* different filtering strategies.
*
* @see ITableFilter
* @see SequenceTableFilter
* @see org.dbunit.dataset.filter.DefaultTableFilter
*
* @author Manuel Laflamme
* @version $Revision: 1049 $
* @since Feb 22, 2002
*/
public class FilteredDataSet extends AbstractDataSet
{
/**
* Logger for this class
*/
private static final Logger logger = LoggerFactory.getLogger(FilteredDataSet.class);
private final IDataSet _dataSet;
private final ITableFilter _filter;
/**
* Creates a FilteredDataSet that decorates the specified dataset and
* exposes only the specified tables using {@link SequenceTableFilter} as
* filtering strategy.
* @throws AmbiguousTableNameException If the given tableNames array contains ambiguous names
*/
public FilteredDataSet(String[] tableNames, IDataSet dataSet)
throws AmbiguousTableNameException
{
_filter = new SequenceTableFilter(tableNames, dataSet.isCaseSensitiveTableNames());
_dataSet = dataSet;
}
/**
* Creates a FilteredDataSet that decorates the specified dataset and
* exposes only the tables allowed by the specified filter.
*
* @param dataSet the filtered dataset
* @param filter the filtering strategy
*/
public FilteredDataSet(ITableFilter filter, IDataSet dataSet)
{
_dataSet = dataSet;
_filter = filter;
}
////////////////////////////////////////////////////////////////////////////
// AbstractDataSet class
protected ITableIterator createIterator(boolean reversed)
throws DataSetException
{
if(logger.isDebugEnabled())
logger.debug("createIterator(reversed={}) - start", String.valueOf(reversed));
return _filter.iterator(_dataSet, reversed);
}
////////////////////////////////////////////////////////////////////////////
// IDataSet interface
public String[] getTableNames() throws DataSetException
{
return _filter.getTableNames(_dataSet);
}
public ITableMetaData getTableMetaData(String tableName)
throws DataSetException
{
if (!_filter.accept(tableName))
{
throw new NoSuchTableException(tableName);
}
return _dataSet.getTableMetaData(tableName);
}
public ITable getTable(String tableName) throws DataSetException
{
logger.debug("getTable(tableName={}) - start", tableName);
if (!_filter.accept(tableName))
{
throw new NoSuchTableException(tableName);
}
return _dataSet.getTable(tableName);
}
}