@Override public void write(OutputStream os, CommonNodeLabelsManager mgr) throws IOException { ((ReplaceLabelsOnNodeRequestPBImpl) ReplaceLabelsOnNodeRequest .newInstance(nodeToLabels)).getProto().writeDelimitedTo(os); }
public static ReplaceLabelsOnNodeRequest newInstance( Map<NodeId, Set<String>> map) { ReplaceLabelsOnNodeRequest request = Records.newRecord(ReplaceLabelsOnNodeRequest.class); request.setNodeToLabels(map); return request; }
private int replaceLabelsOnNodes(Map<NodeId, Set<String>> map, boolean failOnUnknownNodes, boolean directlyAccessNodeLabelStore) throws IOException, YarnException { if (directlyAccessNodeLabelStore) { getNodeLabelManagerInstance(getConf()).replaceLabelsOnNode(map); } else { ResourceManagerAdministrationProtocol adminProtocol = createAdminProtocol(); ReplaceLabelsOnNodeRequest request = ReplaceLabelsOnNodeRequest.newInstance(map); request.setFailOnUnknownNodes(failOnUnknownNodes); adminProtocol.replaceLabelsOnNode(request); } return 0; }
recordFactory.newRecordInstance(ReplaceLabelsOnNodeResponse.class); if (request.getFailOnUnknownNodes()) { for (NodeId requestedNode : request.getNodeToLabels().keySet()) { boolean isKnown = false; request.getNodeToLabels()); RMAuditLogger .logSuccess(user.getShortUserName(), operation, "AdminService");
@Override public ReplaceLabelsOnNodeResponse replaceLabelsOnNode( ReplaceLabelsOnNodeRequest request) throws YarnException, IOException { String argName = "replaceLabelsOnNode"; final String msg = "set node to labels."; UserGroupInformation user = checkAcls(argName); checkRMStatus(user.getShortUserName(), argName, msg); ReplaceLabelsOnNodeResponse response = recordFactory.newRecordInstance(ReplaceLabelsOnNodeResponse.class); try { rmContext.getNodeLabelManager().replaceLabelsOnNode( request.getNodeToLabels()); RMAuditLogger .logSuccess(user.getShortUserName(), argName, "AdminService"); return response; } catch (IOException ioe) { throw logAndWrapException(ioe, user.getShortUserName(), argName, msg); } }
@Override public ReplaceLabelsOnNodeResponse replaceLabelsOnNode( ReplaceLabelsOnNodeRequest request) throws YarnException, IOException { String argName = "replaceLabelsOnNode"; final String msg = "set node to labels."; UserGroupInformation user = checkAcls(argName); checkRMStatus(user.getShortUserName(), argName, msg); ReplaceLabelsOnNodeResponse response = recordFactory.newRecordInstance(ReplaceLabelsOnNodeResponse.class); try { rmContext.getNodeLabelManager().replaceLabelsOnNode( request.getNodeToLabels()); RMAuditLogger .logSuccess(user.getShortUserName(), argName, "AdminService"); return response; } catch (IOException ioe) { throw logAndWrapException(ioe, user.getShortUserName(), argName, msg); } }
private int replaceLabelsOnNodes(Map<NodeId, Set<String>> map) throws IOException, YarnException { if (directlyAccessNodeLabelStore) { getNodeLabelManagerInstance(getConf()).replaceLabelsOnNode(map); } else { ResourceManagerAdministrationProtocol adminProtocol = createAdminProtocol(); ReplaceLabelsOnNodeRequest request = ReplaceLabelsOnNodeRequest.newInstance(map); adminProtocol.replaceLabelsOnNode(request); } return 0; }
private int replaceLabelsOnNodes(Map<NodeId, Set<String>> map, boolean failOnUnknownNodes, boolean directlyAccessNodeLabelStore) throws IOException, YarnException { if (directlyAccessNodeLabelStore) { getNodeLabelManagerInstance(getConf()).replaceLabelsOnNode(map); } else { ResourceManagerAdministrationProtocol adminProtocol = createAdminProtocol(); ReplaceLabelsOnNodeRequest request = ReplaceLabelsOnNodeRequest.newInstance(map); request.setFailOnUnknownNodes(failOnUnknownNodes); adminProtocol.replaceLabelsOnNode(request); } return 0; }
public static ReplaceLabelsOnNodeRequest newInstance( Map<NodeId, Set<String>> map) { ReplaceLabelsOnNodeRequest request = Records.newRecord(ReplaceLabelsOnNodeRequest.class); request.setNodeToLabels(map); return request; }
private int replaceLabelsOnNodes(Map<NodeId, Set<String>> map) throws IOException, YarnException { if (directlyAccessNodeLabelStore) { getNodeLabelManagerInstance(getConf()).replaceLabelsOnNode(map); } else { ResourceManagerAdministrationProtocol adminProtocol = createAdminProtocol(); ReplaceLabelsOnNodeRequest request = ReplaceLabelsOnNodeRequest.newInstance(map); adminProtocol.replaceLabelsOnNode(request); } return 0; }
.newInstance(ImmutableMap.of(NodeId.newInstance("host1", 1111), (Set<String>) ImmutableSet.of("x"))); request1.setFailOnUnknownNodes(true); try { rm.adminService.replaceLabelsOnNode(request1); .newInstance(ImmutableMap.of(NodeId.newInstance("host1", 0), (Set<String>) ImmutableSet.of("x"))); request2.setFailOnUnknownNodes(true); try { rm.adminService.replaceLabelsOnNode(request2); .newInstance(ImmutableMap.of(NodeId.newInstance("host5", 0), (Set<String>) ImmutableSet.of("x"))); request3.setFailOnUnknownNodes(true); try { rm.adminService.replaceLabelsOnNode(request3); .newInstance(ImmutableMap.of(NodeId.newInstance("host2", 1111), (Set<String>) ImmutableSet.of("x"))); request4.setFailOnUnknownNodes(true); try { rm.adminService.replaceLabelsOnNode(request4); .newInstance(ImmutableMap.of(NodeId.newInstance("host5", 0), (Set<String>) ImmutableSet.of("x"))); request5.setFailOnUnknownNodes(false); try { rm.adminService.replaceLabelsOnNode(request5);
public static ReplaceLabelsOnNodeRequest newInstance( Map<NodeId, Set<String>> map) { ReplaceLabelsOnNodeRequest request = Records.newRecord(ReplaceLabelsOnNodeRequest.class); request.setNodeToLabels(map); return request; }
@Override public void updateNodeToLabelsMappings( Map<NodeId, Set<String>> nodeToLabels) throws IOException { ensureAppendEditlogFile(); editlogOs.writeInt(SerializedLogType.NODE_TO_LABELS.ordinal()); ((ReplaceLabelsOnNodeRequestPBImpl) ReplaceLabelsOnNodeRequest .newInstance(nodeToLabels)).getProto().writeDelimitedTo(editlogOs); ensureCloseEditlogFile(); }
public static ReplaceLabelsOnNodeRequest newInstance( Map<NodeId, Set<String>> map) { ReplaceLabelsOnNodeRequest request = Records.newRecord(ReplaceLabelsOnNodeRequest.class); request.setNodeToLabels(map); return request; }
@Override public void updateNodeToLabelsMappings( Map<NodeId, Set<String>> nodeToLabels) throws IOException { ensureAppendEditlogFile(); editlogOs.writeInt(SerializedLogType.NODE_TO_LABELS.ordinal()); ((ReplaceLabelsOnNodeRequestPBImpl) ReplaceLabelsOnNodeRequest .newInstance(nodeToLabels)).getProto().writeDelimitedTo(editlogOs); ensureCloseEditlogFile(); }
.newInstance(mgr.getClusterNodeLabels())).getProto().writeDelimitedTo(os); ((ReplaceLabelsOnNodeRequestPBImpl) ReplaceLabelsOnNodeRequest .newInstance(mgr.getNodeLabels())).getProto().writeDelimitedTo(os); os.close();
.newInstance(mgr.getClusterNodeLabels())).getProto().writeDelimitedTo(os); ((ReplaceLabelsOnNodeRequestPBImpl) ReplaceLabelsOnNodeRequest .newInstance(mgr.getNodeLabels())).getProto().writeDelimitedTo(os); os.close();
@Override public void write(OutputStream os, CommonNodeLabelsManager mgr) throws IOException { ((AddToClusterNodeLabelsRequestPBImpl) AddToClusterNodeLabelsRequestPBImpl .newInstance(mgr.getClusterNodeLabels())).getProto() .writeDelimitedTo(os); if (mgr.isCentralizedConfiguration()) { ((ReplaceLabelsOnNodeRequestPBImpl) ReplaceLabelsOnNodeRequest .newInstance(mgr.getNodeLabels())).getProto().writeDelimitedTo(os); } }
@Test public void testModifyLabelsOnNodesWithDistributedConfigurationDisabled() throws IOException, YarnException { // create RM and set it's ACTIVE MockRM rm = new MockRM(); ((RMContextImpl) rm.getRMContext()) .setHAServiceState(HAServiceState.ACTIVE); RMNodeLabelsManager labelMgr = rm.rmContext.getNodeLabelManager(); // by default, distributed configuration for node label is disabled, this // should pass labelMgr.addToCluserNodeLabelsWithDefaultExclusivity(ImmutableSet.of("x", "y")); rm.adminService.replaceLabelsOnNode(ReplaceLabelsOnNodeRequest .newInstance(ImmutableMap.of(NodeId.newInstance("host", 0), (Set<String>) ImmutableSet.of("x")))); rm.close(); }
@Test(expected = YarnException.class) public void testModifyLabelsOnNodesWithCentralizedConfigurationDisabled() throws IOException, YarnException { // create RM and set it's ACTIVE, and set distributed node label // configuration to true MockRM rm = new MockRM(); rm.adminService.isCentralizedNodeLabelConfiguration = false; ((RMContextImpl) rm.getRMContext()) .setHAServiceState(HAServiceState.ACTIVE); RMNodeLabelsManager labelMgr = rm.rmContext.getNodeLabelManager(); // by default, distributed configuration for node label is disabled, this // should pass labelMgr.addToCluserNodeLabelsWithDefaultExclusivity(ImmutableSet.of("x", "y")); rm.adminService.replaceLabelsOnNode(ReplaceLabelsOnNodeRequest .newInstance(ImmutableMap.of(NodeId.newInstance("host", 0), (Set<String>) ImmutableSet.of("x")))); rm.close(); }