big-data/hadoop
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

YARN-3930. FileSystemNodeLabelsStore should make sure edit log file closed when exception is thrown. (Dian Fu via wangda)

Browse Source
This commit is contained in:
Wangda Tan 2015-07-16 16:06:20 -07:00
parent d96bbe152c
commit fa2b63ed16
2 changed files with 27 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
hadoop-yarn-project/CHANGES.txt
View File

@ -637,6 +637,9 @@ Release 2.8.0 - UNRELEASED
YARN-3805. Update the documentation of Disk Checker based on YARN-90. YARN-3805. Update the documentation of Disk Checker based on YARN-90.
(Masatake Iwasaki via ozawa) (Masatake Iwasaki via ozawa)
YARN-3930. FileSystemNodeLabelsStore should make sure edit log file closed when
exception is thrown. (Dian Fu via wangda)
Release 2.7.2 - UNRELEASED Release 2.7.2 - UNRELEASED
INCOMPATIBLE CHANGES INCOMPATIBLE CHANGES

39
hadoop-yarn-project/hadoop-yarn/hadoop-yarn-common/src/main/java/org/apache/hadoop/yarn/nodelabels/FileSystemNodeLabelsStore.java
View File

@ -127,31 +127,40 @@ private void ensureCloseEditlogFile() throws IOException {
@Override @Override
public void updateNodeToLabelsMappings( public void updateNodeToLabelsMappings(
Map<NodeId, Set<String>> nodeToLabels) throws IOException { Map<NodeId, Set<String>> nodeToLabels) throws IOException {
ensureAppendEditlogFile(); try {
editlogOs.writeInt(SerializedLogType.NODE_TO_LABELS.ordinal()); ensureAppendEditlogFile();
((ReplaceLabelsOnNodeRequestPBImpl) ReplaceLabelsOnNodeRequest editlogOs.writeInt(SerializedLogType.NODE_TO_LABELS.ordinal());
.newInstance(nodeToLabels)).getProto().writeDelimitedTo(editlogOs); ((ReplaceLabelsOnNodeRequestPBImpl) ReplaceLabelsOnNodeRequest
ensureCloseEditlogFile(); .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(); try {
editlogOs.writeInt(SerializedLogType.ADD_LABELS.ordinal()); ensureAppendEditlogFile();
((AddToClusterNodeLabelsRequestPBImpl) AddToClusterNodeLabelsRequest editlogOs.writeInt(SerializedLogType.ADD_LABELS.ordinal());
.newInstance(labels)).getProto().writeDelimitedTo(editlogOs); ((AddToClusterNodeLabelsRequestPBImpl) AddToClusterNodeLabelsRequest
ensureCloseEditlogFile(); .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(); try {
editlogOs.writeInt(SerializedLogType.REMOVE_LABELS.ordinal()); ensureAppendEditlogFile();
((RemoveFromClusterNodeLabelsRequestPBImpl) RemoveFromClusterNodeLabelsRequest.newInstance(Sets editlogOs.writeInt(SerializedLogType.REMOVE_LABELS.ordinal());
.newHashSet(labels.iterator()))).getProto().writeDelimitedTo(editlogOs); ((RemoveFromClusterNodeLabelsRequestPBImpl) RemoveFromClusterNodeLabelsRequest.newInstance(Sets
ensureCloseEditlogFile(); .newHashSet(labels.iterator()))).getProto().writeDelimitedTo(editlogOs);
} finally {
ensureCloseEditlogFile();
}
} }
/* (non-Javadoc) /* (non-Javadoc)