HDFS-13408. MiniDFSCluster to support being built on randomized base directory. Contributed by Xiao Liang

(cherry picked from commit f411de6a79)
This commit is contained in:
Chris Douglas 2018-04-23 11:13:18 -07:00
parent 1f486a0649
commit cf272c5179
2 changed files with 28 additions and 10 deletions

View File

@ -202,6 +202,26 @@ public class MiniDFSCluster implements AutoCloseable {
this.conf = conf;
this.storagesPerDatanode =
FsDatasetTestUtils.Factory.getFactory(conf).getDefaultNumOfDataDirs();
if (null == conf.get(HDFS_MINIDFS_BASEDIR)) {
conf.set(HDFS_MINIDFS_BASEDIR,
new File(getBaseDirectory()).getAbsolutePath());
}
}
public Builder(Configuration conf, File basedir) {
this.conf = conf;
this.storagesPerDatanode =
FsDatasetTestUtils.Factory.getFactory(conf).getDefaultNumOfDataDirs();
if (null == basedir) {
throw new IllegalArgumentException(
"MiniDFSCluster base directory cannot be null");
}
String cdir = conf.get(HDFS_MINIDFS_BASEDIR);
if (cdir != null) {
throw new IllegalArgumentException(
"MiniDFSCluster base directory already defined (" + cdir + ")");
}
conf.set(HDFS_MINIDFS_BASEDIR, basedir.getAbsolutePath());
}
/**

View File

@ -201,9 +201,8 @@ public class TestMiniDFSCluster {
public void testIsClusterUpAfterShutdown() throws Throwable {
Configuration conf = new HdfsConfiguration();
File testDataCluster4 = new File(testDataPath, CLUSTER_4);
String c4Path = testDataCluster4.getAbsolutePath();
conf.set(MiniDFSCluster.HDFS_MINIDFS_BASEDIR, c4Path);
MiniDFSCluster cluster4 = new MiniDFSCluster.Builder(conf).build();
MiniDFSCluster cluster4 =
new MiniDFSCluster.Builder(conf, testDataCluster4).build();
try {
DistributedFileSystem dfs = cluster4.getFileSystem();
dfs.setSafeMode(HdfsConstants.SafeModeAction.SAFEMODE_ENTER);
@ -222,9 +221,8 @@ public class TestMiniDFSCluster {
Configuration conf = new HdfsConfiguration();
conf.set(DFSConfigKeys.DFS_DATANODE_HOST_NAME_KEY, "MYHOST");
File testDataCluster5 = new File(testDataPath, CLUSTER_5);
String c5Path = testDataCluster5.getAbsolutePath();
conf.set(MiniDFSCluster.HDFS_MINIDFS_BASEDIR, c5Path);
try (MiniDFSCluster cluster5 = new MiniDFSCluster.Builder(conf)
try (MiniDFSCluster cluster5 =
new MiniDFSCluster.Builder(conf, testDataCluster5)
.numDataNodes(1)
.checkDataNodeHostConfig(true)
.build()) {