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); } }