YARN-7143. FileNotFound handling in ResourceUtils is inconsistent
Change-Id: Ib1bb487e14a15edd2b5a42cf5078c5a2b295f069 (cherry picked from commit db82a41d94872cea4d0c1bb1336916cebc2faeec)
This commit is contained in:
parent
7025333d31
commit
062fbd084d
|
@ -338,18 +338,14 @@ public class ResourceUtils {
|
|||
if (!initializedResources) {
|
||||
synchronized (ResourceUtils.class) {
|
||||
if (!initializedResources) {
|
||||
if (conf == null) {
|
||||
conf = new YarnConfiguration();
|
||||
}
|
||||
try {
|
||||
addResourcesFileToConf(resourceFile, conf);
|
||||
LOG.debug("Found " + resourceFile + ", adding to configuration");
|
||||
} catch (FileNotFoundException fe) {
|
||||
LOG.info("Unable to find '" + resourceFile
|
||||
+ "'. Falling back to memory and vcores as resources.");
|
||||
}
|
||||
initializeResourcesMap(conf);
|
||||
Configuration resConf = conf;
|
||||
|
||||
if (resConf == null) {
|
||||
resConf = new YarnConfiguration();
|
||||
}
|
||||
|
||||
addResourcesFileToConf(resourceFile, resConf);
|
||||
initializeResourcesMap(resConf);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
@ -386,21 +382,17 @@ public class ResourceUtils {
|
|||
}
|
||||
|
||||
private static void addResourcesFileToConf(String resourceFile,
|
||||
Configuration conf) throws FileNotFoundException {
|
||||
Configuration conf) {
|
||||
try {
|
||||
InputStream ris = getConfInputStream(resourceFile, conf);
|
||||
LOG.debug("Found " + resourceFile + ", adding to configuration");
|
||||
conf.addResource(ris);
|
||||
} catch (FileNotFoundException fe) {
|
||||
throw fe;
|
||||
} catch (IOException ie) {
|
||||
LOG.info("Unable to find '" + resourceFile + "'.");
|
||||
} catch (IOException | YarnException ex) {
|
||||
LOG.fatal("Exception trying to read resource types configuration '"
|
||||
+ resourceFile + "'.", ie);
|
||||
throw new YarnRuntimeException(ie);
|
||||
} catch (YarnException ye) {
|
||||
LOG.fatal("YARN Exception trying to read resource types configuration '"
|
||||
+ resourceFile + "'.", ye);
|
||||
throw new YarnRuntimeException(ye);
|
||||
+ resourceFile + "'.", ex);
|
||||
throw new YarnRuntimeException(ex);
|
||||
}
|
||||
}
|
||||
|
||||
|
@ -462,19 +454,19 @@ public class ResourceUtils {
|
|||
private static Map<String, ResourceInformation> initializeNodeResourceInformation(
|
||||
Configuration conf) {
|
||||
Map<String, ResourceInformation> nodeResources = new HashMap<>();
|
||||
try {
|
||||
addResourcesFileToConf(
|
||||
YarnConfiguration.NODE_RESOURCES_CONFIGURATION_FILE, conf);
|
||||
for (Map.Entry<String, String> entry : conf) {
|
||||
String key = entry.getKey();
|
||||
String value = entry.getValue();
|
||||
if (key.startsWith(YarnConfiguration.NM_RESOURCES_PREFIX)) {
|
||||
addResourceInformation(key, value, nodeResources);
|
||||
}
|
||||
|
||||
addResourcesFileToConf(YarnConfiguration.NODE_RESOURCES_CONFIGURATION_FILE,
|
||||
conf);
|
||||
|
||||
for (Map.Entry<String, String> entry : conf) {
|
||||
String key = entry.getKey();
|
||||
String value = entry.getValue();
|
||||
|
||||
if (key.startsWith(YarnConfiguration.NM_RESOURCES_PREFIX)) {
|
||||
addResourceInformation(key, value, nodeResources);
|
||||
}
|
||||
} catch (FileNotFoundException fe) {
|
||||
LOG.info("Couldn't find node resources file");
|
||||
}
|
||||
|
||||
return nodeResources;
|
||||
}
|
||||
|
||||
|
|
Loading…
Reference in New Issue