|
@@ -127,31 +127,40 @@ public class FileSystemNodeLabelsStore extends NodeLabelsStore {
|
|
@Override
|
|
@Override
|
|
public void updateNodeToLabelsMappings(
|
|
public void updateNodeToLabelsMappings(
|
|
Map<NodeId, Set<String>> nodeToLabels) throws IOException {
|
|
Map<NodeId, Set<String>> nodeToLabels) throws IOException {
|
|
- ensureAppendEditlogFile();
|
|
|
|
- editlogOs.writeInt(SerializedLogType.NODE_TO_LABELS.ordinal());
|
|
|
|
- ((ReplaceLabelsOnNodeRequestPBImpl) ReplaceLabelsOnNodeRequest
|
|
|
|
- .newInstance(nodeToLabels)).getProto().writeDelimitedTo(editlogOs);
|
|
|
|
- ensureCloseEditlogFile();
|
|
|
|
|
|
+ try {
|
|
|
|
+ ensureAppendEditlogFile();
|
|
|
|
+ editlogOs.writeInt(SerializedLogType.NODE_TO_LABELS.ordinal());
|
|
|
|
+ ((ReplaceLabelsOnNodeRequestPBImpl) ReplaceLabelsOnNodeRequest
|
|
|
|
+ .newInstance(nodeToLabels)).getProto().writeDelimitedTo(editlogOs);
|
|
|
|
+ } finally {
|
|
|
|
+ ensureCloseEditlogFile();
|
|
|
|
+ }
|
|
}
|
|
}
|
|
|
|
|
|
@Override
|
|
@Override
|
|
public void storeNewClusterNodeLabels(List<NodeLabel> labels)
|
|
public void storeNewClusterNodeLabels(List<NodeLabel> labels)
|
|
throws IOException {
|
|
throws IOException {
|
|
- ensureAppendEditlogFile();
|
|
|
|
- editlogOs.writeInt(SerializedLogType.ADD_LABELS.ordinal());
|
|
|
|
- ((AddToClusterNodeLabelsRequestPBImpl) AddToClusterNodeLabelsRequest
|
|
|
|
- .newInstance(labels)).getProto().writeDelimitedTo(editlogOs);
|
|
|
|
- ensureCloseEditlogFile();
|
|
|
|
|
|
+ try {
|
|
|
|
+ ensureAppendEditlogFile();
|
|
|
|
+ editlogOs.writeInt(SerializedLogType.ADD_LABELS.ordinal());
|
|
|
|
+ ((AddToClusterNodeLabelsRequestPBImpl) AddToClusterNodeLabelsRequest
|
|
|
|
+ .newInstance(labels)).getProto().writeDelimitedTo(editlogOs);
|
|
|
|
+ } finally {
|
|
|
|
+ ensureCloseEditlogFile();
|
|
|
|
+ }
|
|
}
|
|
}
|
|
|
|
|
|
@Override
|
|
@Override
|
|
public void removeClusterNodeLabels(Collection<String> labels)
|
|
public void removeClusterNodeLabels(Collection<String> labels)
|
|
throws IOException {
|
|
throws IOException {
|
|
- ensureAppendEditlogFile();
|
|
|
|
- editlogOs.writeInt(SerializedLogType.REMOVE_LABELS.ordinal());
|
|
|
|
- ((RemoveFromClusterNodeLabelsRequestPBImpl) RemoveFromClusterNodeLabelsRequest.newInstance(Sets
|
|
|
|
- .newHashSet(labels.iterator()))).getProto().writeDelimitedTo(editlogOs);
|
|
|
|
- ensureCloseEditlogFile();
|
|
|
|
|
|
+ try {
|
|
|
|
+ ensureAppendEditlogFile();
|
|
|
|
+ editlogOs.writeInt(SerializedLogType.REMOVE_LABELS.ordinal());
|
|
|
|
+ ((RemoveFromClusterNodeLabelsRequestPBImpl) RemoveFromClusterNodeLabelsRequest.newInstance(Sets
|
|
|
|
+ .newHashSet(labels.iterator()))).getProto().writeDelimitedTo(editlogOs);
|
|
|
|
+ } finally {
|
|
|
|
+ ensureCloseEditlogFile();
|
|
|
|
+ }
|
|
}
|
|
}
|
|
|
|
|
|
/* (non-Javadoc)
|
|
/* (non-Javadoc)
|