UuidType.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.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 UUID type and Strings.
  37.  *
  38.  * @author Jarvis Cochrane (jarvis@cochrane.com.au)
  39.  * @author Last changed by: $Author$
  40.  * @since Apr 27, 2009
  41.  */
  42. public class UuidType
  43.         extends AbstractDataType {

  44.     /**
  45.      * Logger for this class
  46.      */
  47.     private static final Logger logger = LoggerFactory.getLogger(UuidType.class);

  48.     public UuidType() {
  49.         super("uuid", Types.OTHER, String.class, false);
  50.     }

  51.     public Object getSqlValue(int column, ResultSet resultSet) throws SQLException, TypeCastException {
  52.         return resultSet.getString(column);
  53.     }

  54.     public void setSqlValue(Object uuid, int column,
  55.                             PreparedStatement statement) throws SQLException, TypeCastException {
  56.         statement.setObject(column, getUUID(uuid, statement.getConnection()));
  57.     }

  58.     public Object typeCast(Object arg0) throws TypeCastException {
  59.         return (arg0 == null) ? null : arg0.toString();
  60.     }

  61.     private Object getUUID(Object value, Connection connection) throws TypeCastException {

  62.         logger.debug("getUUID(value={}, connection={}) - start", value, connection);

  63.         Object tempUUID = null;

  64.         try {
  65.             Class aPGObjectClass = super.loadClass("org.postgresql.util.PGobject", connection);
  66.             Constructor ct = aPGObjectClass.getConstructor(null);
  67.             tempUUID = ct.newInstance(null);

  68.             Method setTypeMethod = aPGObjectClass.getMethod("setType", new Class[]{String.class});
  69.             setTypeMethod.invoke(tempUUID, new Object[]{"uuid"});

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

  72.         } catch (ClassNotFoundException e) {
  73.             throw new TypeCastException(value, this, e);
  74.         } catch (InvocationTargetException e) {
  75.             throw new TypeCastException(value, this, e);
  76.         } catch (NoSuchMethodException e) {
  77.             throw new TypeCastException(value, this, e);
  78.         } catch (IllegalAccessException e) {
  79.             throw new TypeCastException(value, this, e);
  80.         } catch (InstantiationException e) {
  81.             throw new TypeCastException(value, this, e);
  82.         }

  83.         return tempUUID;
  84.     }
  85. }