linecounter.java

来自「Java学习笔记」· Java 代码 · 共 51 行

package day16;

import java.io.BufferedReader;
import java.io.File;
import java.io.FileFilter;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.IOException;
import java.io.InputStreamReader;

public class LineCounter {
    private File directory;
    private int sumLine = 0;
    public LineCounter(File directory) {
        super();
        this.directory = directory;
    }
    public int getSumLine(String strFileter) {
        sumLine = lineNumberinDirectory(strFileter);
        return sumLine;
    }
    private int lineNumberinDirectory(final String strFilter) {
        int lineNumber = 0;
        File[] filesFiltered = directory.listFiles(new FileFilter() {
            public boolean accept(File pathname) {
                if (pathname.isFile() && pathname.getName().indexOf(strFilter) != -1)
                    return true;
                return false;
            }
        });
        for (File file : filesFiltered)
            lineNumber += lineNumberinFile(file);
        return lineNumber;
    }
    private int lineNumberinFile(File file) {
        BufferedReader br = null;
        int lineNumber = 1;
        try {
            br = new BufferedReader(new InputStreamReader(new FileInputStream(file)));
            while (br.readLine() != null)
                lineNumber++;
        } catch (FileNotFoundException e) {
            System.out.println(e.getMessage());
            return 0;
        } catch (IOException e) {
            System.out.println(e.getMessage());
            return 0;
        }
        return lineNumber;
    }
}