Java程序——检索文件（含内容）

项目说明

给定一个指定目录和关键字，扫描其中的文件名和文件内容，找到包含关键字的文件

完整代码

import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.util.Scanner;

public class demo4 {
    public static void main(String[] args) throws IOException {
        Scanner scanner = new Scanner(System.in);
        System.out.println("请输入要扫描的路径");
        File rootDir = new File(scanner.next());
        System.out.println("请输入要查询的词");
        String toFind = scanner.next();

        scanDir(rootDir,toFind);
    }

    private static void scanDir(File rootDir, String toFind) throws IOException {
        File[] files = rootDir.listFiles();
        if(files == null){
            return;
        }
        for (File f:files) {
            if(f.isDirectory()){
                scanDir(f,toFind);
            } else checkFile(f,toFind);
        }
    }

    private static void checkFile(File f, String toFind) throws IOException {
        if(f.getName().contains(toFind)){
            System.out.println("该文件名中包含关键词：" + f.getCanonicalPath());
        }
        try(InputStream inputStream = new FileInputStream(f)){
            Scanner scanner = new Scanner(inputStream);
            StringBuilder stringBuilder = new StringBuilder();
            while(scanner.hasNextLine()){
                stringBuilder.append(scanner.nextLine() + "\n");
            }
            if(stringBuilder.indexOf(toFind) > -1){
                System.out.println("该文件内容包含关键字: " + f.getCanonicalPath());
            }
        }
    }
}

核心代码

private static void scanDir(File rootDir, String toFind) throws IOException {
    File[] files = rootDir.listFiles();
    if(files == null){
        return;
    }
    for (File f:files) {
        if(f.isDirectory()){
            scanDir(f,toFind);
        } else checkFile(f,toFind);
    }
}

传入一个文件对象和关键字字符串。通过调用listFiles方法，如果文件对象是空，代表该路径下没有任何文件或目录，那么就返回上级目录，否则就将所有该路径下的文件或目录都存储在File数组中，并依次便利，如果是目录，就再次递归，找该目录下的文件，如果是文件，就比较文件名和文件内容中有没有包含关键字

private static void checkFile(File f, String toFind) throws IOException {
    if(f.getName().contains(toFind)){
        System.out.println("该文件名中包含关键词：" + f.getCanonicalPath());
    }
    try(InputStream inputStream = new FileInputStream(f)){
        Scanner scanner = new Scanner(inputStream);
        StringBuilder stringBuilder = new StringBuilder();
        while(scanner.hasNextLine()){
            stringBuilder.append(scanner.nextLine() + "\n");
        }
        if(stringBuilder.indexOf(toFind) > -1){
            System.out.println("该文件内容包含关键字: " + f.getCanonicalPath());
        }
    }
}

先比较文件名中有没有包含关键字，再通过FileInputStream读取文件中全部内容，将所有的内容写成一个字符串，并比较字符串中是否包含关键字