YARN-4611. Fix scheduler load simulator to support multi-layer network
location. Contributed by Ming Ma.
This commit is contained in:
parent
890a2ebd1a
commit
1708a4cd23
@ -32,6 +32,7 @@
|
||||
import org.apache.hadoop.classification.InterfaceStability.Unstable;
|
||||
import org.apache.hadoop.conf.Configuration;
|
||||
import org.apache.hadoop.fs.Path;
|
||||
import org.apache.hadoop.net.NodeBase;
|
||||
import org.apache.hadoop.tools.rumen.JobTraceReader;
|
||||
import org.apache.hadoop.tools.rumen.LoggedJob;
|
||||
import org.apache.hadoop.tools.rumen.LoggedTask;
|
||||
@ -43,9 +44,15 @@
|
||||
@Unstable
|
||||
public class SLSUtils {
|
||||
|
||||
// hostname includes the network path and the host name. for example
|
||||
// "/default-rack/hostFoo" or "/coreSwitchA/TORSwitchB/hostBar".
|
||||
// the function returns two Strings, the first element is the network
|
||||
// location without "/", the second element is the host name. for example,
|
||||
// {"default-rack", "hostFoo"} or "coreSwitchA/TORSwitchB", "hostBar"
|
||||
public static String[] getRackHostName(String hostname) {
|
||||
hostname = hostname.substring(1);
|
||||
return hostname.split("/");
|
||||
NodeBase node = new NodeBase(hostname);
|
||||
return new String[] {node.getNetworkLocation().substring(1),
|
||||
node.getName()};
|
||||
}
|
||||
|
||||
/**
|
||||
|
@ -51,7 +51,7 @@ public void setup() {
|
||||
public void testNMSimulator() throws Exception {
|
||||
// Register one node
|
||||
NMSimulator node1 = new NMSimulator();
|
||||
node1.init("rack1/node1", GB * 10, 10, 0, 1000, rm);
|
||||
node1.init("/rack1/node1", GB * 10, 10, 0, 1000, rm);
|
||||
node1.middleStep();
|
||||
|
||||
int numClusterNodes = rm.getResourceScheduler().getNumClusterNodes();
|
||||
|
@ -29,6 +29,11 @@ public void testGetRackHostname() {
|
||||
String rackHostname[] = SLSUtils.getRackHostName(str);
|
||||
Assert.assertEquals(rackHostname[0], "rack1");
|
||||
Assert.assertEquals(rackHostname[1], "node1");
|
||||
|
||||
str = "/rackA/rackB/node1";
|
||||
rackHostname = SLSUtils.getRackHostName(str);
|
||||
Assert.assertEquals(rackHostname[0], "rackA/rackB");
|
||||
Assert.assertEquals(rackHostname[1], "node1");
|
||||
}
|
||||
|
||||
}
|
||||
|
@ -131,6 +131,9 @@ Release 2.9.0 - UNRELEASED
|
||||
YARN-3446. FairScheduler headroom calculation should exclude nodes in the
|
||||
blacklist. (Zhihai Xu via kasha)
|
||||
|
||||
YARN-4611. Fix scheduler load simulator to support multi-layer network
|
||||
location. (Ming Ma via xgong)
|
||||
|
||||
Release 2.8.0 - UNRELEASED
|
||||
|
||||
INCOMPATIBLE CHANGES
|
||||
|
Loading…
Reference in New Issue
Block a user