1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
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
33
34
35
36
37 public class DatabaseTableIterator implements ITableIterator
38 {
39
40
41
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
59
60 public boolean next() throws DataSetException
61 {
62 logger.debug("next() - start");
63
64 _index++;
65
66
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 }