YARN-6681. Eliminate double-copy of child queues in canAssignToThisQueue. Contributed by Daryn Sharp.

This commit is contained in:
Naganarasimha 2017-07-01 12:29:39 +05:30
parent 147df300bf
commit fa1aaee87b
2 changed files with 13 additions and 2 deletions

View File

@ -22,6 +22,7 @@
import java.util.HashMap;
import java.util.HashSet;
import java.util.Iterator;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.concurrent.locks.ReentrantReadWriteLock;
@ -615,6 +616,11 @@ Resource getQueueMaxResource(String nodePartition, Resource clusterResource) {
minimumAllocation);
}
public boolean hasChildQueues() {
List<CSQueue> childQueues = getChildQueues();
return childQueues != null && !childQueues.isEmpty();
}
boolean canAssignToThisQueue(Resource clusterResource,
String nodePartition, ResourceLimits currentResourceLimits,
Resource resourceCouldBeUnreserved, SchedulingMode schedulingMode) {
@ -640,7 +646,7 @@ boolean canAssignToThisQueue(Resource clusterResource,
// When queue is a parent queue: Headroom = limit - used + killable
// When queue is a leaf queue: Headroom = limit - used (leaf queue cannot preempt itself)
Resource usedExceptKillable = nowTotalUsed;
if (null != getChildQueues() && !getChildQueues().isEmpty()) {
if (hasChildQueues()) {
usedExceptKillable = Resources.subtract(nowTotalUsed,
getTotalKillableResource(nodePartition));
}

View File

@ -845,7 +845,12 @@ public void updateClusterResource(Resource clusterResource,
writeLock.unlock();
}
}
@Override
public boolean hasChildQueues() {
return true;
}
@Override
public List<CSQueue> getChildQueues() {
try {