TableDecoratorDataSet.java
- /*
- *
- * The DbUnit Database Testing Framework
- * Copyright (C)2002-2019, 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;
- /**
- * Decorates a dataset to allow decorating the returned {@link ITable}s (and
- * {@link ITableMetaData}s, by extension). Intended to be used for things like
- * filtering out columns. Do not use this to filter out entire tables; use
- * {@link FilteredDataSet} for that.
- *
- * @see ColumnFilterTable
- *
- * @author rcd (rcd AT users.sourceforge.net)
- * @since 3.0.0
- */
- public class TableDecoratorDataSet extends AbstractDataSet
- {
- private final IDataSet _dataSet;
- private final TableDecoratorFunction _decoratorFunction;
- public TableDecoratorDataSet(final IDataSet dataSet,
- final TableDecoratorFunction decoratorFunction)
- {
- _dataSet = dataSet;
- _decoratorFunction = decoratorFunction;
- }
- @Override
- protected ITableIterator createIterator(final boolean reversed)
- throws DataSetException
- {
- return new FilterIterator(
- reversed ? _dataSet.reverseIterator() : _dataSet.iterator());
- }
- private class FilterIterator implements ITableIterator
- {
- private final ITableIterator _iterator;
- public FilterIterator(final ITableIterator _iterator)
- {
- this._iterator = _iterator;
- }
- @Override
- public boolean next() throws DataSetException
- {
- return _iterator.next();
- }
- @Override
- public ITableMetaData getTableMetaData() throws DataSetException
- {
- return getTable().getTableMetaData();
- }
- @Override
- public ITable getTable() throws DataSetException
- {
- return _decoratorFunction.apply(_iterator.getTable());
- }
- }
- @FunctionalInterface
- public interface TableDecoratorFunction
- {
- ITable apply(ITable table) throws DataSetException;
- }
- }