@Override protected void serviceStop() throws Exception { // finalize store stopDispatcher(); // only close store when we enabled store persistent if (null != store) { store.close(); } }
protected void handleStoreEvent(NodeLabelsStoreEvent event) { try { switch (event.getType()) { case ADD_LABELS: StoreNewClusterNodeLabels storeNewClusterNodeLabelsEvent = (StoreNewClusterNodeLabels) event; store.storeNewClusterNodeLabels(storeNewClusterNodeLabelsEvent .getLabels()); break; case REMOVE_LABELS: RemoveClusterNodeLabels removeClusterNodeLabelsEvent = (RemoveClusterNodeLabels) event; store.removeClusterNodeLabels(removeClusterNodeLabelsEvent.getLabels()); break; case STORE_NODE_TO_LABELS: UpdateNodeToLabelsMappingsEvent updateNodeToLabelsMappingsEvent = (UpdateNodeToLabelsMappingsEvent) event; store.updateNodeToLabelsMappings(updateNodeToLabelsMappingsEvent .getNodeToLabels()); break; } } catch (IOException e) { LOG.error("Failed to store label modification to storage"); throw new YarnRuntimeException(e); } }
@Override public void initNodeLabelStore(Configuration conf) { this.store = new NodeLabelsStore() { @Override public void recover() throws IOException { } @Override public void removeClusterNodeLabels(Collection<String> labels) throws IOException { lastRemovedlabels = labels; } @Override public void updateNodeToLabelsMappings( Map<NodeId, Set<String>> nodeToLabels) throws IOException { lastNodeToLabels = nodeToLabels; } @Override public void storeNewClusterNodeLabels(List<NodeLabel> label) throws IOException { lastAddedlabels = label; } @Override public void close() throws IOException { // do nothing } }; this.store.setNodeLabelsManager(this); }
protected void handleStoreEvent(NodeLabelsStoreEvent event) { try { switch (event.getType()) { case ADD_LABELS: StoreNewClusterNodeLabels storeNewClusterNodeLabelsEvent = (StoreNewClusterNodeLabels) event; store.storeNewClusterNodeLabels(storeNewClusterNodeLabelsEvent .getLabels()); break; case REMOVE_LABELS: RemoveClusterNodeLabels removeClusterNodeLabelsEvent = (RemoveClusterNodeLabels) event; store.removeClusterNodeLabels(removeClusterNodeLabelsEvent.getLabels()); break; case STORE_NODE_TO_LABELS: UpdateNodeToLabelsMappingsEvent updateNodeToLabelsMappingsEvent = (UpdateNodeToLabelsMappingsEvent) event; store.updateNodeToLabelsMappings(updateNodeToLabelsMappingsEvent .getNodeToLabels()); break; } } catch (IOException e) { LOG.error("Failed to store label modification to storage"); throw new YarnRuntimeException(e); } }
protected void handleStoreEvent(NodeLabelsStoreEvent event) { try { switch (event.getType()) { case ADD_LABELS: StoreNewClusterNodeLabels storeNewClusterNodeLabelsEvent = (StoreNewClusterNodeLabels) event; store.storeNewClusterNodeLabels(storeNewClusterNodeLabelsEvent .getLabels()); break; case REMOVE_LABELS: RemoveClusterNodeLabels removeClusterNodeLabelsEvent = (RemoveClusterNodeLabels) event; store.removeClusterNodeLabels(removeClusterNodeLabelsEvent.getLabels()); break; case STORE_NODE_TO_LABELS: UpdateNodeToLabelsMappingsEvent updateNodeToLabelsMappingsEvent = (UpdateNodeToLabelsMappingsEvent) event; store.updateNodeToLabelsMappings(updateNodeToLabelsMappingsEvent .getNodeToLabels()); break; } } catch (IOException e) { LOG.error("Failed to store label modification to storage"); throw new YarnRuntimeException(e); } }
@Override protected void serviceStop() throws Exception { // finalize store stopDispatcher(); // only close store when we enabled store persistent if (null != store) { store.close(); } }
@Override protected void serviceStop() throws Exception { // finalize store stopDispatcher(); // only close store when we enabled store persistent if (null != store) { store.close(); } }