CaseInsensitiveDataSet.java

  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.dataset;

  22. import org.dbunit.database.AmbiguousTableNameException;
  23. import org.slf4j.Logger;
  24. import org.slf4j.LoggerFactory;

  25. /**
  26.  * Allows access to a decorated dataset in a case insensitive way. Dataset
  27.  * implementations provided by the framework are case sensitive. This class
  28.  * allows using them in situation where case sensitiveness is not desirable.
  29.  *
  30.  * @author Manuel Laflamme
  31.  * @author Last changed by: $Author$
  32.  * @version $Revision$ $Date$
  33.  * @deprecated All IDataSet implementations are case insensitive since DbUnit 1.5 - may change again since tablenames on RDBMSes can be case sensitive
  34.  * @since 1.0
  35.  */
  36. public class CaseInsensitiveDataSet extends AbstractDataSet
  37. {

  38.     /**
  39.      * Logger for this class
  40.      */
  41.     private static final Logger logger = LoggerFactory.getLogger(CaseInsensitiveDataSet.class);

  42.     private final IDataSet _dataSet;
  43.     private OrderedTableNameMap orderedTableMap;

  44.     public CaseInsensitiveDataSet(IDataSet dataSet) throws AmbiguousTableNameException, DataSetException
  45.     {
  46.         _dataSet = dataSet;
  47.        
  48.         // Check for duplicates using the OrderedTableNameMap as helper
  49.         orderedTableMap = new OrderedTableNameMap(false);
  50.         ITableIterator tableIterator = _dataSet.iterator();
  51.         while(tableIterator.next()) {
  52.             ITable table = (ITable) tableIterator.getTable();
  53.             String tableName = table.getTableMetaData().getTableName();
  54.             orderedTableMap.add(tableName.toUpperCase(), tableName);
  55.         }
  56.     }

  57.     private String getInternalTableName(String tableName) throws DataSetException
  58.     {
  59.         logger.debug("getInternalTableName(tableName={}) - start", tableName);

  60.         String originalTableName = (String)orderedTableMap.get(tableName.toUpperCase());
  61.         if(originalTableName==null){
  62.             throw new NoSuchTableException(tableName);
  63.         }
  64.         else {
  65.             return originalTableName;
  66.         }
  67.     }

  68.     ////////////////////////////////////////////////////////////////////////////
  69.     // AbstractDataSet class

  70.     protected ITableIterator createIterator(boolean reversed)
  71.             throws DataSetException
  72.     {
  73.         if(logger.isDebugEnabled())
  74.             logger.debug("createIterator(reversed={}) - start", String.valueOf(reversed));
  75.        
  76.         return new CaseInsensitiveIterator(reversed ?
  77.                 _dataSet.reverseIterator() : _dataSet.iterator());
  78.     }

  79.     ////////////////////////////////////////////////////////////////////////////
  80.     // IDataSet interface

  81.     public String[] getTableNames() throws DataSetException
  82.     {
  83.         return _dataSet.getTableNames();
  84.     }

  85.     public ITableMetaData getTableMetaData(String tableName)
  86.             throws DataSetException
  87.     {
  88.         logger.debug("getTableMetaData(tableName={}) - start", tableName);
  89.         return _dataSet.getTableMetaData(getInternalTableName(tableName));
  90.     }

  91.     public ITable getTable(String tableName) throws DataSetException
  92.     {
  93.         logger.debug("getTable(tableName={}) - start", tableName);
  94.         ITable table = _dataSet.getTable(getInternalTableName(tableName));
  95.         return new CaseInsensitiveTable(table);
  96.     }

  97.     ////////////////////////////////////////////////////////////////////////////
  98.     // CaseInsensitiveIterator class

  99.     private class CaseInsensitiveIterator implements ITableIterator
  100.     {
  101.         private final ITableIterator _iterator;

  102.         public CaseInsensitiveIterator(ITableIterator iterator)
  103.         {
  104.             _iterator = iterator;
  105.         }

  106.         ////////////////////////////////////////////////////////////////////////
  107.         // ITableIterator interface

  108.         public boolean next() throws DataSetException
  109.         {
  110.             return _iterator.next();
  111.         }

  112.         public ITableMetaData getTableMetaData() throws DataSetException
  113.         {
  114.             return _iterator.getTableMetaData();
  115.         }

  116.         public ITable getTable() throws DataSetException
  117.         {
  118.             return new CaseInsensitiveTable(_iterator.getTable());
  119.         }
  120.     }
  121. }