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: package org.dbunit.database;
22:
23: import org.slf4j.Logger;
24: import org.slf4j.LoggerFactory;
25:
26: import org.dbunit.dataset.*;
27:
28: /**
29: * @author Manuel Laflamme
30: * @since Apr 12, 2003
31: * @version $Revision: 554 $
32: */
33: public class DatabaseTableIterator implements ITableIterator {
34:
35: /**
36: * Logger for this class
37: */
38: private static final Logger logger = LoggerFactory
39: .getLogger(DatabaseTableIterator.class);
40:
41: private final String[] _tableNames;
42: private final IDataSet _dataSet;
43: private IResultSetTable _currentTable;
44: private int _index = -1;
45:
46: public DatabaseTableIterator(String[] tableNames, IDataSet dataSet) {
47: _tableNames = tableNames;
48: _dataSet = dataSet;
49: _currentTable = null;
50: }
51:
52: ////////////////////////////////////////////////////////////////////////////
53: // ITableIterator interface
54:
55: public boolean next() throws DataSetException {
56: logger.debug("next() - start");
57:
58: _index++;
59:
60: // Ensure previous table is closed
61: if (_currentTable != null) {
62: _currentTable.close();
63: _currentTable = null;
64: }
65:
66: return _index < _tableNames.length;
67: }
68:
69: public ITableMetaData getTableMetaData() throws DataSetException {
70: logger.debug("getTableMetaData() - start");
71:
72: return _dataSet.getTableMetaData(_tableNames[_index]);
73: }
74:
75: public ITable getTable() throws DataSetException {
76: logger.debug("getTable() - start");
77:
78: if (_currentTable == null) {
79: _currentTable = (IResultSetTable) _dataSet
80: .getTable(_tableNames[_index]);
81: }
82: return _currentTable;
83: }
84: }
|