diff --git a/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/main/java/org/apache/hadoop/yarn/server/resourcemanager/DefaultAMSProcessor.java b/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/main/java/org/apache/hadoop/yarn/server/resourcemanager/DefaultAMSProcessor.java index 9e05edae440..a3bdb2f482a 100644 --- a/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/main/java/org/apache/hadoop/yarn/server/resourcemanager/DefaultAMSProcessor.java +++ b/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/main/java/org/apache/hadoop/yarn/server/resourcemanager/DefaultAMSProcessor.java @@ -116,6 +116,7 @@ final class DefaultAMSProcessor implements ApplicationMasterServiceProcessor { private RMContext rmContext; private ResourceProfilesManager resourceProfilesManager; private boolean timelineServiceV2Enabled; + private boolean nodelabelsEnabled; @Override public void init(ApplicationMasterServiceContext amsContext, @@ -124,6 +125,8 @@ final class DefaultAMSProcessor implements ApplicationMasterServiceProcessor { this.resourceProfilesManager = rmContext.getResourceProfilesManager(); this.timelineServiceV2Enabled = YarnConfiguration. timelineServiceV2Enabled(rmContext.getYarnConfiguration()); + this.nodelabelsEnabled = YarnConfiguration + .areNodeLabelsEnabled(rmContext.getYarnConfiguration()); } @Override @@ -241,7 +244,7 @@ final class DefaultAMSProcessor implements ApplicationMasterServiceProcessor { try { RMServerUtils.normalizeAndValidateRequests(ask, maximumCapacity, app.getQueue(), - getScheduler(), getRmContext()); + getScheduler(), getRmContext(), nodelabelsEnabled); } catch (InvalidResourceRequestException e) { RMAppAttempt rmAppAttempt = app.getRMAppAttempt(appAttemptId); handleInvalidResourceException(e, rmAppAttempt); diff --git a/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/main/java/org/apache/hadoop/yarn/server/resourcemanager/RMAppManager.java b/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/main/java/org/apache/hadoop/yarn/server/resourcemanager/RMAppManager.java index 0679b0eec61..feaff00d2cd 100644 --- a/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/main/java/org/apache/hadoop/yarn/server/resourcemanager/RMAppManager.java +++ b/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/main/java/org/apache/hadoop/yarn/server/resourcemanager/RMAppManager.java @@ -96,6 +96,7 @@ public class RMAppManager implements EventHandler, private Configuration conf; private YarnAuthorizationProvider authorizer; private boolean timelineServiceV2Enabled; + private boolean nodeLabelsEnabled; public RMAppManager(RMContext context, YarnScheduler scheduler, ApplicationMasterService masterService, @@ -118,6 +119,8 @@ public class RMAppManager implements EventHandler, this.authorizer = YarnAuthorizationProvider.getInstance(conf); this.timelineServiceV2Enabled = YarnConfiguration. timelineServiceV2Enabled(conf); + this.nodeLabelsEnabled = YarnConfiguration + .areNodeLabelsEnabled(rmContext.getYarnConfiguration()); } /** @@ -575,7 +578,7 @@ public class RMAppManager implements EventHandler, Resource maxAllocation = scheduler.getMaximumResourceCapability(queue); for (ResourceRequest amReq : amReqs) { SchedulerUtils.normalizeAndValidateRequest(amReq, maxAllocation, - queue, scheduler, isRecovery, rmContext, null); + queue, isRecovery, rmContext, null, nodeLabelsEnabled); amReq.setCapability(scheduler.getNormalizedResource( amReq.getCapability(), maxAllocation)); diff --git a/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/main/java/org/apache/hadoop/yarn/server/resourcemanager/RMServerUtils.java b/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/main/java/org/apache/hadoop/yarn/server/resourcemanager/RMServerUtils.java index bad890fbfdb..25eb979e0e4 100644 --- a/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/main/java/org/apache/hadoop/yarn/server/resourcemanager/RMServerUtils.java +++ b/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/main/java/org/apache/hadoop/yarn/server/resourcemanager/RMServerUtils.java @@ -244,7 +244,8 @@ public class RMServerUtils { */ public static void normalizeAndValidateRequests(List ask, Resource maximumAllocation, String queueName, YarnScheduler scheduler, - RMContext rmContext) throws InvalidResourceRequestException { + RMContext rmContext, boolean nodeLabelsEnabled) + throws InvalidResourceRequestException { // Get queue from scheduler QueueInfo queueInfo = null; try { @@ -256,7 +257,7 @@ public class RMServerUtils { for (ResourceRequest resReq : ask) { SchedulerUtils.normalizeAndValidateRequest(resReq, maximumAllocation, - queueName, scheduler, rmContext, queueInfo); + queueName, rmContext, queueInfo, nodeLabelsEnabled); } } diff --git a/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/main/java/org/apache/hadoop/yarn/server/resourcemanager/scheduler/SchedulerUtils.java b/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/main/java/org/apache/hadoop/yarn/server/resourcemanager/scheduler/SchedulerUtils.java index a048dacb688..36746c02809 100644 --- a/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/main/java/org/apache/hadoop/yarn/server/resourcemanager/scheduler/SchedulerUtils.java +++ b/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/main/java/org/apache/hadoop/yarn/server/resourcemanager/scheduler/SchedulerUtils.java @@ -42,7 +42,6 @@ import org.apache.hadoop.yarn.api.records.QueueInfo; import org.apache.hadoop.yarn.api.records.Resource; import org.apache.hadoop.yarn.api.records.ResourceInformation; import org.apache.hadoop.yarn.api.records.ResourceRequest; -import org.apache.hadoop.yarn.conf.YarnConfiguration; import org.apache.hadoop.yarn.exceptions.InvalidLabelResourceRequestException; import org.apache.hadoop.yarn.exceptions.InvalidResourceRequestException; import org.apache.hadoop.yarn.exceptions.InvalidResourceRequestException @@ -260,12 +259,12 @@ public class SchedulerUtils { } public static void normalizeAndValidateRequest(ResourceRequest resReq, - Resource maximumAllocation, String queueName, YarnScheduler scheduler, - boolean isRecovery, RMContext rmContext, QueueInfo queueInfo) - throws InvalidResourceRequestException { + Resource maximumAllocation, String queueName, boolean isRecovery, + RMContext rmContext, QueueInfo queueInfo, boolean nodeLabelsEnabled) + throws InvalidResourceRequestException { Configuration conf = rmContext.getYarnConfiguration(); // If Node label is not enabled throw exception - if (null != conf && !YarnConfiguration.areNodeLabelsEnabled(conf)) { + if (null != conf && !nodeLabelsEnabled) { String labelExp = resReq.getNodeLabelExpression(); if (!(RMNodeLabelsManager.NO_LABEL.equals(labelExp) || null == labelExp)) { @@ -281,7 +280,8 @@ public class SchedulerUtils { } if (null == queueInfo) { try { - queueInfo = scheduler.getQueueInfo(queueName, false, false); + queueInfo = rmContext.getScheduler().getQueueInfo(queueName, false, + false); } catch (IOException e) { //Queue may not exist since it could be auto-created in case of // dynamic queues @@ -295,11 +295,11 @@ public class SchedulerUtils { } public static void normalizeAndValidateRequest(ResourceRequest resReq, - Resource maximumAllocation, String queueName, YarnScheduler scheduler, - RMContext rmContext, QueueInfo queueInfo) - throws InvalidResourceRequestException { - normalizeAndValidateRequest(resReq, maximumAllocation, queueName, scheduler, - false, rmContext, queueInfo); + Resource maximumAllocation, String queueName, RMContext rmContext, + QueueInfo queueInfo, boolean nodeLabelsEnabled) + throws InvalidResourceRequestException { + normalizeAndValidateRequest(resReq, maximumAllocation, queueName, false, + rmContext, queueInfo, nodeLabelsEnabled); } /** diff --git a/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/test/java/org/apache/hadoop/yarn/server/resourcemanager/TestAppManager.java b/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/test/java/org/apache/hadoop/yarn/server/resourcemanager/TestAppManager.java index 3c39f6494ca..31c16a43def 100644 --- a/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/test/java/org/apache/hadoop/yarn/server/resourcemanager/TestAppManager.java +++ b/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/test/java/org/apache/hadoop/yarn/server/resourcemanager/TestAppManager.java @@ -203,6 +203,7 @@ public class TestAppManager extends AppManagerTestBase{ metricsPublisher = mock(SystemMetricsPublisher.class); context.setSystemMetricsPublisher(metricsPublisher); context.setRMApplicationHistoryWriter(writer); + ((RMContextImpl)context).setYarnConfiguration(new YarnConfiguration()); return context; } diff --git a/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/test/java/org/apache/hadoop/yarn/server/resourcemanager/scheduler/TestSchedulerUtils.java b/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/test/java/org/apache/hadoop/yarn/server/resourcemanager/scheduler/TestSchedulerUtils.java index 696924203de..73027058e0b 100644 --- a/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/test/java/org/apache/hadoop/yarn/server/resourcemanager/scheduler/TestSchedulerUtils.java +++ b/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/test/java/org/apache/hadoop/yarn/server/resourcemanager/scheduler/TestSchedulerUtils.java @@ -1050,7 +1050,8 @@ public class TestSchedulerUtils { Resource maxAllocation) throws InvalidResourceRequestException { SchedulerUtils.normalizeAndValidateRequest(resReq, maxAllocation, queueName, - scheduler, rmContext, null); + rmContext, null, YarnConfiguration + .areNodeLabelsEnabled(rmContext.getYarnConfiguration())); } private static class InvalidResourceRequestExceptionMessageGenerator {