diff --git a/hadoop-hdfs-project/hadoop-hdfs/src/main/java/org/apache/hadoop/hdfs/server/datanode/DataNode.java b/hadoop-hdfs-project/hadoop-hdfs/src/main/java/org/apache/hadoop/hdfs/server/datanode/DataNode.java index e7b72f777b9..f9298f1a268 100644 --- a/hadoop-hdfs-project/hadoop-hdfs/src/main/java/org/apache/hadoop/hdfs/server/datanode/DataNode.java +++ b/hadoop-hdfs-project/hadoop-hdfs/src/main/java/org/apache/hadoop/hdfs/server/datanode/DataNode.java @@ -1139,8 +1139,25 @@ public class DataNode extends ReconfigurableBase * Report a bad block which is hosted on the local DN. */ public void reportBadBlocks(ExtendedBlock block) throws IOException{ - BPOfferService bpos = getBPOSForBlock(block); FsVolumeSpi volume = getFSDataset().getVolume(block); + if (volume == null) { + LOG.warn("Cannot find FsVolumeSpi to report bad block: " + block); + return; + } + reportBadBlocks(block, volume); + } + + /** + * Report a bad block which is hosted on the local DN. + * + * @param block the bad block which is hosted on the local DN + * @param volume the volume that block is stored in and the volume + * must not be null + * @throws IOException + */ + public void reportBadBlocks(ExtendedBlock block, FsVolumeSpi volume) + throws IOException { + BPOfferService bpos = getBPOSForBlock(block); bpos.reportBadBlocks( block, volume.getStorageID(), volume.getStorageType()); } @@ -2056,6 +2073,10 @@ public class DataNode extends ReconfigurableBase private void reportBadBlock(final BPOfferService bpos, final ExtendedBlock block, final String msg) { FsVolumeSpi volume = getFSDataset().getVolume(block); + if (volume == null) { + LOG.warn("Cannot find FsVolumeSpi to report bad block: " + block); + return; + } bpos.reportBadBlocks( block, volume.getStorageID(), volume.getStorageType()); LOG.warn(msg); diff --git a/hadoop-hdfs-project/hadoop-hdfs/src/main/java/org/apache/hadoop/hdfs/server/datanode/VolumeScanner.java b/hadoop-hdfs-project/hadoop-hdfs/src/main/java/org/apache/hadoop/hdfs/server/datanode/VolumeScanner.java index d0dc9edbc82..7a9ecf2aaa7 100644 --- a/hadoop-hdfs-project/hadoop-hdfs/src/main/java/org/apache/hadoop/hdfs/server/datanode/VolumeScanner.java +++ b/hadoop-hdfs-project/hadoop-hdfs/src/main/java/org/apache/hadoop/hdfs/server/datanode/VolumeScanner.java @@ -283,7 +283,7 @@ public class VolumeScanner extends Thread { } LOG.warn("Reporting bad {} on {}", block, volume.getBasePath()); try { - scanner.datanode.reportBadBlocks(block); + scanner.datanode.reportBadBlocks(block, volume); } catch (IOException ie) { // This is bad, but not bad enough to shut down the scanner. LOG.warn("Cannot report bad " + block.getBlockId(), e); diff --git a/hadoop-hdfs-project/hadoop-hdfs/src/main/java/org/apache/hadoop/hdfs/server/datanode/fsdataset/impl/FsDatasetImpl.java b/hadoop-hdfs-project/hadoop-hdfs/src/main/java/org/apache/hadoop/hdfs/server/datanode/fsdataset/impl/FsDatasetImpl.java index b034c0aa373..3dae46e6e62 100644 --- a/hadoop-hdfs-project/hadoop-hdfs/src/main/java/org/apache/hadoop/hdfs/server/datanode/fsdataset/impl/FsDatasetImpl.java +++ b/hadoop-hdfs-project/hadoop-hdfs/src/main/java/org/apache/hadoop/hdfs/server/datanode/fsdataset/impl/FsDatasetImpl.java @@ -2373,7 +2373,8 @@ class FsDatasetImpl implements FsDatasetSpi { LOG.warn("Reporting the block " + corruptBlock + " as corrupt due to length mismatch"); try { - datanode.reportBadBlocks(new ExtendedBlock(bpid, corruptBlock)); + datanode.reportBadBlocks(new ExtendedBlock(bpid, corruptBlock), + memBlockInfo.getVolume()); } catch (IOException e) { LOG.warn("Failed to repot bad block " + corruptBlock, e); } diff --git a/hadoop-hdfs-project/hadoop-hdfs/src/test/java/org/apache/hadoop/hdfs/server/datanode/fsdataset/impl/TestFsDatasetImpl.java b/hadoop-hdfs-project/hadoop-hdfs/src/test/java/org/apache/hadoop/hdfs/server/datanode/fsdataset/impl/TestFsDatasetImpl.java index 68f76dcfee2..e73a6127df4 100644 --- a/hadoop-hdfs-project/hadoop-hdfs/src/test/java/org/apache/hadoop/hdfs/server/datanode/fsdataset/impl/TestFsDatasetImpl.java +++ b/hadoop-hdfs-project/hadoop-hdfs/src/test/java/org/apache/hadoop/hdfs/server/datanode/fsdataset/impl/TestFsDatasetImpl.java @@ -35,6 +35,7 @@ import org.apache.hadoop.hdfs.protocol.Block; import org.apache.hadoop.hdfs.protocol.DatanodeInfo; import org.apache.hadoop.hdfs.protocol.ExtendedBlock; import org.apache.hadoop.hdfs.protocol.LocatedBlock; +import org.apache.hadoop.hdfs.server.blockmanagement.BlockManagerTestUtil; import org.apache.hadoop.hdfs.server.common.HdfsServerConstants; import org.apache.hadoop.hdfs.server.common.Storage; import org.apache.hadoop.hdfs.server.common.StorageInfo; @@ -98,6 +99,7 @@ import static org.mockito.Mockito.never; import static org.mockito.Mockito.spy; import static org.mockito.Mockito.verify; import static org.mockito.Mockito.when; + import org.slf4j.Logger; import org.slf4j.LoggerFactory; @@ -691,4 +693,45 @@ public class TestFsDatasetImpl { cluster.shutdown(); } } + + @Test(timeout = 30000) + public void testReportBadBlocks() throws Exception { + boolean threwException = false; + MiniDFSCluster cluster = null; + try { + Configuration config = new HdfsConfiguration(); + cluster = new MiniDFSCluster.Builder(config).numDataNodes(1).build(); + cluster.waitActive(); + + Assert.assertEquals(0, cluster.getNamesystem().getCorruptReplicaBlocks()); + DataNode dataNode = cluster.getDataNodes().get(0); + ExtendedBlock block = + new ExtendedBlock(cluster.getNamesystem().getBlockPoolId(), 0); + try { + // Test the reportBadBlocks when the volume is null + dataNode.reportBadBlocks(block); + } catch (NullPointerException npe) { + threwException = true; + } + Thread.sleep(3000); + Assert.assertFalse(threwException); + Assert.assertEquals(0, cluster.getNamesystem().getCorruptReplicaBlocks()); + + FileSystem fs = cluster.getFileSystem(); + Path filePath = new Path("testData"); + DFSTestUtil.createFile(fs, filePath, 1, (short) 1, 0); + + block = DFSTestUtil.getFirstBlock(fs, filePath); + // Test for the overloaded method reportBadBlocks + dataNode.reportBadBlocks(block, dataNode.getFSDataset() + .getFsVolumeReferences().get(0)); + Thread.sleep(3000); + BlockManagerTestUtil.updateState(cluster.getNamesystem() + .getBlockManager()); + // Verify the bad block has been reported to namenode + Assert.assertEquals(1, cluster.getNamesystem().getCorruptReplicaBlocks()); + } finally { + cluster.shutdown(); + } + } }