- 首页 > it技术 > >
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;
}
}
推荐阅读
