import java.io.*;
import java.util.*;
import
java.util.concurrent.*;
public class SearchFile
{
public static void main(String ... strings)
{
final int FILE_Queue_SIZE
= 10;
final int SEARCH_THREADS =
100;
Scanner in = new
Scanner(System.in);
System.out.println("enter your base directory:");
String directory = in.nextLine();
System.out.println("enter your
keyword");
String keyword =
in.nextLine();
ExecutorService pool=
Executors.newCachedThreadPool();
MatchCounter counter = new MatchCounter(new
File(directory),keyword,pool);
Future<Integer> result = pool.submit(counter);
try
{
System.out.println("result.get()=" + result.get() + "matching
files.");
}
catch (Exception e)
{
e.printStackTrace();
}
pool.shutdown();
int
largestPoolSize =
((ThreadPoolExecutor)pool).getLargestPoolSize();
System.out.println("largestPoolSize=" +
largestPoolSize);
}
}
class MatchCounter
implements Callable<Integer>
{
private File
directory;
private String
keyword;
private ExecutorService
pool;
private int count;
public MatchCounter(File directory, String keyword, ExecutorService
pool)
{
this.directory = directory;
this.keyword = keyword;
this.pool = pool;
}
public Integer call()
{
count = 0;
try
{
File[] files =
directory.listFiles();
ArrayList<Future<Integer>> results =
new ArrayList<Future<Integer>>();
for(File
file:files)
{
if(file.isDirectory())
{
MatchCounter counter = new
MatchCounter(file,keyword,pool);;
Future<Integer> result = pool.submit(counter);
results.add(result);
}else
{
if
(search(file))
{
count++;
System.out.println("路径：" +
file.getAbsolutePath());
}
}
}
for(Future<Integer>
result:results)
{
try
{
count += result.get();
System.out.println("count=" + count + "-----" +
"result.get()=" + result.get());
}
catch
(Exception e)
{
e.printStackTrace();
}
}
}catch (Exception e)
{
e.printStackTrace();
}
return
count;
}
public boolean
search(File file)
{
try
{
Scanner in = new Scanner(new FileInputStream(file));
boolean found =
false;
while(!found && in.hasNextLine())
{
String line
= in.nextLine();
if(line.contains(keyword)) found =
true;
}
in.close();
return found;
}
catch (Exception
e)
{
return false;
}
}
}
从指定的路径中查找含有特殊字符串的文件,码迷,mamicode.com