01: /*
02: *
03: * The DbUnit Database Testing Framework
04: * Copyright (C)2002-2004, DbUnit.org
05: *
06: * This library is free software; you can redistribute it and/or
07: * modify it under the terms of the GNU Lesser General Public
08: * License as published by the Free Software Foundation; either
09: * version 2.1 of the License, or (at your option) any later version.
10: *
11: * This library is distributed in the hope that it will be useful,
12: * but WITHOUT ANY WARRANTY; without even the implied warranty of
13: * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14: * Lesser General Public License for more details.
15: *
16: * You should have received a copy of the GNU Lesser General Public
17: * License along with this library; if not, write to the Free Software
18: * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
19: *
20: */
21:
22: package org.dbunit.dataset;
23:
24: import org.slf4j.Logger;
25: import org.slf4j.LoggerFactory;
26:
27: import java.util.ArrayList;
28: import java.util.List;
29:
30: /**
31: * @author Manuel Laflamme
32: * @version $Revision: 554 $
33: * @since Feb 18, 2002
34: */
35: public class DefaultDataSet extends AbstractDataSet {
36:
37: /**
38: * Logger for this class
39: */
40: private static final Logger logger = LoggerFactory
41: .getLogger(DefaultDataSet.class);
42:
43: private final List _tableList = new ArrayList();
44:
45: public DefaultDataSet() {
46: }
47:
48: public DefaultDataSet(ITable table) {
49: addTable(table);
50: }
51:
52: public DefaultDataSet(ITable[] tables) {
53: for (int i = 0; i < tables.length; i++) {
54: addTable(tables[i]);
55: }
56: }
57:
58: public DefaultDataSet(ITable table1, ITable table2) {
59: addTable(table1);
60: addTable(table2);
61: }
62:
63: /**
64: * Add a new table in this dataset.
65: */
66: public void addTable(ITable table) {
67: logger.debug("addTable(table=" + table + ") - start");
68:
69: _tableList.add(table);
70: }
71:
72: ////////////////////////////////////////////////////////////////////////////
73: // AbstractDataSet class
74:
75: protected ITableIterator createIterator(boolean reversed)
76: throws DataSetException {
77: logger.debug("createIterator(reversed=" + reversed
78: + ") - start");
79:
80: ITable[] tables = (ITable[]) _tableList.toArray(new ITable[0]);
81: return new DefaultTableIterator(tables, reversed);
82: }
83: }
|