CompressorStream.java
上传用户:quxuerui
上传日期:2018-01-08
资源大小:41811k
文件大小:3k
源码类别:

网格计算

开发平台:

Java

  1. /*
  2.  * Licensed to the Apache Software Foundation (ASF) under one
  3.  * or more contributor license agreements.  See the NOTICE file
  4.  * distributed with this work for additional information
  5.  * regarding copyright ownership.  The ASF licenses this file
  6.  * to you under the Apache License, Version 2.0 (the
  7.  * "License"); you may not use this file except in compliance
  8.  * with the License.  You may obtain a copy of the License at
  9.  *
  10.  *     http://www.apache.org/licenses/LICENSE-2.0
  11.  *
  12.  * Unless required by applicable law or agreed to in writing, software
  13.  * distributed under the License is distributed on an "AS IS" BASIS,
  14.  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  15.  * See the License for the specific language governing permissions and
  16.  * limitations under the License.
  17.  */
  18. package org.apache.hadoop.io.compress;
  19. import java.io.IOException;
  20. import java.io.OutputStream;
  21. import org.apache.hadoop.io.compress.CompressionOutputStream;
  22. import org.apache.hadoop.io.compress.Compressor;
  23. public class CompressorStream extends CompressionOutputStream {
  24.   protected Compressor compressor;
  25.   protected byte[] buffer;
  26.   protected boolean closed = false;
  27.   
  28.   public CompressorStream(OutputStream out, Compressor compressor, int bufferSize) {
  29.     super(out);
  30.     if (out == null || compressor == null) {
  31.       throw new NullPointerException();
  32.     } else if (bufferSize <= 0) {
  33.       throw new IllegalArgumentException("Illegal bufferSize");
  34.     }
  35.     this.compressor = compressor;
  36.     buffer = new byte[bufferSize];
  37.   }
  38.   public CompressorStream(OutputStream out, Compressor compressor) {
  39.     this(out, compressor, 512);
  40.   }
  41.   
  42.   /**
  43.    * Allow derived classes to directly set the underlying stream.
  44.    * 
  45.    * @param out Underlying output stream.
  46.    */
  47.   protected CompressorStream(OutputStream out) {
  48.     super(out);
  49.   }
  50.   public void write(byte[] b, int off, int len) throws IOException {
  51.     // Sanity checks
  52.     if (compressor.finished()) {
  53.       throw new IOException("write beyond end of stream");
  54.     }
  55.     if ((off | len | (off + len) | (b.length - (off + len))) < 0) {
  56.       throw new IndexOutOfBoundsException();
  57.     } else if (len == 0) {
  58.       return;
  59.     }
  60.     compressor.setInput(b, off, len);
  61.     while (!compressor.needsInput()) {
  62.       compress();
  63.     }
  64.   }
  65.   protected void compress() throws IOException {
  66.     int len = compressor.compress(buffer, 0, buffer.length);
  67.     if (len > 0) {
  68.       out.write(buffer, 0, len);
  69.     }
  70.   }
  71.   public void finish() throws IOException {
  72.     if (!compressor.finished()) {
  73.       compressor.finish();
  74.       while (!compressor.finished()) {
  75.         compress();
  76.       }
  77.     }
  78.   }
  79.   public void resetState() throws IOException {
  80.     compressor.reset();
  81.   }
  82.   
  83.   public void close() throws IOException {
  84.     if (!closed) {
  85.       finish();
  86.       out.close();
  87.       closed = true;
  88.     }
  89.   }
  90.   private byte[] oneByte = new byte[1];
  91.   public void write(int b) throws IOException {
  92.     oneByte[0] = (byte)(b & 0xff);
  93.     write(oneByte, 0, oneByte.length);
  94.   }
  95. }