Fix for http://code.google.com/p/mybatis/issues/detail?id=124 - BigIntegerTypeHandler

Author: jeffgbutler

src/main/java/org/apache/ibatis/type/BigIntegerTypeHandler.java

@@ -0,0 +1,26 @@
+package org.apache.ibatis.type;
+
+import java.math.BigInteger;
+import java.sql.CallableStatement;
+import java.sql.PreparedStatement;
+import java.sql.ResultSet;
+import java.sql.SQLException;
+
+public class BigIntegerTypeHandler extends BaseTypeHandler {
+
+  public void setNonNullParameter(PreparedStatement ps, int i, Object parameter, JdbcType jdbcType)
+      throws SQLException {
+    BigInteger bigint = (BigInteger) parameter;
+    ps.setLong(i, bigint.longValue());
+  }
+
+  public Object getNullableResult(ResultSet rs, String columnName)
+      throws SQLException {
+    return BigInteger.valueOf(rs.getLong(columnName));
+  }
+
+  public Object getNullableResult(CallableStatement cs, int columnIndex)
+      throws SQLException {
+    return BigInteger.valueOf(cs.getLong(columnIndex));
+  }
+}

src/main/java/org/apache/ibatis/type/TypeHandlerRegistry.java

@@ -4,6 +4,7 @@
 
 import java.lang.annotation.Annotation;
 import java.math.BigDecimal;
+import java.math.BigInteger;
 import java.util.Date;
 import java.util.HashMap;
 import java.util.EnumMap;
@@ -73,9 +74,13 @@ public TypeHandlerRegistry() {
     register(JdbcType.NCHAR, new NStringTypeHandler());
     register(JdbcType.NCLOB, new NClobTypeHandler());
 
+    register(BigInteger.class, new BigIntegerTypeHandler());
+    register(JdbcType.BIGINT, new LongTypeHandler());
+    
     register(BigDecimal.class, new BigDecimalTypeHandler());
-    register(JdbcType.BIGINT, new BigDecimalTypeHandler());
     register(JdbcType.REAL, new BigDecimalTypeHandler());
+    register(JdbcType.DECIMAL, new BigDecimalTypeHandler());
+    register(JdbcType.NUMERIC, new BigDecimalTypeHandler());
 
     register(byte[].class, new ByteArrayTypeHandler());
     register(byte[].class, JdbcType.BLOB, new BlobTypeHandler());

src/test/java/org/apache/ibatis/submitted/dynsql/CreateDB.sql

@@ -11,3 +11,18 @@ insert into ibtest.names (id, description) values(3, 'Pebbles');
 insert into ibtest.names (id, description) values(4, 'Barney');
 insert into ibtest.names (id, description) values(5, 'Betty');
 insert into ibtest.names (id, description) values(6, 'Bamm Bamm');
+
+create table ibtest.numerics (
+  id int,
+  tinynumber tinyint,
+  smallnumber smallint,
+  longinteger bigint,
+  biginteger bigint,
+  numericnumber numeric(10,2),
+  decimalnumber decimal(10,2),
+  realnumber real,
+  floatnumber float,
+  doublenumber double
+);
+
+insert into ibtest.numerics values(1, 2, 3, 4, 5, 6, 7, 8, 9, 10);

src/test/java/org/apache/ibatis/submitted/dynsql/DynSql.xml

@@ -5,6 +5,19 @@
     "http://mybatis.org/dtd/mybatis-3-mapper.dtd">
 
 <mapper namespace="org.apache.ibatis.submitted.dynsql">
+  <resultMap type="org.apache.ibatis.submitted.dynsql.NumericRow" id="numericRow">
+    <result column="id" property="id" jdbcType="INTEGER"/>
+    <result column="tinynumber" property="tinynumber" jdbcType="TINYINT"/>
+    <result column="smallnumber" property="smallnumber" jdbcType="SMALLINT"/>
+    <result column="longinteger" property="longinteger" jdbcType="BIGINT"/>
+    <result column="biginteger" property="biginteger" jdbcType="BIGINT"/>
+    <result column="numericnumber" property="numericnumber" jdbcType="NUMERIC"/>
+    <result column="decimalnumber" property="decimalnumber" jdbcType="DECIMAL"/>
+    <result column="realnumber" property="realnumber" jdbcType="REAL"/>
+    <result column="floatnumber" property="floatnumber" jdbcType="FLOAT"/>
+    <result column="doublenumber" property="doublenumber" jdbcType="DOUBLE"/>
+  </resultMap>
+  
   <select id="select" parameterType="org.apache.ibatis.submitted.dynsql.Parameter" resultType="map">
     <if test="enabled">
       <foreach collection="ids" item="id" separator="union">
@@ -20,4 +33,8 @@
       select * from ${schema}.names where id = #{id}
     </foreach>
   </select>
+  
+  <select id="selectNumerics" resultMap="numericRow">
+    select * from ibtest.numerics
+  </select>
 </mapper>

src/test/java/org/apache/ibatis/submitted/dynsql/DynSqlTest.java

@@ -1,15 +1,19 @@
 package org.apache.ibatis.submitted.dynsql;
 
+import static org.junit.Assert.assertTrue;
+import static org.junit.Assert.assertEquals;
+
 import org.apache.ibatis.io.Resources;
 import org.apache.ibatis.jdbc.ScriptRunner;
 import org.apache.ibatis.session.SqlSession;
 import org.apache.ibatis.session.SqlSessionFactory;
 import org.apache.ibatis.session.SqlSessionFactoryBuilder;
-import static org.junit.Assert.assertTrue;
 import org.junit.BeforeClass;
 import org.junit.Test;
 
 import java.io.Reader;
+import java.math.BigDecimal;
+import java.math.BigInteger;
 import java.sql.Connection;
 import java.util.ArrayList;
 import java.util.List;
@@ -84,4 +88,29 @@ public void testSelectSimple() {
       sqlSession.close();
     }
   }
