| /** |
| * 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.webapp.dao; |
| |
| import java.util.ArrayList; |
| import java.util.Collections; |
| import java.util.HashMap; |
| import java.util.Map; |
| import java.util.Set; |
| import java.util.regex.Pattern; |
| |
| import javax.xml.bind.annotation.XmlAccessType; |
| import javax.xml.bind.annotation.XmlAccessorType; |
| import javax.xml.bind.annotation.XmlRootElement; |
| import javax.xml.bind.annotation.XmlSeeAlso; |
| import javax.xml.bind.annotation.XmlTransient; |
| |
| import org.apache.hadoop.util.StringUtils; |
| import org.apache.hadoop.security.authorize.AccessControlList; |
| import org.apache.hadoop.yarn.api.records.QueueState; |
| import org.apache.hadoop.yarn.security.AccessType; |
| import org.apache.hadoop.yarn.server.resourcemanager.scheduler.QueueResourceQuotas; |
| import org.apache.hadoop.yarn.server.resourcemanager.scheduler.ResourceUsage; |
| import org.apache.hadoop.yarn.server.resourcemanager.scheduler.capacity.CapacityScheduler; |
| import org.apache.hadoop.yarn.server.resourcemanager.scheduler.capacity.CapacitySchedulerConfiguration; |
| import org.apache.hadoop.yarn.server.resourcemanager.scheduler.capacity.CSQueue; |
| import org.apache.hadoop.yarn.server.resourcemanager.scheduler.capacity.ParentQueue; |
| import org.apache.hadoop.yarn.server.resourcemanager.scheduler.capacity.PlanQueue; |
| import org.apache.hadoop.yarn.server.resourcemanager.scheduler.capacity.QueueCapacities; |
| import org.apache.hadoop.yarn.server.resourcemanager.webapp.dao.helper.CapacitySchedulerInfoHelper; |
| |
| import static org.apache.hadoop.yarn.server.resourcemanager.scheduler.capacity. |
| CapacitySchedulerConfiguration.RESOURCE_PATTERN; |
| import static org.apache.hadoop.yarn.server.resourcemanager.scheduler.capacity. |
| CapacitySchedulerConfiguration.CAPACITY; |
| |
| @XmlRootElement |
| @XmlAccessorType(XmlAccessType.FIELD) |
| @XmlSeeAlso({CapacitySchedulerLeafQueueInfo.class}) |
| public class CapacitySchedulerQueueInfo { |
| |
| @XmlTransient |
| static final float EPSILON = 1e-8f; |
| |
| protected String queuePath; |
| protected float capacity; |
| protected float usedCapacity; |
| protected float maxCapacity; |
| protected float absoluteCapacity; |
| protected float absoluteMaxCapacity; |
| protected float absoluteUsedCapacity; |
| protected float weight; |
| protected float normalizedWeight; |
| protected int numApplications; |
| protected String queueName; |
| protected boolean isAbsoluteResource; |
| protected QueueState state; |
| protected CapacitySchedulerQueueInfoList queues; |
| protected ResourceInfo resourcesUsed; |
| private boolean hideReservationQueues = false; |
| protected ArrayList<String> nodeLabels = new ArrayList<String>(); |
| protected long allocatedContainers; |
| protected long reservedContainers; |
| protected long pendingContainers; |
| protected QueueCapacitiesInfo capacities; |
| protected ResourcesInfo resources; |
| protected ResourceInfo minEffectiveCapacity; |
| protected ResourceInfo maxEffectiveCapacity; |
| protected ResourceInfo maximumAllocation; |
| protected QueueAclsInfo queueAcls; |
| protected int queuePriority; |
| protected String orderingPolicyInfo; |
| protected boolean autoCreateChildQueueEnabled; |
| protected LeafQueueTemplateInfo leafQueueTemplate; |
| protected String mode; |
| protected String queueType; |
| protected String creationMethod; |
| protected String autoCreationEligibility; |
| |
| CapacitySchedulerQueueInfo() { |
| }; |
| |
| CapacitySchedulerQueueInfo(CapacityScheduler cs, CSQueue q) { |
| |
| queuePath = q.getQueuePath(); |
| capacity = q.getCapacity() * 100; |
| usedCapacity = q.getUsedCapacity() * 100; |
| |
| maxCapacity = q.getMaximumCapacity(); |
| if (maxCapacity < EPSILON || maxCapacity > 1f) |
| maxCapacity = 1f; |
| maxCapacity *= 100; |
| |
| absoluteCapacity = |
| cap(q.getAbsoluteCapacity(), 0f, 1f) * 100; |
| absoluteMaxCapacity = |
| cap(q.getAbsoluteMaximumCapacity(), 0f, 1f) * 100; |
| absoluteUsedCapacity = |
| cap(q.getAbsoluteUsedCapacity(), 0f, 1f) * 100; |
| weight = q.getQueueCapacities().getWeight(); |
| normalizedWeight = q.getQueueCapacities().getNormalizedWeight(); |
| numApplications = q.getNumApplications(); |
| allocatedContainers = q.getMetrics().getAllocatedContainers(); |
| pendingContainers = q.getMetrics().getPendingContainers(); |
| reservedContainers = q.getMetrics().getReservedContainers(); |
| queueName = q.getQueueName(); |
| state = q.getState(); |
| resourcesUsed = new ResourceInfo(q.getUsedResources()); |
| if (q instanceof PlanQueue && !((PlanQueue) q).showReservationsAsQueues()) { |
| hideReservationQueues = true; |
| } |
| |
| // add labels |
| Set<String> labelSet = q.getAccessibleNodeLabels(); |
| if (labelSet != null) { |
| nodeLabels.addAll(labelSet); |
| Collections.sort(nodeLabels); |
| } |
| QueueCapacities qCapacities = q.getQueueCapacities(); |
| QueueResourceQuotas qResQuotas = q.getQueueResourceQuotas(); |
| populateQueueCapacities(qCapacities, qResQuotas); |
| |
| mode = CapacitySchedulerInfoHelper.getMode(q); |
| queueType = CapacitySchedulerInfoHelper.getQueueType(q); |
| creationMethod = CapacitySchedulerInfoHelper.getCreationMethod(q); |
| autoCreationEligibility = CapacitySchedulerInfoHelper |
| .getAutoCreationEligibility(q); |
| |
| ResourceUsage queueResourceUsage = q.getQueueResourceUsage(); |
| populateQueueResourceUsage(queueResourceUsage); |
| |
| minEffectiveCapacity = new ResourceInfo( |
| q.getQueueResourceQuotas().getEffectiveMinResource()); |
| maxEffectiveCapacity = new ResourceInfo( |
| q.getQueueResourceQuotas().getEffectiveMaxResource()); |
| maximumAllocation = new ResourceInfo(q.getMaximumAllocation()); |
| |
| CapacitySchedulerConfiguration conf = cs.getConfiguration(); |
| queueAcls = new QueueAclsInfo(); |
| for (Map.Entry<AccessType, AccessControlList> e : conf |
| .getAcls(queuePath).entrySet()) { |
| QueueAclInfo queueAcl = new QueueAclInfo(e.getKey().toString(), |
| e.getValue().getAclString()); |
| queueAcls.add(queueAcl); |
| } |
| |
| String aclApplicationMaxPriority = "acl_" + |
| StringUtils.toLowerCase(AccessType.APPLICATION_MAX_PRIORITY.toString()); |
| String priorityAcls = conf.get(CapacitySchedulerConfiguration |
| .getQueuePrefix(queuePath) + aclApplicationMaxPriority, conf.ALL_ACL); |
| |
| QueueAclInfo queueAcl = new QueueAclInfo( |
| AccessType.APPLICATION_MAX_PRIORITY.toString(), priorityAcls); |
| queueAcls.add(queueAcl); |
| |
| queuePriority = q.getPriority().getPriority(); |
| if (q instanceof ParentQueue) { |
| orderingPolicyInfo = ((ParentQueue) q).getQueueOrderingPolicy() |
| .getConfigName(); |
| } |
| |
| String configuredCapacity = conf.get( |
| CapacitySchedulerConfiguration.getQueuePrefix(queuePath) + CAPACITY); |
| isAbsoluteResource = (configuredCapacity != null) |
| && RESOURCE_PATTERN.matcher(configuredCapacity).find(); |
| |
| autoCreateChildQueueEnabled = conf. |
| isAutoCreateChildQueueEnabled(queuePath); |
| leafQueueTemplate = new LeafQueueTemplateInfo(conf, queuePath); |
| } |
| |
| protected void populateQueueResourceUsage(ResourceUsage queueResourceUsage) { |
| resources = new ResourcesInfo(queueResourceUsage, false); |
| } |
| |
| protected void populateQueueCapacities(QueueCapacities qCapacities, |
| QueueResourceQuotas qResQuotas) { |
| capacities = new QueueCapacitiesInfo(qCapacities, qResQuotas, |
| false); |
| } |
| |
| public float getCapacity() { |
| return this.capacity; |
| } |
| |
| public float getUsedCapacity() { |
| return this.usedCapacity; |
| } |
| |
| public float getMaxCapacity() { |
| return this.maxCapacity; |
| } |
| |
| public float getAbsoluteCapacity() { |
| return absoluteCapacity; |
| } |
| |
| public float getAbsoluteMaxCapacity() { |
| return absoluteMaxCapacity; |
| } |
| |
| public float getAbsoluteUsedCapacity() { |
| return absoluteUsedCapacity; |
| } |
| |
| public int getNumApplications() { |
| return numApplications; |
| } |
| |
| public long getAllocatedContainers() { |
| return allocatedContainers; |
| } |
| |
| public long getReservedContainers() { |
| return reservedContainers; |
| } |
| |
| public long getPendingContainers() { |
| return pendingContainers; |
| } |
| |
| public boolean isAbsoluteResource() { |
| return isAbsoluteResource; |
| } |
| |
| public String getQueueName() { |
| return this.queueName; |
| } |
| |
| public String getQueueState() { |
| return this.state.toString(); |
| } |
| |
| public String getQueuePath() { |
| return this.queuePath; |
| } |
| |
| public CapacitySchedulerQueueInfoList getQueues() { |
| if(hideReservationQueues) { |
| return new CapacitySchedulerQueueInfoList(); |
| } |
| return this.queues; |
| } |
| |
| public ResourceInfo getResourcesUsed() { |
| return resourcesUsed; |
| } |
| |
| /** |
| * Limit a value to a specified range. |
| * @param val the value to be capped |
| * @param low the lower bound of the range (inclusive) |
| * @param hi the upper bound of the range (inclusive) |
| * @return the capped value |
| */ |
| static float cap(float val, float low, float hi) { |
| return Math.min(Math.max(val, low), hi); |
| } |
| |
| public ArrayList<String> getNodeLabels() { |
| return this.nodeLabels; |
| } |
| |
| public QueueCapacitiesInfo getCapacities() { |
| return capacities; |
| } |
| |
| public ResourcesInfo getResources() { |
| return resources; |
| } |
| |
| public ResourceInfo getMinEffectiveCapacity(){ |
| return minEffectiveCapacity; |
| } |
| |
| public ResourceInfo getMaxEffectiveCapacity(){ |
| return maxEffectiveCapacity; |
| } |
| |
| public ResourceInfo getMaximumAllocation() { |
| return maximumAllocation; |
| } |
| |
| public QueueAclsInfo getQueueAcls() { |
| return queueAcls; |
| } |
| |
| public int getPriority() { |
| return queuePriority; |
| } |
| |
| public String getOrderingPolicyInfo() { |
| return orderingPolicyInfo; |
| } |
| |
| public boolean isLeafQueue() { |
| return getQueues() == null; |
| } |
| |
| public boolean isAutoCreateChildQueueEnabled() { |
| return autoCreateChildQueueEnabled; |
| } |
| |
| public LeafQueueTemplateInfo getLeafQueueTemplate() { |
| return leafQueueTemplate; |
| } |
| |
| public String getMode() { |
| return mode; |
| } |
| |
| public String getQueueType() { |
| return queueType; |
| } |
| |
| public float getWeight() { |
| return weight; |
| } |
| |
| public float getNormalizedWeight() { |
| return normalizedWeight; |
| } |
| } |