InetType.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 inet type and Strings.
  37.  *
  38.  * @author Angelo Dipierro (suCrabu@gmail.com)
  39.  */
  40. public class InetType
  41.         extends AbstractDataType {

  42.     /**
  43.      * Logger for this class
  44.      */
  45.     private static final Logger logger = LoggerFactory.getLogger(InetType.class);

  46.     public InetType() {
  47.         super("inet", Types.OTHER, String.class, false);
  48.     }

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

  52.     public void setSqlValue(Object uuid, int column,
  53.                             PreparedStatement statement) throws SQLException, TypeCastException {
  54.         statement.setObject(column, getInet(uuid, statement.getConnection()));
  55.     }

  56.     public Object typeCast(Object arg0) throws TypeCastException {
  57.         return arg0.toString();
  58.     }

  59.     private Object getInet(Object value, Connection connection) throws TypeCastException {

  60.         logger.debug("getInet(value={}, connection={}) - start", value, connection);

  61.         Object tempInet = null;

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

  66.             Method setTypeMethod = aPGObjectClass.getMethod("setType", new Class[]{String.class});
  67.             setTypeMethod.invoke(tempInet, new Object[]{"inet"});

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

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

  81.         return tempInet;
  82.     }
  83. }