support big decimal and big integer

taowen · taowen · commit b40fd4ff2665 · 2017-01-03T19:50:00.000+08:00
diff --git a/src/main/java/com/jsoniter/output/CodegenImplNative.java b/src/main/java/com/jsoniter/output/CodegenImplNative.java
@@ -8,6 +8,8 @@
 import java.io.IOException;
 import java.lang.reflect.ParameterizedType;
 import java.lang.reflect.Type;
+import java.math.BigDecimal;
+import java.math.BigInteger;
 import java.util.HashMap;
 import java.util.IdentityHashMap;
 import java.util.Map;
@@ -236,6 +238,31 @@ public Any wrap(Object obj) {
                 return JsonStream.wrap(obj);
             }
         });
+
+        put(BigDecimal.class, new Encoder() {
+            @Override
+            public void encode(Object obj, JsonStream stream) throws IOException {
+                BigDecimal val = (BigDecimal) obj;
+                stream.writeVal(val.toString());
+            }
+
+            @Override
+            public Any wrap(Object obj) {
+                return Any.wrap(obj.toString());
+            }
+        });
+        put(BigInteger.class, new Encoder() {
+            @Override
+            public void encode(Object obj, JsonStream stream) throws IOException {
+                BigInteger val = (BigInteger) obj;
+                stream.writeVal(val.toString());
+            }
+
+            @Override
+            public Any wrap(Object obj) {
+                return Any.wrap(obj.toString());
+            }
+        });
     }};
 
     public static String genWriteOp(String code, Type valueType) {
diff --git a/src/test/java/com/jsoniter/output/TestNative.java b/src/test/java/com/jsoniter/output/TestNative.java
@@ -4,6 +4,8 @@
 
 import java.io.ByteArrayOutputStream;
 import java.io.IOException;
+import java.math.BigDecimal;
+import java.math.BigInteger;
 
 public class TestNative extends TestCase {
 
@@ -110,4 +112,16 @@ public void test_boolean() throws IOException {
         stream.close();
         assertEquals("truefalse".replace('\'', '"'), baos.toString());
     }
+
+    public void test_big_decimal() throws IOException {
+        stream.writeVal(new BigDecimal("12.34"));
+        stream.close();
+        assertEquals("'12.34'".replace('\'', '"'), baos.toString());
+    }
+
+    public void test_big_integer() throws IOException {
+        stream.writeVal(new BigInteger("1234"));
+        stream.close();
+        assertEquals("'1234'".replace('\'', '"'), baos.toString());
+    }
 }
