@SuppressWarnings("unchecked") @VisibleForTesting public void reSortClusterNodes() { Set<String> nodeLabels = new HashSet<>(); nodeLabels .addAll(rmContext.getNodeLabelManager().getClusterNodeLabelNames()); nodeLabels.add(RMNodeLabelsManager.NO_LABEL); for (String label : nodeLabels) { Map<NodeId, SchedulerNode> nodesByPartition = new HashMap<>(); List<SchedulerNode> nodes = ((AbstractYarnScheduler) rmContext .getScheduler()).getNodeTracker().getNodesPerPartition(label); if (nodes != null && !nodes.isEmpty()) { nodes.forEach(n -> nodesByPartition.put(n.getNodeID(), n)); multiNodePolicy.addAndRefreshNodesSet( (Collection<N>) nodesByPartition.values(), label); } } }
abstractYarnScheduler.getNodeTracker().getAllNodes();
while (waitTime > 0 && ((AbstractYarnScheduler) rm.getRMContext().getScheduler()) .getNodeTracker().nodeCount() < 2) { waitTime -= 10; Thread.sleep(10); .getNodeTracker().nodeCount());
.getNodeTracker().nodeCount()); CapacityScheduler scheduler = (CapacityScheduler) rm.getRMContext().getScheduler();
.getNodeTracker().nodeCount()); CapacityScheduler scheduler = (CapacityScheduler) rm.getRMContext().getScheduler();
.getNodeTracker().nodeCount()); CapacityScheduler cs = (CapacityScheduler) rm.getRMContext().getScheduler();