View Javadoc
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$
34   * @version $Revision$ $Date$
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  }