@Override public void write(OutputStream os, CommonNodeLabelsManager mgr) throws IOException { ((AddToClusterNodeLabelsRequestPBImpl) AddToClusterNodeLabelsRequest .newInstance(labels)).getProto().writeDelimitedTo(os); }
public static AddToClusterNodeLabelsRequest newInstance(Set<String> labels) { AddToClusterNodeLabelsRequest request = Records.newRecord(AddToClusterNodeLabelsRequest.class); request.setNodeLabels(labels); return request; }
@Override public AddToClusterNodeLabelsResponse addToClusterNodeLabels(AddToClusterNodeLabelsRequest request) throws YarnException, IOException { String argName = "addToClusterNodeLabels"; final String msg = "add labels."; UserGroupInformation user = checkAcls(argName); checkRMStatus(user.getShortUserName(), argName, msg); AddToClusterNodeLabelsResponse response = recordFactory.newRecordInstance(AddToClusterNodeLabelsResponse.class); try { rmContext.getNodeLabelManager().addToCluserNodeLabels(request.getNodeLabels()); RMAuditLogger .logSuccess(user.getShortUserName(), argName, "AdminService"); return response; } catch (IOException ioe) { throw logAndWrapException(ioe, user.getShortUserName(), argName, msg); } }
@Override public AddToClusterNodeLabelsResponse addToClusterNodeLabels(AddToClusterNodeLabelsRequest request) throws YarnException, IOException { String argName = "addToClusterNodeLabels"; final String msg = "add labels."; UserGroupInformation user = checkAcls(argName); checkRMStatus(user.getShortUserName(), argName, msg); AddToClusterNodeLabelsResponse response = recordFactory.newRecordInstance(AddToClusterNodeLabelsResponse.class); try { rmContext.getNodeLabelManager().addToCluserNodeLabels(request.getNodeLabels()); RMAuditLogger .logSuccess(user.getShortUserName(), argName, "AdminService"); return response; } catch (IOException ioe) { throw logAndWrapException(ioe, user.getShortUserName(), argName, msg); } }
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; }
public static AddToClusterNodeLabelsRequest newInstance(Set<String> labels) { AddToClusterNodeLabelsRequest request = Records.newRecord(AddToClusterNodeLabelsRequest.class); request.setNodeLabels(labels); return request; }
@Override public AddToClusterNodeLabelsResponse addToClusterNodeLabels(AddToClusterNodeLabelsRequest request) throws YarnException, IOException { final String operation = "addToClusterNodeLabels"; final String msg = "add labels."; UserGroupInformation user = checkAcls(operation); checkRMStatus(user.getShortUserName(), operation, msg); AddToClusterNodeLabelsResponse response = recordFactory.newRecordInstance(AddToClusterNodeLabelsResponse.class); try { rm.getRMContext().getNodeLabelManager() .addToCluserNodeLabels(request.getNodeLabels()); RMAuditLogger.logSuccess(user.getShortUserName(), operation, "AdminService"); return response; } catch (IOException ioe) { throw logAndWrapException(ioe, user.getShortUserName(), operation, msg); } }
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; }
@Public @Unstable public static AddToClusterNodeLabelsRequest newInstance(List<NodeLabel> nodeLabels) { AddToClusterNodeLabelsRequest request = Records .newRecord(AddToClusterNodeLabelsRequest.class); request.setNodeLabels(nodeLabels); return request; }
@Override public void storeNewClusterNodeLabels(Set<String> labels) throws IOException { ensureAppendEditlogFile(); editlogOs.writeInt(SerializedLogType.ADD_LABELS.ordinal()); ((AddToClusterNodeLabelsRequestPBImpl) AddToClusterNodeLabelsRequest.newInstance(labels)).getProto() .writeDelimitedTo(editlogOs); ensureCloseEditlogFile(); }
@Public @Unstable public static AddToClusterNodeLabelsRequest newInstance(List<NodeLabel> nodeLabels) { AddToClusterNodeLabelsRequest request = Records .newRecord(AddToClusterNodeLabelsRequest.class); request.setNodeLabels(nodeLabels); return request; }
@Override public void storeNewClusterNodeLabels(Set<String> labels) throws IOException { ensureAppendEditlogFile(); editlogOs.writeInt(SerializedLogType.ADD_LABELS.ordinal()); ((AddToClusterNodeLabelsRequestPBImpl) AddToClusterNodeLabelsRequest.newInstance(labels)).getProto() .writeDelimitedTo(editlogOs); ensureCloseEditlogFile(); }
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; }
AddToClusterNodeLabelsRequest.newInstance(labels));