1 /*
2 *
3 * The DbUnit Database Testing Framework
4 * Copyright (C)2002-2004, DbUnit.org
5 *
6 * This library is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU Lesser General Public
8 * License as published by the Free Software Foundation; either
9 * 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.dbunit.dataset.DataSetException;
24 import org.dbunit.dataset.IDataSet;
25 import org.dbunit.dataset.ITable;
26 import org.dbunit.dataset.ITableIterator;
27 import org.dbunit.dataset.ITableMetaData;
28 import org.slf4j.Logger;
29 import org.slf4j.LoggerFactory;
30
31 /**
32 * @author Manuel Laflamme
33 * @author Last changed by: $Author: gommma $
34 * @version $Revision: 815 $ $Date: 2008-09-27 14:33:56 +0200 (sab, 27 set 2008) $
35 * @since 1.x (Apr 12, 2003)
36 */
37 public class DatabaseTableIterator implements ITableIterator
38 {
39
40 /**
41 * Logger for this class
42 */
43 private static final Logger logger = LoggerFactory.getLogger(DatabaseTableIterator.class);
44
45 private final String[] _tableNames;
46 private final IDataSet _dataSet;
47 private IResultSetTable _currentTable;
48 private int _index = -1;
49
50 public DatabaseTableIterator(String[] tableNames, IDataSet dataSet)
51 {
52 _tableNames = tableNames;
53 _dataSet = dataSet;
54 _currentTable = null;
55 }
56
57 ////////////////////////////////////////////////////////////////////////////
58 // ITableIterator interface
59
60 public boolean next() throws DataSetException
61 {
62 logger.debug("next() - start");
63
64 _index++;
65
66 // Ensure previous table is closed
67 if (_currentTable != null)
68 {
69 _currentTable.close();
70 _currentTable = null;
71 }
72
73 return _index < _tableNames.length;
74 }
75
76 public ITableMetaData getTableMetaData() throws DataSetException
77 {
78 logger.debug("getTableMetaData() - start");
79
80 return _dataSet.getTableMetaData(_tableNames[_index]);
81 }
82
83 public ITable getTable() throws DataSetException
84 {
85 logger.debug("getTable() - start");
86
87 if (_currentTable == null)
88 {
89 _currentTable = (IResultSetTable)_dataSet.getTable(_tableNames[_index]);
90 }
91 return _currentTable;
92 }
93 }