Addendum to YARN-9730. Support forcing configured partitions to be exclusive based on app node label

(cherry picked from commit d86a1acc866cbda845fb3896dc824baf12217383)
(cherry picked from commit f4f210d2e57e49999579efb7db039cc623798267)
(cherry picked from commit 95ec7050b55490454e1de894da572a7e25374907)
This commit is contained in:
Jonathan Hung 2019-09-25 17:48:40 -07:00
parent eedbf9d195
commit a0e592db5a

View File

@ -587,9 +587,13 @@ public String getAppProxyUrl(Configuration conf, ApplicationId applicationId)
} }
public Set<String> getExclusiveEnforcedPartitions() { public Set<String> getExclusiveEnforcedPartitions() {
String[] configuredPartitions = getYarnConfiguration().getStrings(
YarnConfiguration.EXCLUSIVE_ENFORCED_PARTITIONS);
Set<String> exclusiveEnforcedPartitions = new HashSet<>(); Set<String> exclusiveEnforcedPartitions = new HashSet<>();
Configuration conf = getYarnConfiguration();
if (conf == null) {
return new HashSet<>();
}
String[] configuredPartitions = conf.getStrings(
YarnConfiguration.EXCLUSIVE_ENFORCED_PARTITIONS);
if (configuredPartitions != null) { if (configuredPartitions != null) {
for (String partition : configuredPartitions) { for (String partition : configuredPartitions) {
exclusiveEnforcedPartitions.add(partition); exclusiveEnforcedPartitions.add(partition);