Set<String> labels = new AddToClusterNodeLabelsRequestPBImpl( AddToClusterNodeLabelsRequestProto.parseDelimitedFrom(is)).getNodeLabels(); Map<NodeId, Set<String>> nodeToLabels = new ReplaceLabelsOnNodeRequestPBImpl( case ADD_LABELS: { Collection<String> labels = AddToClusterNodeLabelsRequestProto.parseDelimitedFrom(is) .getNodeLabelsList(); mgr.addToCluserNodeLabels(Sets.newHashSet(labels.iterator()));
Set<String> labels = new AddToClusterNodeLabelsRequestPBImpl( AddToClusterNodeLabelsRequestProto.parseDelimitedFrom(is)).getNodeLabels(); Map<NodeId, Set<String>> nodeToLabels = new ReplaceLabelsOnNodeRequestPBImpl( case ADD_LABELS: { Collection<String> labels = AddToClusterNodeLabelsRequestProto.parseDelimitedFrom(is) .getNodeLabelsList(); mgr.addToCluserNodeLabels(Sets.newHashSet(labels.iterator()));
@Override public void recover(InputStream is, CommonNodeLabelsManager mgr) throws IOException { List<NodeLabel> labels = new AddToClusterNodeLabelsRequestPBImpl( YarnServerResourceManagerServiceProtos .AddToClusterNodeLabelsRequestProto .parseDelimitedFrom(is)).getNodeLabels(); mgr.addToCluserNodeLabels(labels); if (mgr.isCentralizedConfiguration()) { // Only load node to labels mapping while using centralized // configuration Map<NodeId, Set<String>> nodeToLabels = new ReplaceLabelsOnNodeRequestPBImpl( YarnServerResourceManagerServiceProtos .ReplaceLabelsOnNodeRequestProto .parseDelimitedFrom(is)).getNodeToLabels(); mgr.replaceLabelsOnNode(nodeToLabels); } }
@Override public void recover(InputStream is, CommonNodeLabelsManager mgr) throws IOException { labels = new AddToClusterNodeLabelsRequestPBImpl( YarnServerResourceManagerServiceProtos .AddToClusterNodeLabelsRequestProto .parseDelimitedFrom(is)).getNodeLabels(); mgr.addToCluserNodeLabels(labels); }