This is an automated email from the ASF dual-hosted git repository.
snemeth pushed a commit to branch trunk
in repository https://gitbox.apache.org/repos/asf/hadoop.git
The following commit(s) were added to refs/heads/trunk by this push:
new 5440c75c4ab YARN-10946. AbstractCSQueue: Create separate class for
constructing Queue API objects. Contributed by Peter Szucs
5440c75c4ab is described below
commit 5440c75c4abee87aeff6da3c7e3e586978cd27dc
Author: Szilard Nemeth <snem...@apache.org>
AuthorDate: Thu Dec 1 15:11:58 2022 +0100
YARN-10946. AbstractCSQueue: Create separate class for constructing Queue
API objects. Contributed by Peter Szucs
---
.../scheduler/capacity/AbstractCSQueue.java | 80 +-------------
.../scheduler/capacity/CSQueueInfoProvider.java | 117 +++++++++++++++++++++
2 files changed, 118 insertions(+), 79 deletions(-)
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/capacity/AbstractCSQueue.java
b/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/main/java/org/apache/hadoop/yarn/server/resourcemanager/scheduler/capacity/AbstractCSQueue.java
index 1a5a1ce0fd4..7a3ec4f6f4e 100644
---
a/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/main/java/org/apache/hadoop/yarn/server/resourcemanager/scheduler/capacity/AbstractCSQueue.java
+++
b/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/main/java/org/apache/hadoop/yarn/server/resourcemanager/scheduler/capacity/AbstractCSQueue.java
@@ -29,10 +29,8 @@ import org.apache.hadoop.util.Time;
import org.apache.hadoop.yarn.api.records.ApplicationId;
import org.apache.hadoop.yarn.api.records.Priority;
import org.apache.hadoop.yarn.api.records.QueueACL;
-import org.apache.hadoop.yarn.api.records.QueueConfigurations;
import org.apache.hadoop.yarn.api.records.QueueInfo;
import org.apache.hadoop.yarn.api.records.QueueState;
-import org.apache.hadoop.yarn.api.records.QueueStatistics;
import org.apache.hadoop.yarn.api.records.Resource;
import org.apache.hadoop.yarn.api.records.ResourceInformation;
import org.apache.hadoop.yarn.exceptions.YarnException;
@@ -584,83 +582,7 @@ public abstract class AbstractCSQueue implements CSQueue {
// from schedulerApplicationAttempt, to avoid deadlock, sacrifice
// consistency here.
// TODO, improve this
- QueueInfo queueInfo = recordFactory.newRecordInstance(QueueInfo.class);
- queueInfo.setQueueName(queuePath.getLeafName());
- queueInfo.setQueuePath(queuePath.getFullPath());
-
queueInfo.setAccessibleNodeLabels(queueNodeLabelsSettings.getAccessibleNodeLabels());
- queueInfo.setCapacity(queueCapacities.getCapacity());
- queueInfo.setMaximumCapacity(queueCapacities.getMaximumCapacity());
- queueInfo.setQueueState(getState());
-
queueInfo.setDefaultNodeLabelExpression(queueNodeLabelsSettings.getDefaultLabelExpression());
- queueInfo.setCurrentCapacity(getUsedCapacity());
- queueInfo.setQueueStatistics(getQueueStatistics());
- queueInfo.setPreemptionDisabled(preemptionSettings.isPreemptionDisabled());
- queueInfo.setIntraQueuePreemptionDisabled(
- getIntraQueuePreemptionDisabled());
- queueInfo.setQueueConfigurations(getQueueConfigurations());
- queueInfo.setWeight(queueCapacities.getWeight());
-
queueInfo.setMaxParallelApps(queueAppLifetimeSettings.getMaxParallelApps());
- return queueInfo;
- }
-
- public QueueStatistics getQueueStatistics() {
- // Deliberately doesn't use lock here, because this method will be invoked
- // from schedulerApplicationAttempt, to avoid deadlock, sacrifice
- // consistency here.
- // TODO, improve this
- QueueStatistics stats = recordFactory.newRecordInstance(
- QueueStatistics.class);
- stats.setNumAppsSubmitted(getMetrics().getAppsSubmitted());
- stats.setNumAppsRunning(getMetrics().getAppsRunning());
- stats.setNumAppsPending(getMetrics().getAppsPending());
- stats.setNumAppsCompleted(getMetrics().getAppsCompleted());
- stats.setNumAppsKilled(getMetrics().getAppsKilled());
- stats.setNumAppsFailed(getMetrics().getAppsFailed());
- stats.setNumActiveUsers(getMetrics().getActiveUsers());
- stats.setAvailableMemoryMB(getMetrics().getAvailableMB());
- stats.setAllocatedMemoryMB(getMetrics().getAllocatedMB());
- stats.setPendingMemoryMB(getMetrics().getPendingMB());
- stats.setReservedMemoryMB(getMetrics().getReservedMB());
- stats.setAvailableVCores(getMetrics().getAvailableVirtualCores());
- stats.setAllocatedVCores(getMetrics().getAllocatedVirtualCores());
- stats.setPendingVCores(getMetrics().getPendingVirtualCores());
- stats.setReservedVCores(getMetrics().getReservedVirtualCores());
- stats.setPendingContainers(getMetrics().getPendingContainers());
- stats.setAllocatedContainers(getMetrics().getAllocatedContainers());
- stats.setReservedContainers(getMetrics().getReservedContainers());
- return stats;
- }
-
- public Map<String, QueueConfigurations> getQueueConfigurations() {
- Map<String, QueueConfigurations> queueConfigurations = new HashMap<>();
- Set<String> nodeLabels = getNodeLabelsForQueue();
- QueueResourceQuotas queueResourceQuotas =
usageTracker.getQueueResourceQuotas();
- for (String nodeLabel : nodeLabels) {
- QueueConfigurations queueConfiguration =
- recordFactory.newRecordInstance(QueueConfigurations.class);
- float capacity = queueCapacities.getCapacity(nodeLabel);
- float absoluteCapacity = queueCapacities.getAbsoluteCapacity(nodeLabel);
- float maxCapacity = queueCapacities.getMaximumCapacity(nodeLabel);
- float absMaxCapacity =
- queueCapacities.getAbsoluteMaximumCapacity(nodeLabel);
- float maxAMPercentage =
- queueCapacities.getMaxAMResourcePercentage(nodeLabel);
- queueConfiguration.setCapacity(capacity);
- queueConfiguration.setAbsoluteCapacity(absoluteCapacity);
- queueConfiguration.setMaxCapacity(maxCapacity);
- queueConfiguration.setAbsoluteMaxCapacity(absMaxCapacity);
- queueConfiguration.setMaxAMPercentage(maxAMPercentage);
- queueConfiguration.setConfiguredMinCapacity(
- queueResourceQuotas.getConfiguredMinResource(nodeLabel));
- queueConfiguration.setConfiguredMaxCapacity(
- queueResourceQuotas.getConfiguredMaxResource(nodeLabel));
- queueConfiguration.setEffectiveMinCapacity(
- queueResourceQuotas.getEffectiveMinResource(nodeLabel));
- queueConfiguration.setEffectiveMaxCapacity(
- queueResourceQuotas.getEffectiveMaxResource(nodeLabel));
- queueConfigurations.put(nodeLabel, queueConfiguration);
- }
- return queueConfigurations;
+ return CSQueueInfoProvider.getQueueInfo(this);
}
@Private
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/capacity/CSQueueInfoProvider.java
b/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/main/java/org/apache/hadoop/yarn/server/resourcemanager/scheduler/capacity/CSQueueInfoProvider.java
new file mode 100644
index 00000000000..8daca2bc26b
--- /dev/null
+++
b/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/main/java/org/apache/hadoop/yarn/server/resourcemanager/scheduler/capacity/CSQueueInfoProvider.java
@@ -0,0 +1,117 @@
+/**
+ * Licensed to the Apache Software Foundation (ASF) under one
+ * or more contributor license agreements. See the NOTICE file
+ * distributed with this work for additional information
+ * regarding copyright ownership. The ASF licenses this file
+ * to you under the Apache License, Version 2.0 (the
+ * "License"); you may not use this file except in compliance
+ * with the License. You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.apache.hadoop.yarn.server.resourcemanager.scheduler.capacity;
+
+import org.apache.hadoop.yarn.api.records.QueueConfigurations;
+import org.apache.hadoop.yarn.api.records.QueueInfo;
+import org.apache.hadoop.yarn.api.records.QueueStatistics;
+import org.apache.hadoop.yarn.factories.RecordFactory;
+import org.apache.hadoop.yarn.factory.providers.RecordFactoryProvider;
+import
org.apache.hadoop.yarn.server.resourcemanager.scheduler.QueueResourceQuotas;
+
+import java.util.HashMap;
+import java.util.Map;
+import java.util.Set;
+
+public final class CSQueueInfoProvider {
+
+ private static final RecordFactory RECORD_FACTORY =
+ RecordFactoryProvider.getRecordFactory(null);
+
+ private CSQueueInfoProvider() {
+ }
+
+ public static QueueInfo getQueueInfo(AbstractCSQueue csQueue) {
+ QueueInfo queueInfo = RECORD_FACTORY.newRecordInstance(QueueInfo.class);
+ queueInfo.setQueueName(csQueue.getQueuePathObject().getLeafName());
+ queueInfo.setQueuePath(csQueue.getQueuePathObject().getFullPath());
+ queueInfo.setAccessibleNodeLabels(csQueue.getAccessibleNodeLabels());
+ queueInfo.setCapacity(csQueue.getCapacity());
+ queueInfo.setMaximumCapacity(csQueue.getMaximumCapacity());
+ queueInfo.setQueueState(csQueue.getState());
+
queueInfo.setDefaultNodeLabelExpression(csQueue.getDefaultNodeLabelExpression());
+ queueInfo.setCurrentCapacity(csQueue.getUsedCapacity());
+ queueInfo.setQueueStatistics(getQueueStatistics(csQueue));
+ queueInfo.setPreemptionDisabled(csQueue.getPreemptionDisabled());
+ queueInfo.setIntraQueuePreemptionDisabled(
+ csQueue.getIntraQueuePreemptionDisabled());
+ queueInfo.setQueueConfigurations(getQueueConfigurations(csQueue));
+ queueInfo.setWeight(csQueue.getQueueCapacities().getWeight());
+ queueInfo.setMaxParallelApps(csQueue.getMaxParallelApps());
+ return queueInfo;
+ }
+
+ private static QueueStatistics getQueueStatistics(AbstractCSQueue csQueue) {
+ QueueStatistics stats = RECORD_FACTORY.newRecordInstance(
+ QueueStatistics.class);
+ CSQueueMetrics queueMetrics = csQueue.getMetrics();
+ stats.setNumAppsSubmitted(queueMetrics.getAppsSubmitted());
+ stats.setNumAppsRunning(queueMetrics.getAppsRunning());
+ stats.setNumAppsPending(queueMetrics.getAppsPending());
+ stats.setNumAppsCompleted(queueMetrics.getAppsCompleted());
+ stats.setNumAppsKilled(queueMetrics.getAppsKilled());
+ stats.setNumAppsFailed(queueMetrics.getAppsFailed());
+ stats.setNumActiveUsers(queueMetrics.getActiveUsers());
+ stats.setAvailableMemoryMB(queueMetrics.getAvailableMB());
+ stats.setAllocatedMemoryMB(queueMetrics.getAllocatedMB());
+ stats.setPendingMemoryMB(queueMetrics.getPendingMB());
+ stats.setReservedMemoryMB(queueMetrics.getReservedMB());
+ stats.setAvailableVCores(queueMetrics.getAvailableVirtualCores());
+ stats.setAllocatedVCores(queueMetrics.getAllocatedVirtualCores());
+ stats.setPendingVCores(queueMetrics.getPendingVirtualCores());
+ stats.setReservedVCores(queueMetrics.getReservedVirtualCores());
+ stats.setPendingContainers(queueMetrics.getPendingContainers());
+ stats.setAllocatedContainers(queueMetrics.getAllocatedContainers());
+ stats.setReservedContainers(queueMetrics.getReservedContainers());
+ return stats;
+ }
+
+ private static Map<String, QueueConfigurations>
getQueueConfigurations(AbstractCSQueue csQueue) {
+ Map<String, QueueConfigurations> queueConfigurations = new HashMap<>();
+ Set<String> nodeLabels = csQueue.getNodeLabelsForQueue();
+ QueueResourceQuotas queueResourceQuotas = csQueue.getQueueResourceQuotas();
+ for (String nodeLabel : nodeLabels) {
+ QueueConfigurations queueConfiguration =
+ RECORD_FACTORY.newRecordInstance(QueueConfigurations.class);
+ QueueCapacities queueCapacities = csQueue.getQueueCapacities();
+ float capacity = queueCapacities.getCapacity(nodeLabel);
+ float absoluteCapacity = queueCapacities.getAbsoluteCapacity(nodeLabel);
+ float maxCapacity = queueCapacities.getMaximumCapacity(nodeLabel);
+ float absMaxCapacity =
+ queueCapacities.getAbsoluteMaximumCapacity(nodeLabel);
+ float maxAMPercentage =
+ queueCapacities.getMaxAMResourcePercentage(nodeLabel);
+ queueConfiguration.setCapacity(capacity);
+ queueConfiguration.setAbsoluteCapacity(absoluteCapacity);
+ queueConfiguration.setMaxCapacity(maxCapacity);
+ queueConfiguration.setAbsoluteMaxCapacity(absMaxCapacity);
+ queueConfiguration.setMaxAMPercentage(maxAMPercentage);
+ queueConfiguration.setConfiguredMinCapacity(
+ queueResourceQuotas.getConfiguredMinResource(nodeLabel));
+ queueConfiguration.setConfiguredMaxCapacity(
+ queueResourceQuotas.getConfiguredMaxResource(nodeLabel));
+ queueConfiguration.setEffectiveMinCapacity(
+ queueResourceQuotas.getEffectiveMinResource(nodeLabel));
+ queueConfiguration.setEffectiveMaxCapacity(
+ queueResourceQuotas.getEffectiveMaxResource(nodeLabel));
+ queueConfigurations.put(nodeLabel, queueConfiguration);
+ }
+ return queueConfigurations;
+ }
+}
---------------------------------------------------------------------
To unsubscribe, e-mail: common-commits-unsubscr...@hadoop.apache.org
For additional commands, e-mail: common-commits-h...@hadoop.apache.org
