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

网格计算

开发平台:

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.*;
  20. import java.util.List;
  21. import org.apache.hadoop.hdfs.protocol.Block;
  22. import org.apache.hadoop.hdfs.protocol.DatanodeInfo;
  23. import org.apache.hadoop.hdfs.server.namenode.DatanodeDescriptor.BlockTargetPair;
  24. import org.apache.hadoop.io.*;
  25. /****************************************************
  26.  * A BlockCommand is an instruction to a datanode 
  27.  * regarding some blocks under its control.  It tells
  28.  * the DataNode to either invalidate a set of indicated
  29.  * blocks, or to copy a set of indicated blocks to 
  30.  * another DataNode.
  31.  * 
  32.  ****************************************************/
  33. public class BlockCommand extends DatanodeCommand {
  34.   Block blocks[];
  35.   DatanodeInfo targets[][];
  36.   public BlockCommand() {}
  37.   /**
  38.    * Create BlockCommand for transferring blocks to another datanode
  39.    * @param blocktargetlist    blocks to be transferred 
  40.    */
  41.   public BlockCommand(int action, List<BlockTargetPair> blocktargetlist) {
  42.     super(action);
  43.     blocks = new Block[blocktargetlist.size()]; 
  44.     targets = new DatanodeInfo[blocks.length][];
  45.     for(int i = 0; i < blocks.length; i++) {
  46.       BlockTargetPair p = blocktargetlist.get(i);
  47.       blocks[i] = p.block;
  48.       targets[i] = p.targets;
  49.     }
  50.   }
  51.   private static final DatanodeInfo[][] EMPTY_TARGET = {};
  52.   /**
  53.    * Create BlockCommand for the given action
  54.    * @param blocks blocks related to the action
  55.    */
  56.   public BlockCommand(int action, Block blocks[]) {
  57.     super(action);
  58.     this.blocks = blocks;
  59.     this.targets = EMPTY_TARGET;
  60.   }
  61.   public Block[] getBlocks() {
  62.     return blocks;
  63.   }
  64.   public DatanodeInfo[][] getTargets() {
  65.     return targets;
  66.   }
  67.   ///////////////////////////////////////////
  68.   // Writable
  69.   ///////////////////////////////////////////
  70.   static {                                      // register a ctor
  71.     WritableFactories.setFactory
  72.       (BlockCommand.class,
  73.        new WritableFactory() {
  74.          public Writable newInstance() { return new BlockCommand(); }
  75.        });
  76.   }
  77.   public void write(DataOutput out) throws IOException {
  78.     super.write(out);
  79.     out.writeInt(blocks.length);
  80.     for (int i = 0; i < blocks.length; i++) {
  81.       blocks[i].write(out);
  82.     }
  83.     out.writeInt(targets.length);
  84.     for (int i = 0; i < targets.length; i++) {
  85.       out.writeInt(targets[i].length);
  86.       for (int j = 0; j < targets[i].length; j++) {
  87.         targets[i][j].write(out);
  88.       }
  89.     }
  90.   }
  91.   public void readFields(DataInput in) throws IOException {
  92.     super.readFields(in);
  93.     this.blocks = new Block[in.readInt()];
  94.     for (int i = 0; i < blocks.length; i++) {
  95.       blocks[i] = new Block();
  96.       blocks[i].readFields(in);
  97.     }
  98.     this.targets = new DatanodeInfo[in.readInt()][];
  99.     for (int i = 0; i < targets.length; i++) {
  100.       this.targets[i] = new DatanodeInfo[in.readInt()];
  101.       for (int j = 0; j < targets[i].length; j++) {
  102.         targets[i][j] = new DatanodeInfo();
  103.         targets[i][j].readFields(in);
  104.       }
  105.     }
  106.   }
  107. }