+
+  @Test
+  public void testNumerics() {
+    SqlSession sqlSession = sqlSessionFactory.openSession();
+    try {
+      List<NumericRow> answer = sqlSession.selectList("org.apache.ibatis.submitted.dynsql.selectNumerics");
+
+      assertTrue(answer.size() == 1);
+      
+      NumericRow row = answer.get(0);
+      assertEquals(1, (int) row.getId());
+      assertEquals(2, (int) row.getTinynumber());
+      assertEquals(3, (int) row.getSmallnumber());
+      assertEquals(4l, (long) row.getLonginteger());
+      assertEquals(new BigInteger("5"), row.getBiginteger());
+      assertEquals(new BigDecimal("6.00"), row.getNumericnumber());
+      assertEquals(new BigDecimal("7.00"), row.getDecimalnumber());
+      assertEquals((Float) 8.0f, row.getRealnumber());
+      assertEquals((Float) 9.0f, row.getFloatnumber());
+      assertEquals((Double) 10.0, row.getDoublenumber());
+      
+    } finally {
+      sqlSession.close();
+    }
+  }
 }

src/test/java/org/apache/ibatis/submitted/dynsql/NumericRow.java

@@ -0,0 +1,78 @@
+package org.apache.ibatis.submitted.dynsql;
+
+import java.math.BigDecimal;
+import java.math.BigInteger;
+
+
+public class NumericRow {
+    private Integer id;
+    private Byte tinynumber;
+    private Short smallnumber;
+    private Long longinteger;
+    private BigInteger biginteger;
+    private BigDecimal numericnumber;
+    private BigDecimal decimalnumber;
+    private Float realnumber;
+    private Float floatnumber;
+    private Double doublenumber;
+    public Integer getId() {
+        return id;
+    }
+    public void setId(Integer id) {
+        this.id = id;
+    }
+    public Byte getTinynumber() {
+        return tinynumber;
+    }
+    public void setTinynumber(Byte tinynumber) {
+        this.tinynumber = tinynumber;
+    }
+    public Short getSmallnumber() {
+        return smallnumber;
+    }
+    public void setSmallnumber(Short smallnumber) {
+        this.smallnumber = smallnumber;
+    }
+    public Long getLonginteger() {
+        return longinteger;
+    }
+    public void setLonginteger(Long longinteger) {
+        this.longinteger = longinteger;
+    }
+    public BigInteger getBiginteger() {
+        return biginteger;
+    }
+    public void setBiginteger(BigInteger biginteger) {
+        this.biginteger = biginteger;
+    }
+    public BigDecimal getNumericnumber() {
+        return numericnumber;
+    }
+    public void setNumericnumber(BigDecimal numericnumber) {
+        this.numericnumber = numericnumber;
+    }
+    public BigDecimal getDecimalnumber() {
+        return decimalnumber;
+    }
+    public void setDecimalnumber(BigDecimal decimalnumber) {
+        this.decimalnumber = decimalnumber;
+    }
+    public Float getRealnumber() {
+        return realnumber;
+    }
+    public void setRealnumber(Float realnumber) {
+        this.realnumber = realnumber;
+    }
+    public Float getFloatnumber() {
+        return floatnumber;
+    }
+    public void setFloatnumber(Float floatnumber) {
+        this.floatnumber = floatnumber;
+    }
+    public Double getDoublenumber() {
+        return doublenumber;
+    }
+    public void setDoublenumber(Double doublenumber) {
+        this.doublenumber = doublenumber;
+    }
+}