代码行数统计

夙愿已清 提交于 2021-02-18 01:19:10
        public static void main(String[] args) {
		String filePath = "/Volumes/DATA/xxxxx";
		File rootFile = new File(filePath);
		System.out.println(
				"计:"+countFile(rootFile, "java", true) + " 个文件,共 "
		+countFile(rootFile, "java", false)+" 行!");
	}
	
	private static int countFile(File rootFile, String suffix, boolean countLine){
		int result = 0;
		File[] childFiles = rootFile.listFiles(new FileFilter() {
			
			@Override
			public boolean accept(File pathname) {
				return pathname.isDirectory() || pathname.getAbsolutePath().endsWith(suffix);
			}
		});
		for (File f : childFiles) {
			if (f.isFile()) {
				int lineCount = getFileLinesCount(f);
//				System.out.println(f.getAbsolutePath()+" "+lineCount);
				if (countLine) {
					result += lineCount;
				} else {
					result++;
				}
			} else {
				result += countFile(f, suffix, countLine);
			}
		}
		return result;
	}
	
	private static int getFileLinesCount(File file) {
		if (file == null || file.isDirectory()) return 0;
		BufferedReader br = null;
		int result = 0;
		try {
			br = new BufferedReader(new FileReader(file));
			while (br.readLine() != null) {
				result ++;
			}
		} catch (IOException e) {
			e.printStackTrace();
		} finally{
			if (br != null) {
				try {
					br.close();
				} catch (IOException e2) {
				}
			}
		}
		return result;
	}


易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!