GenericEnumType.java

  1. /*
  2.  *
  3.  * The DbUnit Database Testing Framework
  4.  * Copyright (C)2002-2009, 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.postgresql;

  22. import java.lang.reflect.Constructor;
  23. import java.lang.reflect.InvocationTargetException;
  24. import java.lang.reflect.Method;
  25. import java.sql.Connection;
  26. import java.sql.PreparedStatement;
  27. import java.sql.ResultSet;
  28. import java.sql.SQLException;
  29. import java.sql.Types;

  30. import org.dbunit.dataset.datatype.AbstractDataType;
  31. import org.dbunit.dataset.datatype.TypeCastException;
  32. import org.slf4j.Logger;
  33. import org.slf4j.LoggerFactory;

  34. /**
  35.  * Adapter to handle conversion between Postgresql
  36.  * native Enum type and Strings.
  37.  *
  38.  * @author Jarvis Cochrane (jarvis@cochrane.com.au)
  39.  * @author gommma (gommma AT users.sourceforge.net)
  40.  * @author Last changed by: $Author$
  41.  * @version $Revision$ $Date$
  42.  * @since 2.4.6
  43.  */
  44. public class GenericEnumType extends AbstractDataType {

  45.     /**
  46.      * Logger for this class
  47.      */
  48.     private static final Logger logger = LoggerFactory.getLogger(GenericEnumType.class);
  49.    
  50.     private final String sqlTypeName;

  51.     /**
  52.      * @param sqlTypeName The name of the enum type needed to invoke the "setType()" method on
  53.      * the PGObject class.
  54.      */
  55.     public GenericEnumType(String sqlTypeName)
  56.     {
  57.         super(sqlTypeName, Types.OTHER, String.class, false);
  58.        
  59.         if (sqlTypeName == null) {
  60.             throw new NullPointerException(
  61.                     "The parameter 'sqlTypeName' must not be null");
  62.         }
  63.         this.sqlTypeName = sqlTypeName;
  64.     }

  65.     public Object getSqlValue(int column, ResultSet resultSet) throws SQLException, TypeCastException
  66.     {
  67.         return resultSet.getString(column);
  68.     }

  69.     public void setSqlValue(Object enumObject, int column,
  70.                             PreparedStatement statement) throws SQLException, TypeCastException
  71.     {
  72.         statement.setObject(column, getEnum(enumObject, statement.getConnection()));
  73.     }

  74.     public Object typeCast(Object arg0) throws TypeCastException
  75.     {
  76.         return arg0.toString();
  77.     }

  78.     private Object getEnum(Object value, Connection connection) throws TypeCastException {

  79.         logger.debug("getEnum(value={}, connection={}) - start", value, connection);

  80.         Object tempEnum = null;

  81.         try {
  82.             Class aPGObjectClass = super.loadClass("org.postgresql.util.PGobject", connection);
  83.             Constructor ct = aPGObjectClass.getConstructor(null);
  84.             tempEnum = ct.newInstance(null);

  85.             Method setTypeMethod = aPGObjectClass.getMethod("setType", new Class[]{String.class});
  86.             setTypeMethod.invoke(tempEnum, new Object[]{this.sqlTypeName});

  87.             Method setValueMethod = aPGObjectClass.getMethod("setValue", new Class[]{String.class});
  88.             setValueMethod.invoke(tempEnum, new Object[]{value.toString()});

  89.         } catch (ClassNotFoundException e) {
  90.             throw new TypeCastException(value, this, e);
  91.         } catch (InvocationTargetException e) {
  92.             throw new TypeCastException(value, this, e);
  93.         } catch (NoSuchMethodException e) {
  94.             throw new TypeCastException(value, this, e);
  95.         } catch (IllegalAccessException e) {
  96.             throw new TypeCastException(value, this, e);
  97.         } catch (InstantiationException e) {
  98.             throw new TypeCastException(value, this, e);
  99.         }

  100.         return tempEnum;
  101.     }

  102.     public String getSqlTypeName() {
  103.         return sqlTypeName;
  104.     }
  105.    
  106.    
  107. }