Merge pull request #411 from arenadata/add_decimal_rowbinary_stream

add support for decimal types in ClickHouseRowBinaryStream

Author: alex-krash

src/main/java/ru/yandex/clickhouse/util/ClickHouseRowBinaryStream.java

@@ -8,6 +8,8 @@
 
 import java.io.IOException;
 import java.io.OutputStream;
+import java.math.BigDecimal;
+import java.math.BigInteger;
 import java.nio.ByteBuffer;
 import java.nio.ByteOrder;
 import java.util.Date;
@@ -51,6 +53,7 @@ public void writeUnsignedLeb128(int value) throws IOException {
     /**
      * Dangerous. Can only be used for rare optimizations, for example when the string is written in parts
      * without prior concatenation. The size of the string in bytes must be passed through writeUnsignedLeb128.
+     *
      * @param bytes byte array will be written into stream
      * @throws IOException in case if an I/O error occurs
      */
@@ -163,19 +166,41 @@ public void writeFloat64(double value) throws IOException {
         out.writeDouble(value);
     }
 
+    private BigInteger removeComma(BigDecimal num, int scale) {
+        BigDecimal ten = BigDecimal.valueOf(10);
+        BigDecimal s = ten.pow(scale);
+        return num.multiply(s).toBigInteger();
+    }
+
+    public void writeDecimal128(BigDecimal num, int scale) throws IOException {
+        BigInteger bi = removeComma(num, scale);
+        byte[] r = bi.toByteArray();
+        for (int i = r.length; i > 0; i--) {
+            out.write(r[i - 1]);
+        }
+        out.write(new byte[16 - r.length]);
+    }
+
+    public void writeDecimal64(BigDecimal num, int scale) throws IOException {
+        out.writeLong(removeComma(num, scale).longValue());
+    }
+
+    public void writeDecimal32(BigDecimal num, int scale) throws IOException {
+        out.writeInt(removeComma(num, scale).intValue());
+    }
 
     public void writeDateArray(Date[] dates) throws IOException {
         Preconditions.checkNotNull(dates);
         writeUnsignedLeb128(dates.length);
-        for (Date date: dates) {
+        for (Date date : dates) {
             writeDate(date);
         }
     }
 
     public void writeDateTimeArray(Date[] dates) throws IOException {
         Preconditions.checkNotNull(dates);
         writeUnsignedLeb128(dates.length);
-        for (Date date: dates) {
+        for (Date date : dates) {
             writeDateTime(date);
         }
     }
@@ -191,92 +216,94 @@ public void writeStringArray(String[] strings) throws IOException {
     public void writeInt8Array(byte[] bytes) throws IOException {
         Preconditions.checkNotNull(bytes);
         writeUnsignedLeb128(bytes.length);
-        for (byte b: bytes) {
+        for (byte b : bytes) {
             writeInt8(b);
         }
     }
+
     public void writeInt8Array(int[] ints) throws IOException {
         Preconditions.checkNotNull(ints);
         writeUnsignedLeb128(ints.length);
-        for (int i: ints) {
+        for (int i : ints) {
             writeInt8(i);
         }
     }
 
     public void writeUInt8Array(int[] ints) throws IOException {
         Preconditions.checkNotNull(ints);
         writeUnsignedLeb128(ints.length);
-        for (int i: ints) {
+        for (int i : ints) {
             writeUInt8(i);
         }
     }
 
     public void writeInt16Array(short[] shorts) throws IOException {
         Preconditions.checkNotNull(shorts);
         writeUnsignedLeb128(shorts.length);
-        for (short s: shorts) {
+        for (short s : shorts) {
             writeInt16(s);
         }
     }
 
     public void writeUInt16Array(int[] ints) throws IOException {
         Preconditions.checkNotNull(ints);
         writeUnsignedLeb128(ints.length);
-        for (int i: ints) {
+        for (int i : ints) {
             writeUInt16(i);
         }
     }
 
     public void writeInt32Array(int[] ints) throws IOException {
         Preconditions.checkNotNull(ints);
         writeUnsignedLeb128(ints.length);
-        for (int i: ints) {
+        for (int i : ints) {
             writeInt32(i);
         }
     }
 
     public void writeUInt32Array(long[] longs) throws IOException {
         Preconditions.checkNotNull(longs);
         writeUnsignedLeb128(longs.length);
-        for (long l: longs) {
+        for (long l : longs) {
             writeUInt32(l);
         }
     }
 
     public void writeInt64Array(long[] longs) throws IOException {
         Preconditions.checkNotNull(longs);
         writeUnsignedLeb128(longs.length);
-        for (long l: longs) {
+        for (long l : longs) {
             writeInt64(l);
         }
     }
 
     public void writeUInt64Array(long[] longs) throws IOException {
         Preconditions.checkNotNull(longs);
         writeUnsignedLeb128(longs.length);
-        for (long l: longs) {
+        for (long l : longs) {
             writeUInt64(l);
         }
     }
 
     public void writeFloat32Array(float[] floats) throws IOException {
         Preconditions.checkNotNull(floats);
         writeUnsignedLeb128(floats.length);
-        for (float f: floats) {
+        for (float f : floats) {
             writeFloat32(f);
         }
     }
 
     public void writeFloat64Array(double[] doubles) throws IOException {
         Preconditions.checkNotNull(doubles);
         writeUnsignedLeb128(doubles.length);
-        for (double d: doubles) {
+        for (double d : doubles) {
             writeFloat64(d);
         }
     }
 
-    /** Write a marker indicating if value is nullable or not.
-     *
+    /**
+     * Write a marker indicating if value is nullable or not.
+     * <p>
      * E.g., to write Nullable(Int32):
      *
      * <pre>