HDFS源码分析数据块复制选取复制源节点

数据块的复制当然需要一个源数据节点，从其上拷贝数据块至目标数据节点。那么数据块复制是如何选取复制源节点的呢？本文我们将针对这一问题进行研究。

        在BlockManager中，chooseSourceDatanode()方法就是用来选取数据块复制时的源节点的，它负责解析数据块所属数据节点列表，并选择一个，用它作为数据块的复制源。其核心逻辑如下：

        我们优先选择正处于退役过程中的数据节点而不是其他节点，因为前者没有写数据传输量因此相对不是很繁忙。我们不使用已退役节点作为数据源。否则我们从它们之中随机选择一个数据节点，其复制工作量还没有达到阈值，然而，如果一个复制是最高优先级的复制的话，我们会随机选择一个数据节点，而不管复制阈值的限制。

        chooseSourceDatanode()方法代码如下：

  @VisibleForTesting
  DatanodeDescriptor chooseSourceDatanode(Block block,
      List<DatanodeDescriptor> containingNodes,
      List<DatanodeStorageInfo>  nodesContainingLiveReplicas,
      NumberReplicas numReplicas,
      int priority) {

// 清空containingNodes列表
// 包含指定block的节点列表
   containingNodes.clear();

   // 清空nodesContainingLiveReplicas列表
   // 包含指定block活跃副本的节点列表
   nodesContainingLiveReplicas.clear();

   DatanodeDescriptor srcNode = null;
   int live = 0;
   int decommissioned = 0;
   int corrupt = 0;
   int excess = 0;

   // 根据Block实例block从corruptReplicas中获取坏块副本所在数据节点集合nodesCorrupt
   Collection<DatanodeDescriptor> nodesCorrupt = corruptReplicas.getNodes(block);

   // 根据Block实例block从blocksMap中获取其对应的数据节点存储DatanodeStorageInfo实例storage
   for(DatanodeStorageInfo storage : blocksMap.getStorages(block)) {

     // 从数据节点存储DatanodeStorageInfo实例storage中获取数据节点描述信息node
     final DatanodeDescriptor node = storage.getDatanodeDescriptor();

     // 从excessReplicateMap集合中获取数据块集合excessBlocks，
     // 这些块对数据节点来说是多余的。我们最终会将这些多余的块删除。
     LightWeightLinkedSet<Block> excessBlocks =
       excessReplicateMap.get(node.getDatanodeUuid());

     // 根据数据节点的存储状态确定其是否为可用副本
     int countableReplica = storage.getState() == State.NORMAL ? 1 : 0;

     // 如果坏块节点集合nodesCorrupt中包含该节点，坏块数corrupt累加
     if ((nodesCorrupt != null) && (nodesCorrupt.contains(node)))
       corrupt += countableReplica;

     // 如果节点正在退役或者已经退役，退役数decommissioned累加
     else if (node.isDecommissionInProgress() || node.isDecommissioned())
       decommissioned += countableReplica;

     // 如果多余数据块集合中包含该数据块，则多余数excess累加
     else if (excessBlocks != null && excessBlocks.contains(block)) {
       excess += countableReplica;

     // 其他情况下
     } else {

    // 将该存储添加到nodesContainingLiveReplicas集合
       nodesContainingLiveReplicas.add(storage);

       // 累加活跃副本数live
       live += countableReplica;
     }

     // 将该节点添加到containingNodes集合
     containingNodes.add(node);

     // Check if this replica is corrupt
     // If so, do not select the node as src node

     // 如果为坏块，跳过
     if ((nodesCorrupt != null) && nodesCorrupt.contains(node))
       continue;

     // 如果复制级别不是最高级别，且数据节点正在复制的数据块数目大于等于最大复制块数maxReplicationStreams，跳过
     if(priority != UnderReplicatedBlocks.QUEUE_HIGHEST_PRIORITY
         && node.getNumberOfBlocksToBeReplicated() >= maxReplicationStreams)
     {
       continue; // already reached replication limit
     }

     // 如果数据节点getNumberOfBlocksToBeReplicated大于等于复制块数上线replicationStreamsHardLimit，跳过
     if (node.getNumberOfBlocksToBeReplicated() >= replicationStreamsHardLimit)
     {
       continue;
     }

     // the block must not be scheduled for removal on srcNode
     // 如果数据块为多余的数据块，直接跳过
     if(excessBlocks != null && excessBlocks.contains(block))
       continue;

     // never use already decommissioned nodes
     // 如果数据节点为已退役节点，跳过
     if(node.isDecommissioned())
       continue;
     // we prefer nodes that are in DECOMMISSION_INPROGRESS state
     // 如果数据节点正在退役，且srcNode还未选中，那么选择该数据节点为srcNode，并跳过
     if(node.isDecommissionInProgress() || srcNode == null) {
       srcNode = node;
       continue;
     }

     // 如果源数据节点srcNode正在退役，则跳过
     if(srcNode.isDecommissionInProgress())
       continue;
     // switch to a different node randomly
     // this to prevent from deterministically selecting the same node even
     // if the node failed to replicate the block on previous iterations
     if(DFSUtil.getRandom().nextBoolean())
       srcNode = node;
   }

   // 初始化数据块副本复制统计对象numReplicas
   if(numReplicas != null)
     numReplicas.initialize(live, decommissioned, corrupt, excess, 0);

   // 返回srcNode
   return srcNode;
 }

        chooseSourceDatanode()方法的整体逻辑如下：

        1、清空containingNodes列表：containingNodes为包含指定block的节点描述信息DatanodeDescriptor列表；

        2、清空nodesContainingLiveReplicas列表：nodesContainingLiveReplicas为包含指定block活跃副本的节点存储DatanodeStorageInfo列表；

        3、根据Block实例block从corruptReplicas中获取坏块副本所在数据节点集合nodesCorrupt；

        4、根据Block实例block从blocksMap中获取其对应的数据节点存储DatanodeStorageInfo集合，并遍历每一个数据节点存储DatanodeStorageInfo实例storage：

              4.1、从数据节点存储DatanodeStorageInfo实例storage中获取数据节点描述信息node；

              4.2、从excessReplicateMap集合中获取数据块集合excessBlocks：这些块对数据节点来说是多余的，我们最终会将这些多余的块删除；

              4.3、根据数据节点的存储状态确定其是否为可用副本countableReplica；

              以下为统计数据块副本情况：

              4.4、如果坏块节点集合nodesCorrupt中包含该节点，坏块数corrupt累加；

              4.5、如果节点正在退役或者已经退役，退役数decommissioned累加；

              4.6、如果多余数据块集合中包含该数据块，则多余数excess累加；

              4.7、其他情况下：

                      4.7.1、将该存储添加到nodesContainingLiveReplicas集合；

                      4.7.2、累加活跃副本数live；

              4.8、将该节点添加到containingNodes集合；

              4.9、如果为坏块，跳过；

              4.10、如果复制级别不是最高级别，且节点正在复制的数据块数目大于等于最大复制块数maxReplicationStreams，跳过；

              4.11、如果数据节点getNumberOfBlocksToBeReplicated大于等于复制块数上线replicationStreamsHardLimit，跳过；

              4.12、如果数据块为多余的数据块，直接跳过；

              4.13、如果数据节点为已退役节点，跳过；

              4.14、如果数据节点正在退役，且srcNode还未选中，那么选择该数据节点为srcNode，并跳过；

              4.15、如果源数据节点srcNode正在退役，则跳过；

              4.16、随机选择源数据节点；

        5、初始化数据块副本复制统计对象numReplicas；

        6、返回块复制源数据节点srcNode。

        其中，有两个阈值需要单独说下，如下：

        1、maxReplicationStreams：一个给定节点除最高优先级复制外复制流的最大数目，取参数dfs.namenode.replication.max-streams，参数未配置默认为2；

        2、replicationStreamsHardLimit：一个给定节点全部优先级复制复制流的最大数目，取参数dfs.namenode.replication.max-streams-hard-limit，参数未配置默认为4。

        

        从上述整理流程中，大致总结如下：

        根据block从blocksMap中取数据块所在数据节点存储实例集合并遍历，统计数据块副本情况，包括损坏副本、多余副本、退役副本、活跃副本等，然后损坏副本、多余副本、退役节点直接跳过，这三种情况不能被选中为复制源数据节点，并且还有两种情况，一是如果复制级别不是最高级别，且数据节点正在复制的数据块数目大于等于最大复制块数maxReplicationStreams，二是如果数据节点正在复制的数据块数目大于等于复制块数上线replicationStreamsHardLimit，这两种情况也直接跳过，不能被选中为复制源数据节点，剩下的，则是随机选择源数据节点，并且其最喜欢选择正在退役的数据节点，这个最喜欢的意思是，选择的方式是随机选择，但是一旦正在退役节点被选中，则源节点不会再做变更，否则还是要通过随机选择来变更的。