package omq.common.util;

import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.util.zip.GZIPInputStream;
import java.util.zip.GZIPOutputStream;

public class Zipper {

	public static byte[] zip(byte[] b) throws IOException {
		ByteArrayOutputStream baos = new ByteArrayOutputStream();

		GZIPOutputStream zos = null;
		try {
			zos = new GZIPOutputStream(baos);
			zos.write(b);
		} finally{
			if(zos != null){
				zos.close();
			}
			
			baos.close();
		}
		
		return baos.toByteArray();
	}

	public static byte[] unzip(byte[] b) throws IOException {
		ByteArrayOutputStream baos = new ByteArrayOutputStream();
		ByteArrayInputStream bais = new ByteArrayInputStream(b);

		GZIPInputStream zis = null;
		try {
			zis = new GZIPInputStream(bais);
			
			byte[] tmpBuffer = new byte[256];
			int n;
			while ((n = zis.read(tmpBuffer)) >= 0) {
				baos.write(tmpBuffer, 0, n);
			}
		} finally {		
			if(zis != null){
				zis.close();
			}
			
			bais.close();
			baos.close();
		}		
		
		return baos.toByteArray();
	}	
}
