无忧无虑

Java找出某一目录下包含某一关键词的所有文件

之前写过FileHelper类，其中的readFile和writeFile方法分别用于文件的读和写。这次在原来的基础上添加了如下方法：

listFiles

用于递归地查找一个目录下的所有文件并输出。

hasWords

用于判断一个文件中是否含有某一关键词。

findFilesContainsWords

递归地查找某一目录下所有包含某一关键字的所有文件（这里我加了一个过滤器，即我只找了所有后缀为“.h”的文件）。

加了新功能后的FileHelper类代码如下：

import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.File;
import java.io.FileReader;
import java.io.FileWriter;



public class FileHelper {
    public static String readFile(String filename) throws Exception {
        BufferedReader reader = new BufferedReader(new FileReader(filename));  
        String ans = "", line = null;
        while((line = reader.readLine()) != null){
            ans += line + "\r\n";
        }
        reader.close();
        return ans;
    }
    public static void writeFile(String content, String filename) throws Exception {
        BufferedWriter writer  = new BufferedWriter(new FileWriter(filename));
        writer.write(content);
        writer.flush();
        writer.close();
    }
    
    public static void listFiles(String path) {
        File file = new File(path);
        File[] files = file.listFiles();
        if(files == null) 
            return;
        for(File f : files) {
            if(f.isFile()) {
                System.out.println(f.toString());
            } else if(f.isDirectory()) {
                System.out.println(f.toString());
                listFiles(f.toString());
            }
        }
    }
    public static boolean hasWords(String file, String words) {
        try {
            String s = readFile(file);
            int w_len = words.length();
            int len = s.length();
            for(int i=0;i+w_len<=len;i++) {
                if(s.substring(i, i+w_len).equals(words))
                    return true;
            }
        } catch (Exception e) {
            e.printStackTrace();
        }
        return false;
    }
    public static void findFilesContainsWords(String path, String words) throws Exception {
        File file = new File(path);
        File[] files = file.listFiles();
        if(files == null) return;
        for(File f : files) {
            if(f.isFile()) {
                if(hasWords(f.toString(), words))
                    System.out.println(f.toString());
            } else if(f.isDirectory()) {
                findFilesContainsWords(f.toString(), words);
            }
        }
    }
    
    public static void main(String[] args) throws Exception {
        findFilesContainsWords("D:\\test", args[0]);//在命令行中找
    }
}