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

[do-not-reply at jboss.org]

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 at 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 at 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 at 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);
+      
+   }
+
+}