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

(cherry picked from commit d86a1acc866cbda845fb3896dc824baf12217383)
This commit is contained in:
Jonathan Hung 2019-09-25 17:48:40 -07:00
parent 806c7b7dfb
commit f4f210d2e5
1 changed files with 6 additions and 2 deletions

View File

@ -648,9 +648,13 @@ public class RMContextImpl implements RMContext {
} }
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);