代码行数统计器

import java.io.BufferedReader;
import java.io.File;
import java.io.FileReader;
import java.io.IOException;


/**
 * @Title 代码行数统计器
 */
public class CodeLinesStatistic {

	private static long sums = 0;


	private static String suffixs[];

	private static String target;

	/** buffer flush 的最大值 * */
	private static int FLUSH_FLAG = 1024 * 64;
	private static final String PROJECT_PATH="D:/eclipseProject/src/main";
	private static StringBuffer statistics = new StringBuffer();

	public static void main(String... args) throws IOException {

		// 这里模拟命令行下的参数进行测试
		args = new String[] {PROJECT_PATH, // 这里是项目的根目录
				"java", "xml", "properties","jsp" }; // 这里是统计文件的后缀名
		long startTimes = System.currentTimeMillis();
		if (args.length > 1)
			suffixs = new String[args.length - 1];
		else {
			System.out.println("As that : targetLocation , fileSuffix , fileSuffix . . .");
			return;
		}

		for (int i = 0; i < args.length; i++) {
			if (i == 0) {
				target = args[i];
			} else {
				suffixs[i - 1] = args[i];
			}
		}

		File targetFile = new File(target);
		if (targetFile.exists()) {
			statistic(targetFile);
			System.out.print(statistics.toString());
			System.out.println("该项目中总共的代码行数：" + sums);
		} else {
			System.out.println("文件或者目录不存在: " + target);
		}
		System.out.println("总花费时间： "+ (System.currentTimeMillis() - startTimes) + " ms");
	}

	/**
	 * 深度优先,统计文件行数
	 * @param file
	 * @throws IOException 
	 * @Author Aaron
	 */
	private static void statistic(File file) throws IOException {

		if (file.isDirectory()) {
			File[] files = file.listFiles();
			for (int i = 0; i < files.length; i++) {
				statistic(files[i]);
			}
		}
		if (file.isFile()) {
			if (isMatchSuffixs(file)) {
				sums += countFileTextLines(file);
			}
		}
	}

	
	/**
	 * 检查文件是否是制定后缀的文件
	 * @param file
	 * @return 
	 * @Author Aaron
	 */
	private static boolean isMatchSuffixs(File file) {
		String fileName = file.getName();
		if (fileName.indexOf(".") != -1) {
			String extName = fileName.substring(fileName.indexOf(".") + 1);
			for (int i = 0; i < suffixs.length; i++) {
				if (suffixs[i].equals(extName)) {
					return true;
				}
			}
		}
		return false;
	}

	/**
	 * 统计文件行数
	 * @param file
	 * @return
	 * @throws IOException 
	 * @Author Aaron
	 */
	private static long countFileTextLines(File file) throws IOException {
		long result = 0;
		if (statistics.length() > FLUSH_FLAG) {
			System.out.print(statistics.toString());
			statistics = new StringBuffer();
		}
		statistics.append("文件行数： ").append(file.getAbsolutePath());
		BufferedReader br = new BufferedReader(new FileReader(file));
		while (br.readLine() != null)
			result++;
		br.close();
		statistics.append("   -  ").append(result).append("\n");
		return result;
	}
}