private void waitSchedulerNodeHasUpdatedLabels(CapacityScheduler cs, MockNM nm, String partition) throws InterruptedException { FiCaSchedulerNode node = cs.getNode(nm.getNodeId()); int totalWaitTick = 20; // wait 2 sec at most. while (!node.getLabels().contains(partition) && totalWaitTick > 0) { Thread.sleep(100); totalWaitTick--; } }
rm.getRMContext()); SchedulerContainer reservedContainer = new SchedulerContainer(schedulerApp, scheduler.getNode(sn2.getNodeID()), rmContainer, "", false); ContainerAllocationProposal reservedForAttempt1Proposal =
FiCaSchedulerNode node = getNode(nm.getNodeID());
((CapacityScheduler) scheduler).getNode(idToUnreserve); if (nodeToUnreserve == null) { LOG.error("node to unreserve doesn't exist, nodeid: " + idToUnreserve);
FiCaSchedulerNode node = getNode(nm.getNodeID());
private int checkNumNonAMContainersOnNode(CapacityScheduler cs, MockNM nm) { SchedulerNode node = cs.getNode(nm.getNodeId()); int nonAMContainer = 0; for (RMContainer c : node.getCopiedListOfRunningContainers()) { if (!c.isAMContainer()) { nonAMContainer++; } } return nonAMContainer; }
FiCaSchedulerNode node = getNode(container.getNodeId());
FiCaSchedulerNode node = getNode(container.getNodeId());
private void verifyAvailableResourceOfSchedulerNode(MockRM rm, NodeId nodeId, int expectedMemory) { CapacityScheduler cs = (CapacityScheduler) rm.getResourceScheduler(); SchedulerNode node = cs.getNode(nodeId); Assert .assertEquals(expectedMemory, node.getUnallocatedResource().getMemorySize()); }
private void verifyAvailableResourceOfSchedulerNode(MockRM rm, NodeId nodeId, int expectedMemory) { CapacityScheduler cs = (CapacityScheduler) rm.getResourceScheduler(); SchedulerNode node = cs.getNode(nodeId); Assert .assertEquals(expectedMemory, node.getUnallocatedResource().getMemorySize()); } }
private SchedulerContainer<FiCaSchedulerApp, FiCaSchedulerNode> getSchedulerContainer( RMContainer rmContainer, boolean allocated) { if (null == rmContainer) { return null; } FiCaSchedulerApp app = getApplicationAttempt( rmContainer.getApplicationAttemptId()); if (null == app) { return null; } NodeId nodeId; // Get nodeId if (rmContainer.getState() == RMContainerState.RESERVED) { nodeId = rmContainer.getReservedNode(); } else { nodeId = rmContainer.getNodeId(); } FiCaSchedulerNode node = getNode(nodeId); if (null == node) { return null; } return new SchedulerContainer<>(app, node, rmContainer, // TODO, node partition should come from CSAssignment to avoid partition // get updated before submitting the commit node.getPartition(), allocated); }
FiCaSchedulerNode node = getNode(container.getNodeId()); if (null == node) { LOG.info("Container " + container + " of" + " removed node " + container
FiCaSchedulerNode node = getNode(nodeId); if (null != node) { int offswitchCount = 0;
Collections.<ContainerId>emptyList(), null, null, NULL_UPDATE_REQUESTS); spyCs.handle(new NodeUpdateSchedulerEvent( spyCs.getNode(nm.getNodeId()).getRMNode()));
Assert.assertEquals(0 * GB, cs.getNode(nm1.getNodeId()) .getUnallocatedResource().getMemorySize()); am2.allocate("*", 1 * GB, 1, new ArrayList<ContainerId>());
.getApplicationAttemptId()); FiCaSchedulerNode node = cs.getNode(rmContainer.getAllocatedNode()); for (ResourceRequest request : requests) {
.getApplicationAttemptId()); FiCaSchedulerNode node = cs.getNode(rmContainer.getAllocatedNode()); for (ResourceRequest request : requests) {
Assert.assertEquals(0 * GB, cs.getNode(nm1.getNodeId()) .getUnallocatedResource().getMemorySize()); am2.allocate("*", 3 * GB, 1, new ArrayList<ContainerId>());