@Override public void handle(Event event) { if (event instanceof RMNodeResourceUpdateEvent) { RMNodeResourceUpdateEvent resourceEvent = (RMNodeResourceUpdateEvent) event; resourceManager .getResourceScheduler() .getSchedulerNode(resourceEvent.getNodeId()) .updateTotalResource(resourceEvent.getResourceOption().getResource()); } } });
@Override public void handle(Event event) { if (event instanceof RMNodeResourceUpdateEvent) { RMNodeResourceUpdateEvent resourceEvent = (RMNodeResourceUpdateEvent) event; resourceManager .getResourceScheduler() .getSchedulerNode(resourceEvent.getNodeId()) .updateTotalResource(resourceEvent.getResourceOption().getResource()); } } });
@Override public void handle(Event event) { if (event instanceof RMNodeResourceUpdateEvent) { RMNodeResourceUpdateEvent resourceEvent = (RMNodeResourceUpdateEvent) event; resourceManager .getResourceScheduler() .getSchedulerNode(resourceEvent.getNodeId()) .updateTotalResource(resourceEvent.getResourceOption().getResource()); } } });
private void handleIncreaseRequests( SchedulerApplicationAttempt applicationAttempt, List<UpdateContainerRequest> updateContainerRequests) { for (UpdateContainerRequest uReq : updateContainerRequests) { RMContainer rmContainer = rmContext.getScheduler().getRMContainer(uReq.getContainerId()); // Check if this is a container update // And not in the middle of a Demotion if (rmContainer != null) { // Check if this is an executionType change request // If so, fix the rr to make it look like a normal rr // with relaxLocality=false and numContainers=1 SchedulerNode schedulerNode = rmContext.getScheduler() .getSchedulerNode(rmContainer.getContainer().getNodeId()); // Add only if no outstanding promote requests exist. if (!applicationAttempt.getUpdateContext() .checkAndAddToOutstandingIncreases( rmContainer, schedulerNode, uReq)) { applicationAttempt.addToUpdateContainerErrors( UpdateContainerError.newInstance( RMServerUtils.UPDATE_OUTSTANDING_ERROR, uReq)); } } else { LOG.warn("Cannot promote non-existent (or completed) Container [" + uReq.getContainerId() + "]"); } } }
.getSchedulerNode(nm1.getNodeId()).containerStarted(cid);
if (rmContainer != null) { SchedulerNode schedulerNode = rmContext.getScheduler() .getSchedulerNode(rmContainer.getContainer().getNodeId()); if (appAttempt.getUpdateContext() .checkAndAddToOutstandingDecreases(uReq, schedulerNode,
.getSchedulerNode(nm_0.getNodeId()).getAllocatedResource(); Assert.assertEquals(usedResource.getMemorySize(), 0); Assert.assertEquals(usedResource.getVirtualCores(), 0); .getSchedulerNode(nm_0.getNodeId()).getTotalResource(); Assert.assertEquals(totalResource.getMemorySize(), 0 * GB); Assert.assertEquals(totalResource.getVirtualCores(), 0); .getSchedulerNode(nm_0.getNodeId()).getUnallocatedResource(); Assert.assertEquals(availableResource.getMemorySize(), 0); Assert.assertEquals(availableResource.getVirtualCores(), 0);
.getSchedulerNode(nm_0.getNodeId()).getAllocatedResource(); Assert.assertEquals(usedResource.getMemorySize(), 1 * GB); Assert.assertEquals(usedResource.getVirtualCores(), 1); .getSchedulerNode(nm_0.getNodeId()).getTotalResource(); Assert.assertEquals(totalResource.getMemorySize(), 1 * GB); Assert.assertEquals(totalResource.getVirtualCores(), 1); .getSchedulerNode(nm_0.getNodeId()).getUnallocatedResource(); Assert.assertEquals(availableResource.getMemorySize(), 0); Assert.assertEquals(availableResource.getVirtualCores(), 0);
rmContext.getScheduler().getSchedulerNode(ni.getNodeID()); cu = scheduler.getAggregatedContainersUtilization(); assertEquals("Containers Utillization not propagated to SchedulerNode",
Resource usedResource = resourceManager.getResourceScheduler() .getSchedulerNode(nm_0.getNodeId()).getAllocatedResource(); Assert.assertEquals("Used Resource Memory Size should be 1GB", 1 * GB, usedResource.getMemorySize()); .getSchedulerNode(nm_0.getNodeId()).getTotalResource(); Assert.assertEquals("Total Resource Memory Size should be 1GB", 1 * GB, totalResource.getMemorySize()); .getSchedulerNode(nm_0.getNodeId()).getUnallocatedResource(); Assert.assertEquals("Available Resource Memory Size should be 0", 0, availableResource.getMemorySize());
rm.getRMContext().getScheduler().getSchedulerNode(nodeId); Assert.assertNotNull(schedulerNode); Resource nodeResources = schedulerNode.getAllocatedResource();