BlocksWithLocations.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.hdfs.server.protocol;
  19. import java.io.DataInput;
  20. import java.io.DataOutput;
  21. import java.io.IOException;
  22. import org.apache.hadoop.hdfs.protocol.Block;
  23. import org.apache.hadoop.io.Text;
  24. import org.apache.hadoop.io.Writable;
  25. import org.apache.hadoop.io.WritableUtils;
  26. /** A class to implement an array of BlockLocations
  27.  *  It provide efficient customized serialization/deserialization methods
  28.  *  in stead of using the default array (de)serialization provided by RPC
  29.  */
  30. public class BlocksWithLocations implements Writable {
  31.   /**
  32.    * A class to keep track of a block and its locations
  33.    */
  34.   public static class BlockWithLocations  implements Writable {
  35.     Block block;
  36.     String datanodeIDs[];
  37.     
  38.     /** default constructor */
  39.     public BlockWithLocations() {
  40.       block = new Block();
  41.       datanodeIDs = null;
  42.     }
  43.     
  44.     /** constructor */
  45.     public BlockWithLocations(Block b, String[] datanodes) {
  46.       block = b;
  47.       datanodeIDs = datanodes;
  48.     }
  49.     
  50.     /** get the block */
  51.     public Block getBlock() {
  52.       return block;
  53.     }
  54.     
  55.     /** get the block's locations */
  56.     public String[] getDatanodes() {
  57.       return datanodeIDs;
  58.     }
  59.     
  60.     /** deserialization method */
  61.     public void readFields(DataInput in) throws IOException {
  62.       block.readFields(in);
  63.       int len = WritableUtils.readVInt(in); // variable length integer
  64.       datanodeIDs = new String[len];
  65.       for(int i=0; i<len; i++) {
  66.         datanodeIDs[i] = Text.readString(in);
  67.       }
  68.     }
  69.     
  70.     /** serialization method */
  71.     public void write(DataOutput out) throws IOException {
  72.       block.write(out);
  73.       WritableUtils.writeVInt(out, datanodeIDs.length); // variable length int
  74.       for(String id:datanodeIDs) {
  75.         Text.writeString(out, id);
  76.       }
  77.     }
  78.   }
  79.   private BlockWithLocations[] blocks;
  80.   /** default constructor */
  81.   BlocksWithLocations() {
  82.   }
  83.   /** Constructor with one parameter */
  84.   public BlocksWithLocations( BlockWithLocations[] blocks ) {
  85.     this.blocks = blocks;
  86.   }
  87.   /** getter */
  88.   public BlockWithLocations[] getBlocks() {
  89.     return blocks;
  90.   }
  91.   /** serialization method */
  92.   public void write( DataOutput out ) throws IOException {
  93.     WritableUtils.writeVInt(out, blocks.length);
  94.     for(int i=0; i<blocks.length; i++) {
  95.       blocks[i].write(out);
  96.     }
  97.   }
  98.   /** deserialization method */
  99.   public void readFields(DataInput in) throws IOException {
  100.     int len = WritableUtils.readVInt(in);
  101.     blocks = new BlockWithLocations[len];
  102.     for(int i=0; i<len; i++) {
  103.       blocks[i] = new BlockWithLocations();
  104.       blocks[i].readFields(in);
  105.     }
  106.   }
  107. }