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.postgresql;
22  
23  import java.lang.reflect.Constructor;
24  import java.lang.reflect.InvocationTargetException;
25  import java.lang.reflect.Method;
26  import java.sql.Connection;
27  import java.sql.PreparedStatement;
28  import java.sql.ResultSet;
29  import java.sql.SQLException;
30  import java.sql.Types;
31  
32  import org.dbunit.dataset.datatype.AbstractDataType;
33  import org.dbunit.dataset.datatype.TypeCastException;
34  import org.slf4j.Logger;
35  import org.slf4j.LoggerFactory;
36  
37  /**
38   * Adapter to handle conversion between Postgresql
39   * native inet type and Strings.
40   *
41   * @author Angelo Dipierro (suCrabu@gmail.com)
42   */
43  public class InetType
44          extends AbstractDataType {
45  
46      /**
47       * Logger for this class
48       */
49      private static final Logger logger = LoggerFactory.getLogger(InetType.class);
50  
51      public InetType() {
52          super("inet", Types.OTHER, String.class, false);
53      }
54  
55      public Object getSqlValue(int column, ResultSet resultSet) throws SQLException, TypeCastException {
56          return resultSet.getString(column);
57      }
58  
59      public void setSqlValue(Object uuid, int column,
60                              PreparedStatement statement) throws SQLException, TypeCastException {
61          statement.setObject(column, getInet(uuid, statement.getConnection()));
62      }
63  
64      public Object typeCast(Object arg0) throws TypeCastException {
65          return arg0.toString();
66      }
67  
68      private Object getInet(Object value, Connection connection) throws TypeCastException {
69  
70          logger.debug("getInet(value={}, connection={}) - start", value, connection);
71  
72          Object tempInet = null;
73  
74          try {
75              Class aPGObjectClass = super.loadClass("org.postgresql.util.PGobject", connection);
76              Constructor ct = aPGObjectClass.getConstructor(null);
77              tempInet = ct.newInstance(null);
78  
79              Method setTypeMethod = aPGObjectClass.getMethod("setType", new Class[]{String.class});
80              setTypeMethod.invoke(tempInet, new Object[]{"inet"});
81  
82              Method setValueMethod = aPGObjectClass.getMethod("setValue", new Class[]{String.class});
83              setValueMethod.invoke(tempInet, new Object[]{value.toString()});
84  
85          } catch (ClassNotFoundException e) {
86              throw new TypeCastException(value, this, e);
87          } catch (InvocationTargetException e) {
88              throw new TypeCastException(value, this, e);
89          } catch (NoSuchMethodException e) {
90              throw new TypeCastException(value, this, e);
91          } catch (IllegalAccessException e) {
92              throw new TypeCastException(value, this, e);
93          } catch (InstantiationException e) {
94              throw new TypeCastException(value, this, e);
95          }
96  
97          return tempInet;
98      }
99  }