@Override public void addToCluserNodeLabels(Collection<NodeLabel> labels) throws IOException { try { writeLock.lock(); super.addToCluserNodeLabels(labels); } finally { writeLock.unlock(); } }
/** * Add multiple node labels to repository * * @param labels * new node labels added */ @VisibleForTesting public void addToCluserNodeLabelsWithDefaultExclusivity(Set<String> labels) throws IOException { Set<NodeLabel> nodeLabels = new HashSet<NodeLabel>(); for (String label : labels) { nodeLabels.add(NodeLabel.newInstance(label)); } addToCluserNodeLabels(nodeLabels); }
@Override public void recover(InputStream is, CommonNodeLabelsManager mgr) throws IOException { labels = new AddToClusterNodeLabelsRequestPBImpl( YarnServerResourceManagerServiceProtos .AddToClusterNodeLabelsRequestProto .parseDelimitedFrom(is)).getNodeLabels(); mgr.addToCluserNodeLabels(labels); }
private int handleAddToClusterNodeLabels(String[] args, String cmd, boolean isHAEnabled) throws IOException, YarnException, ParseException { Options opts = new Options(); opts.addOption("addToClusterNodeLabels", true, "Add to cluster node labels."); opts.addOption("directlyAccessNodeLabelStore", false, "Directly access node label store."); int exitCode = -1; CommandLine cliParser = null; try { cliParser = new GnuParser().parse(opts, args); } catch (MissingArgumentException ex) { System.err.println(NO_LABEL_ERR_MSG); printUsage(args[0], isHAEnabled); return exitCode; } List<NodeLabel> labels = YarnClientUtils.buildNodeLabelsFromStr( cliParser.getOptionValue("addToClusterNodeLabels")); if (cliParser.hasOption("directlyAccessNodeLabelStore")) { getNodeLabelManagerInstance(getConf()).addToCluserNodeLabels(labels); } else { ResourceManagerAdministrationProtocol adminProtocol = createAdminProtocol(); AddToClusterNodeLabelsRequest request = AddToClusterNodeLabelsRequest.newInstance(labels); adminProtocol.addToClusterNodeLabels(request); } return 0; }
private int handleAddToClusterNodeLabels(String[] args, String cmd, boolean isHAEnabled) throws IOException, YarnException, ParseException { Options opts = new Options(); opts.addOption("addToClusterNodeLabels", true, "Add to cluster node labels."); opts.addOption("directlyAccessNodeLabelStore", false, "Directly access node label store."); int exitCode = -1; CommandLine cliParser = null; try { cliParser = new GnuParser().parse(opts, args); } catch (MissingArgumentException ex) { System.err.println(NO_LABEL_ERR_MSG); printUsage(args[0], isHAEnabled); return exitCode; } List<NodeLabel> labels = buildNodeLabelsFromStr( cliParser.getOptionValue("addToClusterNodeLabels")); if (cliParser.hasOption("directlyAccessNodeLabelStore")) { getNodeLabelManagerInstance(getConf()).addToCluserNodeLabels(labels); } else { ResourceManagerAdministrationProtocol adminProtocol = createAdminProtocol(); AddToClusterNodeLabelsRequest request = AddToClusterNodeLabelsRequest.newInstance(labels); adminProtocol.addToClusterNodeLabels(request); } return 0; }
ReplaceLabelsOnNodeRequestProto.parseDelimitedFrom(is)) .getNodeToLabels(); mgr.addToCluserNodeLabels(labels); mgr.replaceLabelsOnNode(nodeToLabels); is.close(); AddToClusterNodeLabelsRequestProto.parseDelimitedFrom(is) .getNodeLabelsList(); mgr.addToCluserNodeLabels(Sets.newHashSet(labels.iterator())); break;
ReplaceLabelsOnNodeRequestProto.parseDelimitedFrom(is)) .getNodeToLabels(); mgr.addToCluserNodeLabels(labels); mgr.replaceLabelsOnNode(nodeToLabels); is.close(); AddToClusterNodeLabelsRequestProto.parseDelimitedFrom(is) .getNodeLabelsList(); mgr.addToCluserNodeLabels(Sets.newHashSet(labels.iterator())); break;
private int addToClusterNodeLabels(String args) throws IOException, YarnException { Set<String> labels = buildNodeLabelsSetFromStr(args); if (directlyAccessNodeLabelStore) { getNodeLabelManagerInstance(getConf()).addToCluserNodeLabels(labels); } else { ResourceManagerAdministrationProtocol adminProtocol = createAdminProtocol(); AddToClusterNodeLabelsRequest request = AddToClusterNodeLabelsRequest.newInstance(labels); adminProtocol.addToClusterNodeLabels(request); } return 0; }
private int addToClusterNodeLabels(String args) throws IOException, YarnException { Set<String> labels = buildNodeLabelsSetFromStr(args); if (directlyAccessNodeLabelStore) { getNodeLabelManagerInstance(getConf()).addToCluserNodeLabels(labels); } else { ResourceManagerAdministrationProtocol adminProtocol = createAdminProtocol(); AddToClusterNodeLabelsRequest request = AddToClusterNodeLabelsRequest.newInstance(labels); adminProtocol.addToClusterNodeLabels(request); } return 0; }
@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); } }