浏览代码

HDFS-16377. Should CheckNotNull before access FsDatasetSpi (#3784)

Reviewed-by: Viraj Jasani <vjasani@apache.org>
Signed-off-by: Takanobu Asanuma <tasanuma@apache.org>
litao 3 年之前
父节点
当前提交
22f5e1885d

+ 7 - 1
hadoop-hdfs-project/hadoop-hdfs/src/main/java/org/apache/hadoop/hdfs/server/datanode/DataNode.java

@@ -865,6 +865,7 @@ public class DataNode extends ReconfigurableBase
               .newFixedThreadPool(changedVolumes.newLocations.size());
               .newFixedThreadPool(changedVolumes.newLocations.size());
           List<Future<IOException>> exceptions = Lists.newArrayList();
           List<Future<IOException>> exceptions = Lists.newArrayList();
 
 
+          Preconditions.checkNotNull(data, "Storage not yet initialized");
           for (final StorageLocation location : changedVolumes.newLocations) {
           for (final StorageLocation location : changedVolumes.newLocations) {
             exceptions.add(service.submit(new Callable<IOException>() {
             exceptions.add(service.submit(new Callable<IOException>() {
               @Override
               @Override
@@ -964,6 +965,7 @@ public class DataNode extends ReconfigurableBase
         clearFailure, Joiner.on(",").join(storageLocations)));
         clearFailure, Joiner.on(",").join(storageLocations)));
 
 
     IOException ioe = null;
     IOException ioe = null;
+    Preconditions.checkNotNull(data, "Storage not yet initialized");
     // Remove volumes and block infos from FsDataset.
     // Remove volumes and block infos from FsDataset.
     data.removeVolumes(storageLocations, clearFailure);
     data.removeVolumes(storageLocations, clearFailure);
 
 
@@ -2040,6 +2042,7 @@ public class DataNode extends ReconfigurableBase
     FileInputStream fis[] = new FileInputStream[2];
     FileInputStream fis[] = new FileInputStream[2];
     
     
     try {
     try {
+      Preconditions.checkNotNull(data, "Storage not yet initialized");
       fis[0] = (FileInputStream)data.getBlockInputStream(blk, 0);
       fis[0] = (FileInputStream)data.getBlockInputStream(blk, 0);
       fis[1] = DatanodeUtil.getMetaDataInputStream(blk, data);
       fis[1] = DatanodeUtil.getMetaDataInputStream(blk, data);
     } catch (ClassCastException e) {
     } catch (ClassCastException e) {
@@ -3069,6 +3072,7 @@ public class DataNode extends ReconfigurableBase
   @Override // InterDatanodeProtocol
   @Override // InterDatanodeProtocol
   public ReplicaRecoveryInfo initReplicaRecovery(RecoveringBlock rBlock)
   public ReplicaRecoveryInfo initReplicaRecovery(RecoveringBlock rBlock)
       throws IOException {
       throws IOException {
+    Preconditions.checkNotNull(data, "Storage not yet initialized");
     return data.initReplicaRecovery(rBlock);
     return data.initReplicaRecovery(rBlock);
   }
   }
 
 
@@ -3079,6 +3083,7 @@ public class DataNode extends ReconfigurableBase
   public String updateReplicaUnderRecovery(final ExtendedBlock oldBlock,
   public String updateReplicaUnderRecovery(final ExtendedBlock oldBlock,
       final long recoveryId, final long newBlockId, final long newLength)
       final long recoveryId, final long newBlockId, final long newLength)
       throws IOException {
       throws IOException {
+    Preconditions.checkNotNull(data, "Storage not yet initialized");
     final Replica r = data.updateReplicaUnderRecovery(oldBlock,
     final Replica r = data.updateReplicaUnderRecovery(oldBlock,
         recoveryId, newBlockId, newLength);
         recoveryId, newBlockId, newLength);
     // Notify the namenode of the updated block info. This is important
     // Notify the namenode of the updated block info. This is important
@@ -3360,7 +3365,7 @@ public class DataNode extends ReconfigurableBase
           "The block pool is still running. First do a refreshNamenodes to " +
           "The block pool is still running. First do a refreshNamenodes to " +
           "shutdown the block pool service");
           "shutdown the block pool service");
     }
     }
-   
+    Preconditions.checkNotNull(data, "Storage not yet initialized");
     data.deleteBlockPool(blockPoolId, force);
     data.deleteBlockPool(blockPoolId, force);
   }
   }
 
 
@@ -3804,6 +3809,7 @@ public class DataNode extends ReconfigurableBase
   @Override
   @Override
   public List<DatanodeVolumeInfo> getVolumeReport() throws IOException {
   public List<DatanodeVolumeInfo> getVolumeReport() throws IOException {
     checkSuperuserPrivilege();
     checkSuperuserPrivilege();
+    Preconditions.checkNotNull(data, "Storage not yet initialized");
     Map<String, Object> volumeInfoMap = data.getVolumeInfoMap();
     Map<String, Object> volumeInfoMap = data.getVolumeInfoMap();
     if (volumeInfoMap == null) {
     if (volumeInfoMap == null) {
       LOG.warn("DataNode volume info not available.");
       LOG.warn("DataNode volume info not available.");