浏览代码

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)
Jonathan Hung 5 年之前
父节点
当前提交
a0e592db5a

+ 6 - 2
hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/main/java/org/apache/hadoop/yarn/server/resourcemanager/RMContextImpl.java

@@ -587,9 +587,13 @@ public class RMContextImpl implements RMContext {
   }
 
   public Set<String> getExclusiveEnforcedPartitions() {
-    String[] configuredPartitions = getYarnConfiguration().getStrings(
-        YarnConfiguration.EXCLUSIVE_ENFORCED_PARTITIONS);
     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) {
       for (String partition : configuredPartitions) {
         exclusiveEnforcedPartitions.add(partition);