文件读取和操作类
import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream; public class Test3 { private String keyWords; public Test3(String keyWords){
this.keyWords = keyWords;
} //计算文件数量
public static int count = 0; //搜索后查询到的文件路径汇总文件地址
public static String searchedFilePath = "G:/Document/HongDaXingYe/SearchedDir/searchedFile.txt"; public static File searchedFile = null; public static FileOutputStream fos = null; public void search(){
String path = "G:/Document/HongDaXingYe/Project/oa/workflow/home/weaver/ecology/workflow/";
File file = new File(path);
File[] files = file.listFiles();
getFiles(files);
System.out.println("count:" + count);
} //递归搜索文件并写入搜索到的路径到文件
public void getFiles(File[] files){
FileInputStream fis = null;
try{
for(File file : files){
count++;
if(file.isDirectory()){
getFiles(file.listFiles());
}else{
StringBuffer sb = new StringBuffer();
byte[] bytes = new byte[1024];
fis = new FileInputStream(file);
int len = 0;
while((len = fis.read(bytes)) != -1){
sb.append(new String(bytes, 0, len));
}
fis.close();
if(sb.indexOf(keyWords) >= 0){
System.out.println("包含关键字(" + keyWords + ")的文件路径:" + file.getAbsolutePath());
fos.write(("包含关键字(" + keyWords + ")的文件路径:" + file.getAbsolutePath() + System.lineSeparator()).getBytes());
fos.flush();
}
}
}
}catch(Exception e){
e.printStackTrace();
}
} }
多线程处理类
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.util.Scanner;
import java.util.concurrent.LinkedBlockingQueue;
import java.util.concurrent.ThreadPoolExecutor;
import java.util.concurrent.TimeUnit; public class TestThread2 implements Runnable{ private int taskNum;
private String keyWords; public TestThread2(int taskNum, String keyWords){
this.taskNum = taskNum;
this.keyWords = keyWords;
} public static void main(String[] args){
String searchedFilePath = "G:/Document/HongDaXingYe/SearchedDir/searchedFile.txt";
File searchedFile = new File(searchedFilePath);
FileOutputStream fos = null;
try{
fos = new FileOutputStream(searchedFile);
if(!searchedFile.exists()){
searchedFile.createNewFile();
}
Test3.fos = fos;
}catch(Exception e){
e.printStackTrace();
}
Scanner input = new Scanner(System.in);
System.out.println("请输入需要搜索的关键字(如果有多个,用英文半角逗号隔开):");
String keyWordsStr = input.next();
String[] keyWordsArr = null;
if(keyWordsStr.indexOf(",") > 0){
keyWordsArr = keyWordsStr.split(",");
}else{
keyWordsArr = new String[]{ keyWordsStr };
}
input.close();
LinkedBlockingQueue<Runnable> workQueue = new LinkedBlockingQueue<Runnable>();
ThreadPoolExecutor executor = new ThreadPoolExecutor(3, 100, 3600, TimeUnit.SECONDS, workQueue);
for(int i = 0; i < keyWordsArr.length; i++){
String keyWords = keyWordsArr[i];
TestThread testThread = new TestThread(i, keyWords);
executor.execute(testThread);
}
if(executor.isTerminated() && Test3.fos != null){
try {
Test3.fos.close();
} catch (IOException e) {
e.printStackTrace();
}
}
} @Override
public void run() {
System.out.println("正在执行task " + taskNum);
System.out.println("当前关键字:" + keyWords);
Test3 test3 = new Test3(keyWords);
test3.search();
System.out.println("task " + taskNum + "执行完毕");
} }