Java使用阻塞队列实现指定文件夹下面搜索指定关键字

import java.io.File;
import java.io.FileNotFoundException;
import java.util.Scanner;
import java.util.concurrent.ArrayBlockingQueue;
import java.util.concurrent.BlockingQueue;

public class BlockingQueueTest {

	public static void main(String[] args) {
		// TODO Auto-generated method stub
		final int File_QUEUE_SIZE = 10;
		final int SEARCH_THREADS = 100;
		BlockingQueue<File> queue = new ArrayBlockingQueue<>(File_QUEUE_SIZE);
		File startingDirectory = new File("F:\\");
		String keyWords = "java";
		new Thread(new FileEnumerationTask(queue, startingDirectory)).start();
		for(int i=0; i<SEARCH_THREADS; ++i){
			new Thread(new SearchTask(queue, keyWords)).start();
		}
	}

}

/**
 * 将文件添加到队列中
 * @author Administrator
 *
 */
class FileEnumerationTask implements Runnable{
	
	public static File DUMMY = new File("");
	
	private BlockingQueue<File> queue;
	
	private File startingDirectory;
	
	public FileEnumerationTask(BlockingQueue<File> queue, File startingDirectory) {
		super();
		this.queue = queue;
		this.startingDirectory = startingDirectory;
	}
	
	@Override
	public void run() {
		// TODO Auto-generated method stub
		try {
			enumerate(startingDirectory);
			queue.put(DUMMY);
		} catch (InterruptedException e) {
			// TODO Auto-generated catch block
			e.printStackTrace();
		}
		
	}
	
	public void enumerate(File directory) throws InterruptedException{
		
		File[] files = directory.listFiles();
		
		for(File f : files){
			if(f.isDirectory()){
				enumerate(f);
			}else{
				queue.put(f);
			}
		}
	}

}

/**
 * 启动多线程搜索关键字
 * @author Administrator
 *
 */
class SearchTask implements Runnable{
	
	private BlockingQueue<File> queue;
	
	private String keyWords;
	
	public SearchTask(BlockingQueue<File> queue, String keyWords) {
		super();
		this.queue = queue;
		this.keyWords = keyWords;
	}

	@Override
	public void run() {
		// TODO Auto-generated method stub
		boolean isDone = false;
		while(!isDone){
			try {
				File f = queue.take();
				if(f==FileEnumerationTask.DUMMY){
					queue.put(f);
					isDone = true;
				}else{
					search(f);
				}
			} catch (InterruptedException e) {
				// TODO Auto-generated catch block
				e.printStackTrace();
			} catch (FileNotFoundException e) {
				// TODO Auto-generated catch block
				e.printStackTrace();
			}
			
		}
		
	}
	
	public void search(File f) throws FileNotFoundException{
		
		try(Scanner in = new Scanner(f)){
			int lineNumber = 0;
			while(in.hasNextLine()){
				++lineNumber;
				String line = in.nextLine();
				if(line.contains(keyWords)){
					System.out.println(f.getPath() + " " + lineNumber + " " + line);
				}
			}
		}
	}
}