SimpleFileVisitor快速遍历文件夹内容输出到文件

最新推荐文章于 2023-06-02 20:38:40 发布

weixin_33962621

最新推荐文章于 2023-06-02 20:38:40 发布

阅读量513

点赞数

文章标签： java python

原文链接：https://my.oschina.net/TonyTaotao/blog/2979104

版权

2019独角兽企业重金招聘Python工程师标准>>>

有时候，为了快速复制很多文件里的内容合并到一个文件中，则可以采用如下方式：

public class CodeOutput {
    public static void main(String[] args) throws Exception {
        // 需要遍历的文件夹
        String fileDirectory = "E:\\doc";
        // 包含的文件名后缀
        String includeFileNameSuffix = ".java";
        // 输出的文件名
        String outputFilePath = "D:/code.txt";

        Path startingDir = Paths.get(fileDirectory);
        FindFileVisitor findJavaVisitor = new FindFileVisitor(includeFileNameSuffix);
        Files.walkFileTree(startingDir, findJavaVisitor);

        try {

            FileOutputStream fos = new FileOutputStream(outputFilePath);
            FileChannel channel = fos.getChannel();
            findJavaVisitor.getFileNameList().forEach(name -> {
                System.out.println(name);
                if (!name.endsWith("CodeOutput.java")) {
                    Path path = Paths.get(name);
                    try {
                        byte[] bytes = Files.readAllBytes(path);
                        channel.write(ByteBuffer.wrap(bytes));
                    } catch (IOException e) {
                        e.printStackTrace();
                    }
                }

            });

        }  catch (Exception e) {
            e.printStackTrace();
        }

    }


    static class FindFileVisitor extends SimpleFileVisitor<Path> {

        private String fileNameSuffix;
        public FindFileVisitor(String fileNameSuffix) {
            this.fileNameSuffix = fileNameSuffix;
        }

        private List<String> fileNameList = new ArrayList<>();
        public List<String> getFileNameList() {
            return fileNameList;
        }

        @Override
        public FileVisitResult visitFile(Path file, BasicFileAttributes attrs) {
            String fileName = file.toString();
            if (fileName.endsWith(fileNameSuffix)) {
                fileNameList.add(fileName);
            }
            return FileVisitResult.CONTINUE;
        }

        @Override
        public FileVisitResult visitFileFailed(Path file, IOException exc) throws IOException {
            return FileVisitResult.CONTINUE;
        }

    }

}

转载于:https://my.oschina.net/TonyTaotao/blog/2979104