View Javadoc
1   /*
2    *
3    * The DbUnit Database Testing Framework
4    * Copyright (C)2008, 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.ext.h2;
22  
23  import java.util.Arrays;
24  import java.util.Collection;
25  
26  import org.dbunit.dataset.datatype.DataType;
27  import org.dbunit.dataset.datatype.DataTypeException;
28  import org.dbunit.dataset.datatype.DefaultDataTypeFactory;
29  import org.slf4j.Logger;
30  import org.slf4j.LoggerFactory;
31  
32  /**
33   * Specialized factory that recognizes H2 data types.
34   *
35   * @author Felipe Leme
36   * @author Last changed by: $Author$
37   * @version $Revision$ $Date$
38   * @since 2.2.1
39   */
40  public class H2DataTypeFactory extends DefaultDataTypeFactory
41  {
42      /**
43       * Logger for this class
44       */
45      private static final Logger logger = LoggerFactory.getLogger(H2DataTypeFactory.class);
46      /**
47       * Database product names supported.
48       */
49      private static final Collection DATABASE_PRODUCTS = Arrays.asList(new String[] {"h2"});
50  
51      /**
52       * @see org.dbunit.dataset.datatype.IDbProductRelatable#getValidDbProducts()
53       */
54      @Override
55      public Collection getValidDbProducts()
56      {
57          return DATABASE_PRODUCTS;
58      }
59  
60      @Override
61      public DataType createDataType(int sqlType, String sqlTypeName) throws DataTypeException
62      {
63          if(logger.isDebugEnabled())
64          {
65              logger.debug("createDataType(sqlType={}, sqlTypeName={}) - start", String.valueOf(sqlType), sqlTypeName);
66          }
67  
68          if (sqlTypeName.equals("BOOLEAN"))
69          {
70              return DataType.BOOLEAN;
71          } else if ("UUID".equals(sqlTypeName))
72          {
73              return DataType.NVARCHAR;
74          }
75  
76          return super.createDataType(sqlType, sqlTypeName);
77      }
78  }