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:
parent
1f486a0649
commit
cf272c5179
|
@ -202,6 +202,26 @@ public class MiniDFSCluster implements AutoCloseable {
|
||||||
this.conf = conf;
|
this.conf = conf;
|
||||||
this.storagesPerDatanode =
|
this.storagesPerDatanode =
|
||||||
FsDatasetTestUtils.Factory.getFactory(conf).getDefaultNumOfDataDirs();
|
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());
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
|
|
|
@ -201,9 +201,8 @@ public class TestMiniDFSCluster {
|
||||||
public void testIsClusterUpAfterShutdown() throws Throwable {
|
public void testIsClusterUpAfterShutdown() throws Throwable {
|
||||||
Configuration conf = new HdfsConfiguration();
|
Configuration conf = new HdfsConfiguration();
|
||||||
File testDataCluster4 = new File(testDataPath, CLUSTER_4);
|
File testDataCluster4 = new File(testDataPath, CLUSTER_4);
|
||||||
String c4Path = testDataCluster4.getAbsolutePath();
|
MiniDFSCluster cluster4 =
|
||||||
conf.set(MiniDFSCluster.HDFS_MINIDFS_BASEDIR, c4Path);
|
new MiniDFSCluster.Builder(conf, testDataCluster4).build();
|
||||||
MiniDFSCluster cluster4 = new MiniDFSCluster.Builder(conf).build();
|
|
||||||
try {
|
try {
|
||||||
DistributedFileSystem dfs = cluster4.getFileSystem();
|
DistributedFileSystem dfs = cluster4.getFileSystem();
|
||||||
dfs.setSafeMode(HdfsConstants.SafeModeAction.SAFEMODE_ENTER);
|
dfs.setSafeMode(HdfsConstants.SafeModeAction.SAFEMODE_ENTER);
|
||||||
|
@ -222,9 +221,8 @@ public class TestMiniDFSCluster {
|
||||||
Configuration conf = new HdfsConfiguration();
|
Configuration conf = new HdfsConfiguration();
|
||||||
conf.set(DFSConfigKeys.DFS_DATANODE_HOST_NAME_KEY, "MYHOST");
|
conf.set(DFSConfigKeys.DFS_DATANODE_HOST_NAME_KEY, "MYHOST");
|
||||||
File testDataCluster5 = new File(testDataPath, CLUSTER_5);
|
File testDataCluster5 = new File(testDataPath, CLUSTER_5);
|
||||||
String c5Path = testDataCluster5.getAbsolutePath();
|
try (MiniDFSCluster cluster5 =
|
||||||
conf.set(MiniDFSCluster.HDFS_MINIDFS_BASEDIR, c5Path);
|
new MiniDFSCluster.Builder(conf, testDataCluster5)
|
||||||
try (MiniDFSCluster cluster5 = new MiniDFSCluster.Builder(conf)
|
|
||||||
.numDataNodes(1)
|
.numDataNodes(1)
|
||||||
.checkDataNodeHostConfig(true)
|
.checkDataNodeHostConfig(true)
|
||||||
.build()) {
|
.build()) {
|
||||||
|
|
Loading…
Reference in New Issue