如何获取具有指定扩展名并排除少数目录的文件列表？

Question

我需要从不包括几个子目录的目录中列出文件（指定的扩展名）。我可以使用扩展名进行过滤，但不能使用两个条件 1）文件应按扩展名过滤 2）应排除少数子目录    File dir = new File("D:\\SomeDir");    String[] extensions = new String[] { "txt", "java" };    List<File> files = (List<File>) FileUtils.listFiles(dir, extensions, true);    for (File file : files) {        System.out.println("file: " + file.getCanonicalPath());    }如何排除选定的子目录？

Answer 1

试试这个代码：

// for exlude file extension  

NotFileFilter suffixFileFilterFileFilter=new NotFileFilter(new 

SuffixFileFilter(new String[] { "txt", "java" }));  

//to include only selected extension use below one

//IOFileFilter suffixFileFilterFileFilter= new SuffixFileFilter(new String[] { "java", "txt" });

NotFileFilter directoryFileFilterFileFilter=new NotFileFilter(new 

PrefixFileFilter(new String[] { "tolearn", "dir2" }));   // for directory

File dir = new File("D:\\SomeDir");

List<File> fileList  = (List<File>)  FileUtils.listFiles(dir,suffixFileFilterFileFilter,directoryFileFilterFileFilter);

    for (File file : fileList) {

        System.out.println("file: " + file.getCanonicalPath());

    }

Answer 2

干得好

public static void main(String[] args) throws IOException {

    File dir = new File("C:\\Users\\cosmap\\Documents\\travel-expenses");

    List<String> excludedDirs = Arrays.asList("src", ".git");

    List<String> extensions = Arrays.asList("xml", "yml");

    listFiles(dir, excludedDirs, extensions).forEach(System.out::println);

}

public static List<File> listFiles(File currentFile, List<String> excludedDirs, List<String> extensions) {

    List<File> out = new ArrayList<>();

    File[] files = currentFile.listFiles();

    // might be null

    if(files == null) {

        return out;

    }

    for (File file : files) {

        // add recursively only if is directory and not in the excluded list

        if(file.isDirectory() && !excludedDirs.contains(file.getName())) {

            out.addAll(listFiles(file, excludedDirs, extensions));

        // add if is file and in the allowed extensions list

        } else if (file.isFile() && extensions.contains(file.getName().substring(file.getName().lastIndexOf('.') + 1))) {

            out.add(file);

        }

    }

    return out;

}