[hornetq-commits] JBoss hornetq SVN: r9951 - branches/Branch_Large_Message_Compression/src/main/org/hornetq/utils.

Author: gaohoward Date: 2010-11-30 09:30:46 -0500 (Tue, 30 Nov 2010) New Revision: 9951 Added: branches/Branch_Large_Message_Compression/src/main/org/hornetq/utils/InflaterWriter.java Modified: branches/Branch_Large_Message_Compression/src/main/org/hornetq/utils/DeflaterReader.java branches/Branch_Large_Message_Compression/src/main/org/hornetq/utils/InflaterReader.java Log: added InflaterWriter Modified: branches/Branch_Large_Message_Compression/src/main/org/hornetq/utils/DeflaterReader.java =================================================================== --- branches/Branch_Large_Message_Compression/src/main/org/hornetq/utils/DeflaterReader.java 2010-11-30 11:51:12 UTC (rev 9950) +++ branches/Branch_Large_Message_Compression/src/main/org/hornetq/utils/DeflaterReader.java 2010-11-30 14:30:46 UTC (rev 9951) @@ -22,6 +22,8 @@ * A DeflaterReader * The reader takes an inputstream and compress it. * Not for concurrent use. + + * @author <a href="mailto:hgao@redhat.com">Howard Gao</a> * */ public class DeflaterReader Modified: branches/Branch_Large_Message_Compression/src/main/org/hornetq/utils/InflaterReader.java =================================================================== --- branches/Branch_Large_Message_Compression/src/main/org/hornetq/utils/InflaterReader.java 2010-11-30 11:51:12 UTC (rev 9950) +++ branches/Branch_Large_Message_Compression/src/main/org/hornetq/utils/InflaterReader.java 2010-11-30 14:30:46 UTC (rev 9951) @@ -26,6 +26,7 @@ * It takes an compressed input stream and decompressed it as it is being read. * Not for concurrent use. * + * @author <a href="mailto:hgao@redhat.com">Howard Gao</a> */ public class InflaterReader extends InputStream { Added: branches/Branch_Large_Message_Compression/src/main/org/hornetq/utils/InflaterWriter.java =================================================================== --- branches/Branch_Large_Message_Compression/src/main/org/hornetq/utils/InflaterWriter.java (rev 0) +++ branches/Branch_Large_Message_Compression/src/main/org/hornetq/utils/InflaterWriter.java 2010-11-30 14:30:46 UTC (rev 9951) @@ -0,0 +1,147 @@ +/* + * Copyright 2010 Red Hat, Inc. + * Red Hat 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.hornetq.utils; + +import java.io.ByteArrayInputStream; +import java.io.ByteArrayOutputStream; +import java.io.IOException; +import java.io.OutputStream; +import java.util.zip.DataFormatException; +import java.util.zip.Deflater; +import java.util.zip.Inflater; + +/** + * A InflaterWriter + * + * This class takes an OutputStream. Compressed bytes + * can directly be written into this class. The class will + * decompress the bytes and write them to the output stream. + * + * Not for concurrent use. + * + * @author <a href="mailto:hgao@redhat.com">Howard Gao</a> + * + */ +public class InflaterWriter extends OutputStream +{ + private Inflater inflater = new Inflater(); + private OutputStream output; + + private byte[] writeBuffer = new byte[1024]; + private int writePointer = 0; + + private byte[] outputBuffer = new byte[writeBuffer.length*2]; + + public InflaterWriter(OutputStream output) + { + this.output = output; + } + + /* + * Write a compressed byte. + */ + @Override + public void write(int b) throws IOException + { + writeBuffer[writePointer] = (byte)(b & 0xFF); + writePointer++; + + if (writePointer == writeBuffer.length) + { + writePointer = 0; + try + { + doWrite(); + } + catch (DataFormatException e) + { + throw new IOException("Error decompressing data", e); + } + } + } + + @Override + public void close() throws IOException + { + if (writePointer > 0) + { + inflater.setInput(writeBuffer, 0, writePointer); + try + { + int n = inflater.inflate(outputBuffer); + while (n > 0) + { + output.write(outputBuffer, 0, n); + n = inflater.inflate(outputBuffer); + } + output.close(); + } + catch (DataFormatException e) + { + throw new IOException(e); + } + } + } + + private void doWrite() throws DataFormatException, IOException + { + inflater.setInput(writeBuffer); + int n = inflater.inflate(outputBuffer); + + while (n > 0) + { + output.write(outputBuffer, 0, n); + n = inflater.inflate(outputBuffer); + } + } + + public static void main(String[] args) throws IOException + { + String inputString = "blahblahblah??blahblahblahblahblah??blablahblah??blablahblah??bla"; + byte[] input = inputString.getBytes("UTF-8"); + byte[] output = new byte[30]; + Deflater compresser = new Deflater(); + compresser.setInput(input); + compresser.finish(); + int compressedDataLength = compresser.deflate(output); + System.err.println("compress len: " + compressedDataLength); + + byte[] zipBytes = new byte[compressedDataLength]; + + System.arraycopy(output, 0, zipBytes, 0, compressedDataLength); + ByteArrayInputStream byteInput = new ByteArrayInputStream(zipBytes); + + ByteArrayOutputStream byteOutput = new ByteArrayOutputStream(); + InflaterWriter writer = new InflaterWriter(byteOutput); + + byte[] zipBuffer = new byte[12]; + + int n = byteInput.read(zipBuffer); + while (n > 0) + { + System.out.println("Writing: " + n); + writer.write(zipBuffer, 0, n); + n = byteInput.read(zipBuffer); + } + + writer.close(); + + byte[] outcome = byteOutput.toByteArray(); + String outStr = new String(outcome); + + System.out.println("Outcome: " + outStr); + + } + +}