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:
parent
806c7b7dfb
commit
f4f210d2e5
|
@ -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);
|
||||||
|
|
Loading…
Reference in New Issue