Java - 如何使用此程序更有效地 "scan"我的文件系统的一部分？

Question

我正在为我的防病毒想法开发 Java 概念验证程序的一部分。现在我仍然只是在思考这个想法，细节并不重要，但我希望我正在编写的程序能够获取每个文件在一定范围内的文件路径(例如相隔 5 个级别)目录并将它们写入文本文件。

我现在所拥有的(我将在下面包含我的代码)可以通过检查目录中给定文件夹中是否有文件并将它们的文件路径写入文本文件，然后向下执行，在有限的范围内做到这一点另一个级别，然后再做一次。我目前已将其设置为在目录中执行 2 个级别，并且可以正常工作。但仅当给定级别的目录中只有一项时，它才有效。如果有一个文本文件，它将将该文件路径写入另一个文本文件，然后终止。但是，如果存在文本文件和文件夹，它会忽略该文本文件并进入下一级目录并记录在那里找到的任何文本文件的文件路径。如果有两个或多个文件夹，它总是会选择其中一个而不是其他文件夹。

我现在意识到这是因为我使用了错误的条件。我使用了 if else 并且应该做其他事情，但我不确定我应该使用哪一个。然而我必须这样做，我想修复它，以便它在每个级别都有分支。例如，我启动程序并为其指定起始目录C:/Users/"Name"/Desktop/test/。 Test 有 2 个文件夹和一个文本文件。按照我想要的方式工作，然后它会记录 .txt 的文件路径，向下进入两个文件夹，记录在那里找到的任何 .txt 或其他文件，然后再向下进入它找到的每个文件夹在这两个文件夹中，记录它在那里找到的内容，依此类推，直到完成预定数量的级别。

编辑:为了澄清对问题所在的困惑，我将总结一下。我希望程序将它在目录的每个级别中找到的任何文件的文件路径写入另一个文本文件中。它会执行此操作，但前提是给定级别的目录中有一个文件。例如，如果只有一个 .txt，它将将该 .txt 的文件路径写入另一文本文件。但是，如果该级别目录中有多个文件(例如，两个.txt)，它将只写入其中一个文件的文件路径，而忽略另一个。如果有.txt和文件夹，则忽略.txt并进入文件夹进入下一级目录。我希望它记录给定位置的所有文件，然后分支到同一位置的所有文件夹。

编辑2:我得到了从这个问题获取文件路径的代码部分(Read all files in a folder)以及从这个问题写入我的其他文本文件的部分(How do I create a file and write to it in Java?)

编辑3:如何编辑代码以实现递归，正如 @horatius 指出的那样？

编辑 4:如何编辑我的代码，以便它不需要硬编码的起始文件路径即可工作，而是可以检测可执行文件 .jar 的位置并将其用作其起始目录？

这是我的代码:

public class ScanFolder {

private static final int LEVELS = 5;

private static final String START_DIR = "C:/Users/Joe/Desktop/Test-Level1/";
private static final String REPORT_FILE = "C:/Users/Joe/Desktop/reports.txt";

public static void main(String[] args) throws IOException {
    try (PrintWriter writer = new PrintWriter(REPORT_FILE, "UTF-8");
            Stream<Path> pathStream = Files.walk(Paths.get(START_DIR), LEVELS)) {

         pathStream.filter(Files::isRegularFile).forEach(writer::println);

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

提前致谢

Answer 1

如果您使用Files.walk(...)它会为您完成所有递归操作。

打开并写入PrintWriter每次打开/写入输出文件时都会截断输出文件，只留下最后写入的文件名。

我认为下面的内容就是您所追求的。随着您的进展，您可能希望将找到的 Path 放入文件中，而不是写入文件。对象进入 ArrayList<Path>或类似的以便于以后处理，但从您的问题中不清楚您在这里有什么要求。

public class Walk
{
  public static void main(String[] args) throws IOException {
    try (PrintWriter writer = new PrintWriter("C:/Users/Joe/Desktop/reports.txt", "UTF-8")) {
      Files.walk(Paths.get("C:/Users/Joe/Desktop/test")).forEach(filePath -> {
        if (Files.isRegularFile(filePath)) {
          writer.println(filePath);
        }
      });
    }
  }
}

<小时/>

这是一个改进的示例，您可以使用它来限制深度。它还涉及正确关闭 Stream返回者 Files.walk(...)前面的示例没有，并且更符合流/lambda 惯用语:

public class Walk
{
  // Can use Integer.MAX_VALUE for all
  private static final int LEVELS = 2;

  private static final String START_DIR = "C:/Users/Joe/Desktop/test";
  private static final String REPORT_FILE = "C:/Users/Joe/Desktop/reports.txt";

  public static void main(String[] args) {
    try (PrintWriter writer = new PrintWriter(REPORT_FILE, "UTF-8");
         Stream<Path> pathStream = Files.walk(Paths.get(START_DIR), LEVELS)) {

      pathStream.filter(Files::isRegularFile).forEach(writer::println);

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