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.ext.mssql;
22  
23  import java.sql.Types;
24  import java.util.Arrays;
25  import java.util.Collection;
26  
27  import org.dbunit.dataset.datatype.DataType;
28  import org.dbunit.dataset.datatype.DataTypeException;
29  import org.dbunit.dataset.datatype.DefaultDataTypeFactory;
30  import org.slf4j.Logger;
31  import org.slf4j.LoggerFactory;
32  
33  /**
34   * Specialized factory that recognizes MS SQL Server data types.
35   *
36   * @author Manuel Laflamme
37   * @since May 19, 2003
38   * @version $Revision$
39   */
40  public class MsSqlDataTypeFactory extends DefaultDataTypeFactory
41  {
42  
43      /**
44       * Logger for this class
45       */
46      private static final Logger logger = LoggerFactory.getLogger(MsSqlDataTypeFactory.class);
47      /**
48       * Database product names supported.
49       */
50      private static final Collection DATABASE_PRODUCTS = Arrays.asList(new String[] {"mssql", "Microsoft SQL Server"});
51  
52      public static final int NCHAR = -8;
53      public static final int NVARCHAR = -9;
54      public static final int NTEXT = -10;
55      public static final int NTEXT_MSSQL_2005 = -16;
56      
57      /**
58       * @see org.dbunit.dataset.datatype.IDbProductRelatable#getValidDbProducts()
59       */
60      public Collection getValidDbProducts()
61      {
62        return DATABASE_PRODUCTS;
63      }
64  
65      public DataType createDataType(int sqlType, String sqlTypeName) throws DataTypeException
66      {
67      	if(logger.isDebugEnabled())
68      		logger.debug("createDataType(sqlType={}, sqlTypeName={}) - start", String.valueOf(sqlType), sqlTypeName);
69  
70          // TODO : Process MS SQL Server custom datatype here
71          if(sqlType == Types.CHAR)
72          {
73              if (UniqueIdentifierType.UNIQUE_IDENTIFIER_TYPE.equals(sqlTypeName))
74              {
75                  return new UniqueIdentifierType();
76              }
77          }
78  
79          switch(sqlType) {
80              case NCHAR: return DataType.CHAR; // nchar
81              case NVARCHAR: return DataType.VARCHAR; // nvarchar
82              case NTEXT: return DataType.LONGVARCHAR; // ntext
83              case NTEXT_MSSQL_2005: return DataType.LONGVARCHAR; // ntext
84              default: return super.createDataType(sqlType, sqlTypeName);
85          }
86      }
87  }