private void updateNodeLabelsInternal(Set<NodeId> nodes) throws IOException { Map<NodeId, Set<NodeLabel>> labelsUpdated = rmNodeLabelsMappingProvider.getNodeLabels(nodes); if (labelsUpdated != null && labelsUpdated.size() != 0) { Map<NodeId, Set<String>> nodeToLabels = new HashMap<NodeId, Set<String>>(labelsUpdated.size()); for (Map.Entry<NodeId, Set<NodeLabel>> entry : labelsUpdated.entrySet()) { nodeToLabels.put(entry.getKey(), NodeLabelsUtils.convertToStringSet(entry.getValue())); } rmContext.getNodeLabelManager().replaceLabelsOnNode(nodeToLabels); } }
Set<String> nodeLabels = NodeLabelsUtils.convertToStringSet( request.getNodeLabels()); if (isDistributedNodeLabelsConf && nodeLabels != null) {
try { updateNodeLabelsFromNMReport( NodeLabelsUtils.convertToStringSet(request.getNodeLabels()), nodeId); nodeHeartBeatResponse.setAreNodeLabelsAcceptedByRM(true);
NodeAction.NORMAL, response.getNodeAction()); assertCollectionEquals(nodeLabelsMgr.getNodeLabels().get(nodeId), NodeLabelsUtils.convertToStringSet(registerReq.getNodeLabels())); Assert.assertTrue("Valid Node Labels were not accepted by RM", response.getAreNodeLabelsAcceptedByRM());
NodeAction.NORMAL, nodeHeartbeatResponse.getNodeAction()); assertCollectionEquals(nodeLabelsMgr.getNodeLabels().get(nodeId), NodeLabelsUtils.convertToStringSet(heartbeatReq.getNodeLabels())); Assert.assertTrue("Valid Node Labels were not accepted by RM", nodeHeartbeatResponse.getAreNodeLabelsAcceptedByRM());