@Override public void write(OutputStream os, CommonNodeLabelsManager mgr) throws IOException { ((RemoveFromClusterNodeLabelsRequestPBImpl) RemoveFromClusterNodeLabelsRequest .newInstance(Sets.newHashSet(labels.iterator()))).getProto() .writeDelimitedTo(os); }
@Override public void removeClusterNodeLabels(Collection<String> labels) throws IOException { ensureAppendEditlogFile(); editlogOs.writeInt(SerializedLogType.REMOVE_LABELS.ordinal()); ((RemoveFromClusterNodeLabelsRequestPBImpl) RemoveFromClusterNodeLabelsRequest.newInstance(Sets .newHashSet(labels.iterator()))).getProto().writeDelimitedTo(editlogOs); ensureCloseEditlogFile(); }
@Override public void removeClusterNodeLabels(Collection<String> labels) throws IOException { ensureAppendEditlogFile(); editlogOs.writeInt(SerializedLogType.REMOVE_LABELS.ordinal()); ((RemoveFromClusterNodeLabelsRequestPBImpl) RemoveFromClusterNodeLabelsRequest.newInstance(Sets .newHashSet(labels.iterator()))).getProto().writeDelimitedTo(editlogOs); ensureCloseEditlogFile(); }
createAdminProtocol(); RemoveFromClusterNodeLabelsRequest request = RemoveFromClusterNodeLabelsRequest.newInstance(labels); adminProtocol.removeFromClusterNodeLabels(request);
createAdminProtocol(); RemoveFromClusterNodeLabelsRequest request = RemoveFromClusterNodeLabelsRequest.newInstance(labels); adminProtocol.removeFromClusterNodeLabels(request);
private int removeFromClusterNodeLabels(String args) throws IOException, YarnException { Set<String> labels = buildNodeLabelsSetFromStr(args); if (directlyAccessNodeLabelStore) { getNodeLabelManagerInstance(getConf()).removeFromClusterNodeLabels( labels); } else { ResourceManagerAdministrationProtocol adminProtocol = createAdminProtocol(); RemoveFromClusterNodeLabelsRequest request = RemoveFromClusterNodeLabelsRequest.newInstance(labels); adminProtocol.removeFromClusterNodeLabels(request); } return 0; }
private int removeFromClusterNodeLabels(String args) throws IOException, YarnException { Set<String> labels = buildNodeLabelsSetFromStr(args); if (directlyAccessNodeLabelStore) { getNodeLabelManagerInstance(getConf()).removeFromClusterNodeLabels( labels); } else { ResourceManagerAdministrationProtocol adminProtocol = createAdminProtocol(); RemoveFromClusterNodeLabelsRequest request = RemoveFromClusterNodeLabelsRequest.newInstance(labels); adminProtocol.removeFromClusterNodeLabels(request); } return 0; }
@Test public void testRemoveClusterNodeLabelsWithCentralizedConfigurationDisabled() 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(); rm.adminService.isCentralizedNodeLabelConfiguration = false; // by default, distributed configuration for node label is disabled, this // should pass labelMgr.addToCluserNodeLabelsWithDefaultExclusivity(ImmutableSet.of("x", "y")); rm.adminService .removeFromClusterNodeLabels(RemoveFromClusterNodeLabelsRequest .newInstance((Set<String>) ImmutableSet.of("x"))); Set<String> clusterNodeLabels = labelMgr.getClusterNodeLabelNames(); assertEquals(1,clusterNodeLabels.size()); rm.close(); }