@Override public void replaceLabelsOnNode(Map<NodeId, Set<String>> replaceLabelsToNode) throws IOException { try { writeLock.lock(); // get nodesCollection before edition Map<String, Host> before = cloneNodeMap(replaceLabelsToNode.keySet()); super.replaceLabelsOnNode(replaceLabelsToNode); // get nodesCollection after edition Map<String, Host> after = cloneNodeMap(replaceLabelsToNode.keySet()); // update running nodes resources updateResourceMappings(before, after); } finally { writeLock.unlock(); } }
@Override public void replaceLabelsOnNode(Map<NodeId, Set<String>> replaceLabelsToNode) throws IOException { try { writeLock.lock(); // get nodesCollection before edition Map<String, Host> before = cloneNodeMap(replaceLabelsToNode.keySet()); super.replaceLabelsOnNode(replaceLabelsToNode); // get nodesCollection after edition Map<String, Host> after = cloneNodeMap(replaceLabelsToNode.keySet()); // update running nodes resources updateResourceMappings(before, after); } finally { writeLock.unlock(); } }
@Override public void replaceLabelsOnNode(Map<NodeId, Set<String>> replaceLabelsToNode) throws IOException { try { writeLock.lock(); Map<NodeId, Set<String>> effectiveModifiedLabelMappings = getModifiedNodeLabelsMappings(replaceLabelsToNode); if(effectiveModifiedLabelMappings.isEmpty()) { LOG.info("No Modified Node label Mapping to replace"); return; } // get nodesCollection before edition Map<String, Host> before = cloneNodeMap(effectiveModifiedLabelMappings.keySet()); super.replaceLabelsOnNode(effectiveModifiedLabelMappings); // get nodesCollection after edition Map<String, Host> after = cloneNodeMap(effectiveModifiedLabelMappings.keySet()); // update running nodes resources updateResourceMappings(before, after); } finally { writeLock.unlock(); } }
@Override public void recover(InputStream is, CommonNodeLabelsManager mgr) throws IOException { nodeToLabels = new ReplaceLabelsOnNodeRequestPBImpl( YarnServerResourceManagerServiceProtos.ReplaceLabelsOnNodeRequestProto .parseDelimitedFrom(is)).getNodeToLabels(); if (mgr.isCentralizedConfiguration()) { mgr.replaceLabelsOnNode(nodeToLabels); } }
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) throws IOException, YarnException { if (directlyAccessNodeLabelStore) { getNodeLabelManagerInstance(getConf()).replaceLabelsOnNode(map); } else { ResourceManagerAdministrationProtocol adminProtocol = createAdminProtocol(); ReplaceLabelsOnNodeRequest request = ReplaceLabelsOnNodeRequest.newInstance(map); adminProtocol.replaceLabelsOnNode(request); } return 0; }
.getNodeToLabels(); mgr.addToCluserNodeLabels(labels); mgr.replaceLabelsOnNode(nodeToLabels); is.close(); ReplaceLabelsOnNodeRequestProto.parseDelimitedFrom(is)) .getNodeToLabels(); mgr.replaceLabelsOnNode(map); break;
.getNodeToLabels(); mgr.addToCluserNodeLabels(labels); mgr.replaceLabelsOnNode(nodeToLabels); is.close(); ReplaceLabelsOnNodeRequestProto.parseDelimitedFrom(is)) .getNodeToLabels(); mgr.replaceLabelsOnNode(map); break;
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; }
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; }
@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); } }