IFileOutputStream.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.mapred;
  19. import java.io.IOException;
  20. import java.io.OutputStream; 
  21. import java.io.FilterOutputStream;
  22. import org.apache.hadoop.util.DataChecksum;
  23. /**
  24.  * A Checksum output stream.
  25.  * Checksum for the contents of the file is calculated and
  26.  * appended to the end of the file on close of the stream.
  27.  * Used for IFiles
  28.  */
  29. class IFileOutputStream extends FilterOutputStream {
  30.   /**
  31.    * The output stream to be checksummed. 
  32.    */
  33.   private final DataChecksum sum;
  34.   private byte[] barray;
  35.   private boolean closed = false;
  36.   private boolean finished = false;
  37.   /**
  38.    * Create a checksum output stream that writes
  39.    * the bytes to the given stream.
  40.    * @param out
  41.    */
  42.   public IFileOutputStream(OutputStream out) {
  43.     super(out);
  44.     sum = DataChecksum.newDataChecksum(DataChecksum.CHECKSUM_CRC32,
  45.         Integer.MAX_VALUE);
  46.     barray = new byte[sum.getChecksumSize()];
  47.   }
  48.   
  49.   @Override
  50.   public void close() throws IOException {
  51.     if (closed) {
  52.       return;
  53.     }
  54.     closed = true;
  55.     finish();
  56.     out.close();
  57.   }
  58.   /**
  59.    * Finishes writing data to the output stream, by writing
  60.    * the checksum bytes to the end. The underlying stream is not closed.
  61.    * @throws IOException
  62.    */
  63.   public void finish() throws IOException {
  64.     if (finished) {
  65.       return;
  66.     }
  67.     finished = true;
  68.     sum.writeValue(barray, 0, false);
  69.     out.write (barray, 0, sum.getChecksumSize());
  70.     out.flush();
  71.   }
  72.   /**
  73.    * Write bytes to the stream.
  74.    */
  75.   @Override
  76.   public void write(byte[] b, int off, int len) throws IOException {
  77.     sum.update(b, off,len);
  78.     out.write(b,off,len);
  79.   }
  80.  
  81.   @Override
  82.   public void write(int b) throws IOException {
  83.     barray[0] = (byte) (b & 0xFF);
  84.     write(barray,0,1);
  85.   }
  86. }