lanyuanxiaoyao/hudi
1
0
Fork 0
Code Activity

[MINOR] Ensure directory exists before listing all marker files. (#2594)

Browse Source
This commit is contained in:
Prashant Wason
2021-02-23 08:05:59 -08:00
committed by GitHub
parent 2efd0760ac
commit d2f360f5dd
2 changed files with 7 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
hudi-client/hudi-client-common/src/main/java/org/apache/hudi/table/MarkerFiles.java
View File

@@ -166,10 +166,12 @@ public class MarkerFiles implements Serializable {
public List<String> allMarkerFilePaths() throws IOException { public List<String> allMarkerFilePaths() throws IOException {
List<String> markerFiles = new ArrayList<>(); List<String> markerFiles = new ArrayList<>();
FSUtils.processFiles(fs, markerDirPath.toString(), fileStatus -> { if (doesMarkerDirExist()) {
markerFiles.add(stripMarkerFolderPrefix(fileStatus.getPath().toString())); FSUtils.processFiles(fs, markerDirPath.toString(), fileStatus -> {
return true; markerFiles.add(stripMarkerFolderPrefix(fileStatus.getPath().toString()));
}, false); return true;
}, false);
}
return markerFiles; return markerFiles;
} }

1
hudi-client/hudi-spark-client/src/test/java/org/apache/hudi/table/TestMarkerFiles.java
View File

@@ -119,6 +119,7 @@ public class TestMarkerFiles extends HoodieCommonTestHarness {
public void testDeletionWhenMarkerDirNotExists() throws IOException { public void testDeletionWhenMarkerDirNotExists() throws IOException {
// then // then
assertFalse(markerFiles.doesMarkerDirExist()); assertFalse(markerFiles.doesMarkerDirExist());
assertTrue(markerFiles.allMarkerFilePaths().isEmpty());
assertFalse(markerFiles.deleteMarkerDir(context, 2)); assertFalse(markerFiles.deleteMarkerDir(context, 2));
} }