Github user xuchuanyin commented on a diff in the pull request:
https://github.com/apache/carbondata/pull/2847#discussion_r227615128
--- Diff:
core/src/main/java/org/apache/carbondata/core/datastore/compression/GzipCompressor.java
---
@@ -0,0 +1,201 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements. See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership.
+ * The ASF licenses this file to You under the Apache License, Version 2.0
+ * (the "License"); you may not use this file except in compliance with
+ * the License. You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.apache.carbondata.core.datastore.compression;
+
+import java.io.ByteArrayInputStream;
+import java.io.ByteArrayOutputStream;
+import java.io.IOException;
+import java.nio.ByteBuffer;
+import java.nio.DoubleBuffer;
+import java.nio.FloatBuffer;
+import java.nio.IntBuffer;
+import java.nio.LongBuffer;
+import java.nio.ShortBuffer;
+
+import org.apache.carbondata.core.util.ByteUtil;
+
+import
org.apache.commons.compress.compressors.gzip.GzipCompressorInputStream;
+import
org.apache.commons.compress.compressors.gzip.GzipCompressorOutputStream;
+
+public class GzipCompressor implements Compressor {
+
+ public GzipCompressor() {
+ }
+
+ @Override public String getName() {
+ return "gzip";
+ }
+
+ /*
+ * Method called for compressing the data and
+ * return a byte array
+ */
+ private byte[] compressData(byte[] data) {
+
+ ByteArrayOutputStream bt = new ByteArrayOutputStream();
+ try {
+ GzipCompressorOutputStream gzos = new GzipCompressorOutputStream(bt);
+ try {
+ gzos.write(data);
+ } catch (IOException e) {
+ e.printStackTrace();
+ } finally {
+ gzos.close();
+ }
+ } catch (IOException e) {
+ e.printStackTrace();
+ }
+
+ return bt.toByteArray();
+ }
+
+ /*
+ * Method called for decompressing the data and
+ * return a byte array
+ */
+ private byte[] decompressData(byte[] data) {
+
+ ByteArrayInputStream bt = new ByteArrayInputStream(data);
+ ByteArrayOutputStream bot = new ByteArrayOutputStream();
+
+ try {
+ GzipCompressorInputStream gzis = new GzipCompressorInputStream(bt);
+ byte[] buffer = new byte[1024];
+ int len;
+
+ while ((len = gzis.read(buffer)) != -1) {
+ bot.write(buffer, 0, len);
+ }
+
+ } catch (IOException e) {
+ e.printStackTrace();
+ }
+
+ return bot.toByteArray();
+ }
+
+ @Override public byte[] compressByte(byte[] unCompInput) {
+ return compressData(unCompInput);
+ }
+
+ @Override public byte[] compressByte(byte[] unCompInput, int byteSize) {
+ return compressData(unCompInput);
+ }
+
+ @Override public byte[] unCompressByte(byte[] compInput) {
+ return decompressData(compInput);
+ }
+
+ @Override public byte[] unCompressByte(byte[] compInput, int offset, int
length) {
+ byte[] data = new byte[length];
+ System.arraycopy(compInput, offset, data, 0, length);
+ return decompressData(data);
+ }
+
+ @Override public byte[] compressShort(short[] unCompInput) {
+ ByteBuffer unCompBuffer = ByteBuffer.allocate(unCompInput.length *
ByteUtil.SIZEOF_SHORT);
+ unCompBuffer.asShortBuffer().put(unCompInput);
+ return compressData(unCompBuffer.array());
+ }
+
+ @Override public short[] unCompressShort(byte[] compInput, int offset,
int length) {
+ byte[] unCompArray = unCompressByte(compInput, offset, length);
+ ShortBuffer unCompBuffer =
ByteBuffer.wrap(unCompArray).asShortBuffer();
+ short[] shorts = new short[unCompArray.length / ByteUtil.SIZEOF_SHORT];
+ unCompBuffer.get(shorts);
+ return shorts;
+ }
+
+ @Override public byte[] compressInt(int[] unCompInput) {
+ ByteBuffer unCompBuffer = ByteBuffer.allocate(unCompInput.length *
ByteUtil.SIZEOF_INT);
+ unCompBuffer.asIntBuffer().put(unCompInput);
+ return compressData(unCompBuffer.array());
+ }
+
+ @Override public int[] unCompressInt(byte[] compInput, int offset, int
length) {
+ byte[] unCompArray = unCompressByte(compInput, offset, length);
+ IntBuffer unCompBuffer = ByteBuffer.wrap(unCompArray).asIntBuffer();
+ int[] ints = new int[unCompArray.length / ByteUtil.SIZEOF_INT];
+ unCompBuffer.get(ints);
+ return ints;
+ }
+
+ @Override public byte[] compressLong(long[] unCompInput) {
+ ByteBuffer unCompBuffer = ByteBuffer.allocate(unCompInput.length *
ByteUtil.SIZEOF_LONG);
+ unCompBuffer.asLongBuffer().put(unCompInput);
+ return compressData(unCompBuffer.array());
+ }
+
+ @Override public long[] unCompressLong(byte[] compInput, int offset, int
length) {
+ byte[] unCompArray = unCompressByte(compInput, offset, length);
+ LongBuffer unCompBuffer = ByteBuffer.wrap(unCompArray).asLongBuffer();
+ long[] longs = new long[unCompArray.length / ByteUtil.SIZEOF_LONG];
+ unCompBuffer.get(longs);
+ return longs;
+ }
+
+ @Override public byte[] compressFloat(float[] unCompInput) {
+ ByteBuffer unCompBuffer = ByteBuffer.allocate(unCompInput.length *
ByteUtil.SIZEOF_FLOAT);
+ unCompBuffer.asFloatBuffer().put(unCompInput);
+ return compressData(unCompBuffer.array());
+ }
+
+ @Override public float[] unCompressFloat(byte[] compInput, int offset,
int length) {
+ byte[] unCompArray = unCompressByte(compInput, offset, length);
+ FloatBuffer unCompBuffer =
ByteBuffer.wrap(unCompArray).asFloatBuffer();
+ float[] floats = new float[unCompArray.length / ByteUtil.SIZEOF_FLOAT];
+ unCompBuffer.get(floats);
+ return floats;
+ }
+
+ @Override public byte[] compressDouble(double[] unCompInput) {
+ ByteBuffer unCompBuffer = ByteBuffer.allocate(unCompInput.length *
ByteUtil.SIZEOF_DOUBLE);
+ unCompBuffer.asDoubleBuffer().put(unCompInput);
+ return compressData(unCompBuffer.array());
+ }
+
+ @Override public double[] unCompressDouble(byte[] compInput, int offset,
int length) {
+ byte[] unCompArray = unCompressByte(compInput, offset, length);
+ DoubleBuffer unCompBuffer =
ByteBuffer.wrap(unCompArray).asDoubleBuffer();
+ double[] doubles = new double[unCompArray.length /
ByteUtil.SIZEOF_DOUBLE];
+ unCompBuffer.get(doubles);
+ return doubles;
+ }
+
+ @Override public long rawCompress(long inputAddress, int inputSize, long
outputAddress)
+ throws IOException {
+ throw new RuntimeException("Not implemented rawUncompress for gzip
yet");
+ }
+
+ @Override public long rawUncompress(byte[] input, byte[] output) throws
IOException {
+ //gzip api doesnt have rawCompress yet.
--- End diff --
if it is so, just throw exception, otherwise JVM may crash if you pass the
illegal address/length
---
