搜索目录下所有文件内容是否包含指定的关键词

搜索目录下所有文件内容是否包含指定的关键词

import java.util.*;
import java.io.*;
import java.util.concurrent.*;
/**
 * Created by QYT on 2017/8/2.
 */
public class BlockingQueueTest {
    private static final int FILE_QUEUE_SIZE=10;
    private static final int SEARCH_THREADS=100;
    private static final File DUMMY=new File("");
    private static BlockingQueue<File> queue=new ArrayBlockingQueue<>(FILE_QUEUE_SIZE);

    public static void main(String[] args) {
        try(Scanner in=new Scanner(System.in)){
            System.out.print("Enter base directory(e.g. /opt/jdk1.8.0/src): ");
            String directory=in.nextLine();
            System.out.print("Enter keyword(e.g. volatile): ");
            String keyword=in.nextLine();
            Runnable enumerator=()->{
                try{
                    enumerate(new File(directory));
                    queue.put(DUMMY);
                }catch (InterruptedException e){}

            };
            new Thread(enumerator).start();
            for (int i = 0; i < SEARCH_THREADS; i++) {
                Runnable searcher=()->{
                  try {
                      boolean done=false;
                      while (!done){
                          File file=queue.take();
                          if(file == DUMMY){
                              queue.put(file);
                              done =true;
                          }else {
                              search(file,keyword);
                          }
                      }
                  }catch (Exception e){}
                };
                new Thread(searcher).start();
            }
        }
    }

    public static void enumerate(File directory) throws InterruptedException {
        File[] files=directory.listFiles();
        for(File f:files){
            if(f.isDirectory())enumerate(f);
            else queue.put(f);
        }
    }

    public static void search(File file,String keyword) throws FileNotFoundException {
        try(Scanner in=new Scanner(file,"UTF-8") ){
            int lineNumber=0;
            while (in.hasNextLine()){
                lineNumber++;
                String line=in.nextLine();
                if(line.contains(keyword))
                    System.out.printf("%s:%d:%s%n",file.getPath(),lineNumber,line);
            }
        }
    }
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值