YARN-5533. JMX AM Used metrics for queue wrong when app submited to nodelabel partition (Bibin A Chundatt via Varun Saxena)
This commit is contained in:
parent
8179f9a493
commit
59557e85a4
|
@ -774,7 +774,8 @@ public class LeafQueue extends AbstractCSQueue {
|
||||||
application.getAMResource(partitionName));
|
application.getAMResource(partitionName));
|
||||||
user.getResourceUsage().decAMUsed(partitionName,
|
user.getResourceUsage().decAMUsed(partitionName,
|
||||||
application.getAMResource(partitionName));
|
application.getAMResource(partitionName));
|
||||||
metrics.decAMUsed(application.getUser(), application.getAMResource());
|
metrics.decAMUsed(application.getUser(),
|
||||||
|
application.getAMResource(partitionName));
|
||||||
}
|
}
|
||||||
applicationAttemptMap.remove(application.getApplicationAttemptId());
|
applicationAttemptMap.remove(application.getApplicationAttemptId());
|
||||||
|
|
||||||
|
|
|
@ -41,6 +41,7 @@ import org.apache.hadoop.yarn.server.resourcemanager.ResourceManager;
|
||||||
import org.apache.hadoop.yarn.server.resourcemanager.nodelabels.NullRMNodeLabelsManager;
|
import org.apache.hadoop.yarn.server.resourcemanager.nodelabels.NullRMNodeLabelsManager;
|
||||||
import org.apache.hadoop.yarn.server.resourcemanager.nodelabels.RMNodeLabelsManager;
|
import org.apache.hadoop.yarn.server.resourcemanager.nodelabels.RMNodeLabelsManager;
|
||||||
import org.apache.hadoop.yarn.server.resourcemanager.rmapp.RMApp;
|
import org.apache.hadoop.yarn.server.resourcemanager.rmapp.RMApp;
|
||||||
|
import org.apache.hadoop.yarn.server.resourcemanager.rmapp.RMAppState;
|
||||||
import org.apache.hadoop.yarn.server.resourcemanager.rmapp.attempt.RMAppAttemptState;
|
import org.apache.hadoop.yarn.server.resourcemanager.rmapp.attempt.RMAppAttemptState;
|
||||||
import org.apache.hadoop.yarn.server.resourcemanager.rmcontainer.RMContainer;
|
import org.apache.hadoop.yarn.server.resourcemanager.rmcontainer.RMContainer;
|
||||||
import org.apache.hadoop.yarn.server.resourcemanager.rmcontainer.RMContainerState;
|
import org.apache.hadoop.yarn.server.resourcemanager.rmcontainer.RMContainerState;
|
||||||
|
@ -1959,6 +1960,14 @@ public class TestNodeLabelContainerAllocation {
|
||||||
LeafQueue leafQueue = (LeafQueue) cs.getQueue("a");
|
LeafQueue leafQueue = (LeafQueue) cs.getQueue("a");
|
||||||
assertEquals(0 * GB, leafQueue.getMetrics().getAvailableMB());
|
assertEquals(0 * GB, leafQueue.getMetrics().getAvailableMB());
|
||||||
assertEquals(5 * GB, leafQueue.getMetrics().getAllocatedMB());
|
assertEquals(5 * GB, leafQueue.getMetrics().getAllocatedMB());
|
||||||
|
|
||||||
|
// Kill all apps in queue a
|
||||||
|
cs.killAllAppsInQueue("a");
|
||||||
|
rm1.waitForState(app1.getApplicationId(), RMAppState.KILLED);
|
||||||
|
rm1.waitForAppRemovedFromScheduler(app1.getApplicationId());
|
||||||
|
|
||||||
|
assertEquals(0 * GB, leafQueue.getMetrics().getUsedAMResourceMB());
|
||||||
|
assertEquals(0 * GB, leafQueue.getMetrics().getUsedAMResourceVCores());
|
||||||
rm1.close();
|
rm1.close();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue