项目说明
给定一个指定目录和关键字,扫描其中的文件名和文件内容,找到包含关键字的文件
完整代码
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读取文件中全部内容,将所有的内容写成一个字符串,并比较字符串中是否包含关键字