001/**
002 * Licensed to the Apache Software Foundation (ASF) under one
003 * or more contributor license agreements.  See the NOTICE file
004 * distributed with this work for additional information
005 * regarding copyright ownership.  The ASF licenses this file
006 * to you under the Apache License, Version 2.0 (the
007 * "License"); you may not use this file except in compliance
008 * with the License.  You may obtain a copy of the License at
009 *
010 *      http://www.apache.org/licenses/LICENSE-2.0
011 *
012 * Unless required by applicable law or agreed to in writing, software
013 * distributed under the License is distributed on an "AS IS" BASIS,
014 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
015 * See the License for the specific language governing permissions and
016 * limitations under the License.
017 */
018package org.apache.oozie.compression;
019
020import java.io.ByteArrayOutputStream;
021import java.io.DataInputStream;
022import java.io.IOException;
023import java.util.zip.GZIPInputStream;
024import java.util.zip.GZIPOutputStream;
025import org.apache.commons.io.IOUtils;
026
027/**
028 * Class to compress and decompress data using Gzip codec
029 *
030 */
031public class GzipCompressionCodec implements CompressionCodec {
032
033    public static final String CODEC_NAME = "gz";
034
035    public byte[] compressBytes(byte[] header, byte[] data) throws IOException {
036        ByteArrayOutputStream byteOutput = new ByteArrayOutputStream(2000);
037        byteOutput.write(header);
038        GZIPOutputStream gzipOut = new GZIPOutputStream(byteOutput);
039        gzipOut.write(data);
040        gzipOut.close();
041        return byteOutput.toByteArray();
042    }
043
044    public byte[] compressString(byte[] header, String data) throws IOException {
045        ByteArrayOutputStream byteOutput = new ByteArrayOutputStream(1000);
046        byteOutput.write(header);
047        GZIPOutputStream gzipOut = new GZIPOutputStream(byteOutput);
048        gzipOut.write(data.getBytes(CodecFactory.UTF_8_ENCODING));
049        gzipOut.close();
050        return byteOutput.toByteArray();
051    }
052
053    public String decompressToString(DataInputStream dais) throws IOException {
054        GZIPInputStream gzipIn = new GZIPInputStream(dais);
055        String decompress = IOUtils.toString(gzipIn, CodecFactory.UTF_8_ENCODING);
056        gzipIn.close();
057        return decompress;
058    }
059
060    public byte[] decompressToBytes(DataInputStream dais) throws IOException {
061        GZIPInputStream gzipIn = new GZIPInputStream(dais);
062        byte[] decompress = IOUtils.toByteArray(gzipIn);
063        gzipIn.close();
064        return decompress;
065    }
066
067}