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  
22  package org.dbunit.ext.db2;
23  
24  import java.sql.Types;
25  import java.util.Arrays;
26  import java.util.Collection;
27  
28  import org.dbunit.dataset.datatype.DataType;
29  import org.dbunit.dataset.datatype.DataTypeException;
30  import org.dbunit.dataset.datatype.DefaultDataTypeFactory;
31  import org.dbunit.dataset.datatype.StringDataType;
32  import org.slf4j.Logger;
33  import org.slf4j.LoggerFactory;
34  
35  /**
36   * Specialized factory that recognizes DB2 data types.
37   *
38   * @author Federico Spinazzi
39   * @author Manuel Laflamme
40   * @since Jul 17, 2003
41   * @version $Revision$
42   */
43  public class Db2DataTypeFactory extends DefaultDataTypeFactory
44  {
45  
46      /**
47       * Logger for this class
48       */
49      private static final Logger logger = LoggerFactory.getLogger(Db2DataTypeFactory.class);
50  
51      /**
52       * Database product names supported.
53       */
54      private static final Collection DATABASE_PRODUCTS = Arrays.asList(new String[] {"db2"});
55  
56      static final DataType DB2XML_XMLVARCHAR = new StringDataType(
57              "DB2XML.XMLVARCHAR", Types.DISTINCT);
58      static final DataType DB2XML_XMLCLOB = new StringDataType(
59              "DB2XML.XMLCLOB", Types.DISTINCT);
60      static final DataType DB2XML_XMLFILE = new StringDataType(
61              "DB2XML.XMLFILE", Types.DISTINCT);
62  
63      /**
64       * @see org.dbunit.dataset.datatype.IDbProductRelatable#getValidDbProducts()
65       */
66      public Collection getValidDbProducts()
67      {
68        return DATABASE_PRODUCTS;
69      }
70  
71      public DataType createDataType(int sqlType, String sqlTypeName) throws DataTypeException
72      {
73      	if(logger.isDebugEnabled())
74      		logger.debug("createDataType(sqlType={}, sqlTypeName={}) - start", String.valueOf(sqlType), sqlTypeName);
75  
76          if (sqlType == Types.DISTINCT)
77          {
78              if (sqlTypeName.equals(DB2XML_XMLVARCHAR.toString()))
79              {
80                  return DB2XML_XMLVARCHAR;
81              }
82  
83              if (sqlTypeName.equals(DB2XML_XMLCLOB.toString()))
84              {
85                  return DB2XML_XMLCLOB;
86              }
87  
88              if (sqlTypeName.equals(DB2XML_XMLFILE.toString()))
89              {
90                  return DB2XML_XMLFILE;
91              }
92          }
93  
94          return super.createDataType(sqlType, sqlTypeName);
95      }
96  }