Java DistributedFileSystem.listLocatedStatus方法代码示例

import org.apache.hadoop.hdfs.DistributedFileSystem; //导入方法依赖的package包/类
/**
 * Verify a DN remains in DECOMMISSION_INPROGRESS state if it is marked
 * as dead before decommission has completed. That will allow DN to resume
 * the replication process after it rejoins the cluster.
 */
@Test(timeout=120000)
public void testDecommissionStatusAfterDNRestart() throws Exception {
  DistributedFileSystem fileSys =
      (DistributedFileSystem)cluster.getFileSystem();

  // Create a file with one block. That block has one replica.
  Path f = new Path("decommission.dat");
  DFSTestUtil.createFile(fileSys, f, fileSize, fileSize, fileSize,
      (short)1, seed);

  // Find the DN that owns the only replica.
  RemoteIterator<LocatedFileStatus> fileList = fileSys.listLocatedStatus(f);
  BlockLocation[] blockLocations = fileList.next().getBlockLocations();
  String dnName = blockLocations[0].getNames()[0];

  // Decommission the DN.
  FSNamesystem fsn = cluster.getNamesystem();
  final DatanodeManager dm = fsn.getBlockManager().getDatanodeManager();
  decommissionNode(fsn, localFileSys, dnName);
  dm.refreshNodes(conf);

  // Stop the DN when decommission is in progress.
  // Given DFS_DATANODE_BALANCE_BANDWIDTHPERSEC_KEY is to 1 and the size of
  // the block, it will take much longer time that test timeout value for
  // the decommission to complete. So when stopDataNode is called,
  // decommission should be in progress.
  DataNodeProperties dataNodeProperties = cluster.stopDataNode(dnName);
  final List<DatanodeDescriptor> dead = new ArrayList<DatanodeDescriptor>();
  while (true) {
    dm.fetchDatanodes(null, dead, false);
    if (dead.size() == 1) {
      break;
    }
    Thread.sleep(1000);
  }

  // Force removal of the dead node's blocks.
  BlockManagerTestUtil.checkHeartbeat(fsn.getBlockManager());

  // Force DatanodeManager to check decommission state.
  BlockManagerTestUtil.recheckDecommissionState(dm);

  // Verify that the DN remains in DECOMMISSION_INPROGRESS state.
  assertTrue("the node should be DECOMMISSION_IN_PROGRESSS",
      dead.get(0).isDecommissionInProgress());

  // Check DatanodeManager#getDecommissionNodes, make sure it returns
  // the node as decommissioning, even if it's dead
  List<DatanodeDescriptor> decomlist = dm.getDecommissioningNodes();
  assertTrue("The node should be be decommissioning", decomlist.size() == 1);
  
  // Delete the under-replicated file, which should let the 
  // DECOMMISSION_IN_PROGRESS node become DECOMMISSIONED
  cleanupFile(fileSys, f);
  BlockManagerTestUtil.recheckDecommissionState(dm);
  assertTrue("the node should be decommissioned",
      dead.get(0).isDecommissioned());

  // Add the node back
  cluster.restartDataNode(dataNodeProperties, true);
  cluster.waitActive();

  // Call refreshNodes on FSNamesystem with empty exclude file.
  // This will remove the datanodes from decommissioning list and
  // make them available again.
  writeConfigFile(localFileSys, excludeFile, null);
  dm.refreshNodes(conf);
}