2

我想列出一个目录中的所有文件以及该目录中与文件掩码匹配的子目录

例如“M:\SOURCE\*.doc”,而 SOURCE 可能如下所示:

|-- SOURCE
|   |-- Folder1
|   |   |-- File1.doc
|   |   |-- File1.txt
|   |-- File2.doc
|   |-- File3.xml

应该返回 File1.doc 和 File2.doc。

最初,我使用DirectoryStream,因为它已经对掩码/全局语法进行了一些检查,并且能够将其用于过滤,因为这不仅仅是一些正则表达式,而是普通用户发现更容易理解的实际文件掩码

Files.newDirectoryStream(path, mask);

问题是 DirectoryStream 只检查您提供的直接路径目录,而不是它的子目录

然后是Files.walk的“扁平化”方法,实际上能够查看所有子目录,问题是,它不提供通过文件掩码“过滤”的可能性,就像 DirectoryStream 一样

Files.walk(path, Integer.MAX_VALUE);

所以我被困住了,无法在这里结合这两种方法中最好的......

4

3 回答 3

2

您还可以使用自定义FileVisitor[1] 和PathMatcher[2] 的组合,它与 GLOB 完美配合。

代码可能如下所示:

public static void main(String[] args) throws IOException {
    System.out.println(getFiles(Paths.get("/tmp/SOURCE"), "*.doc"));
}

public static List<Path> getFiles(final Path directory, final String glob) throws IOException {
    final var docFileVisitor = new GlobFileVisitor(glob);
    Files.walkFileTree(directory, docFileVisitor);

    return docFileVisitor.getMatchedFiles();
}

public static class GlobFileVisitor extends SimpleFileVisitor<Path> {

    private final PathMatcher pathMatcher;
    private List<Path> matchedFiles = new ArrayList<>();

    public GlobFileVisitor(final String glob) {
        this.pathMatcher = FileSystems.getDefault().getPathMatcher("glob:" + glob);
    }

    @Override
    public FileVisitResult visitFile(Path path, BasicFileAttributes basicFileAttributes) throws IOException {
        if (pathMatcher.matches(path.getFileName())) {
            matchedFiles.add(path);
        }
        return FileVisitResult.CONTINUE;
    }

    public List<Path> getMatchedFiles() {
        return matchedFiles;
    }
}

[1] https://docs.oracle.com/en/java/javase/11/docs/api/java.base/java/nio/file/FileVisitor.html

[2] https://docs.oracle.com/en/java/javase/11/docs/api/java.base/java/nio/file/PathMatcher.html

于 2020-11-13T21:41:40.397 回答
1

PathMatcher我想我可能已经用这里收到的洞察力和其他提到对象的问题解决了我自己的问题

final PathMatcher maskMatcher = FileSystems.getDefault()
                  .getPathMatcher("glob:" + mask);

final List<Path> matchedFiles = Files.walk(path)
                  .collect(Collectors.toList());

final List<Path> filesToRemove = new ArrayList<>(matchedFiles.size());

matchedFiles.forEach(foundPath -> {
            if (!maskMatcher.matches(foundPath.getFileName()) || Files.isDirectory(foundPath)) {
              filesToRemove.add(foundPath);
            }
          });

 matchedFiles.removeAll(filesToRemove);

所以基本上与DirectoryStream过滤文件所做.getPathMatcher("glob:" + mask);的事情是一样的

之后我现在要做的就是通过删除与我的 PathMatcher 不匹配且不是 File 类型的元素来过滤我使用Files.walk获得的路径列表

于 2020-11-13T22:29:14.917 回答
0

可以使用 common StreamfilterFiles.walk使用String::matches适当的正则表达式中检索过滤的文件名:

final String SOURCE_DIR = "test";

Files.walk(Paths.get(SOURCE_DIR));
     .filter(p -> p.getFileName().toString().matches(".*\\.docx?"))
     .forEach(System.out::println);

输出

test\level01\level11\test.doc
test\level02\test-level2.doc
test\t1.doc
test\t3.docx

输入目录结构:

│   t1.doc
│   t2.txt
│   t3.docx
│   t4.bin
│
├───level01
│   │   test.do
│   │
│   └───level11
│           test.doc
│
└───level02
        test-level2.doc

更新

可以使用递归解决方案,newDirectoryStream但需要将其转换为 Stream:

static Stream<Path> readFilesByMaskRecursively(Path start, String mask) {
        
    List<Stream<Path>> sub = new ArrayList<>();
        
    try {
        sub.add(StreamSupport.stream( // read files by mask in current dir
                Files.newDirectoryStream(start, mask).spliterator(), false));
            
        Files.newDirectoryStream(start, (path) -> path.toFile().isDirectory())
             .forEach(path -> sub.add(recursive(path, mask)));
    } catch (IOException ioex) {
        ioex.printStackTrace();
    }
        
    return sub.stream().flatMap(s -> s); // convert to Stream<Path>
}

// test
readFilesByMaskRecursively(Paths.get(SOURCE_DIR), "*.doc*")
             .forEach(System.out::println);

输出:

test\t1.doc
test\t3.docx
test\level01\level11\test.doc
test\level02\test-level2.doc

更新 2

可以将前缀**/添加到PathMatcher跨目录边界,然后Files.walk基于 - 的解决方案可以使用简化过滤器,而无需删除特定条目:

String mask = "*.doc*";
PathMatcher maskMatcher = FileSystems.getDefault().getPathMatcher("glob:**/" + mask);
Files.walk(Paths.get(SOURCE_DIR))
     .filter(path -> maskMatcher.matches(path))
     .forEach(System.out::println);

输出(与递归解决方案相同):

test\level01\level11\test.doc
test\level02\test-level2.doc
test\t1.doc
test\t3.docx
于 2020-11-13T21:22:59.440 回答