Java统计指定格式文件数量及文件中代码行数

前言

最近为了创新学分在申请软著，软著申请表上要求标注代码行数，而且还需要贴源代码。但是作为懒宅晚期患者，我感觉数代码行数和粘贴代码超级无聊。我就想到之前老师说一个学长用python写过类似的工具来统计代码行数，然后我就想Java能不能实现。经过简单的封装就用Java写了这个简单的工具类来统计代码。当然你也可以使用这个工具类将源代码输出到txt文件里。代码比较简单，没有处理注释和空行，如果想处理注释和空行也可以自己加点逻辑（空行主要使用StringUtils中的isEmpty()方法判断，注释根据注释习惯可以使用正则表达式处理）。

代码

import java.io.*;

/**
 * @ClassName: CodeCount
 * @date: 2020/11/27 14:50
 * @author 索半斤
 * @Description: 小工具，用来申请软著时自动统计代码行数并能够将代码输出到txt文本
 */
public class CodeCount {
    private static int i,j = 0;
    /**
     * @MethodName: countCodeLine
     * @date: 2020/11/27 14:36
     * @author 索半斤
     * @Description: 用于统计代码行数
     * @param path 要扫描的包的绝对路径
     * @param suffix 要统计的文件后缀，例如java
     */
    public static Integer countCodeLine(String path,String suffix){
        return getFileOrCodeCount(path, suffix, "i");
    }

    /**
     * @MethodName: countFiles
     * @date: 2020/11/27 14:37
     * @author 索半斤
     * @Description: 用于统计文件数
     * @param path 要扫描的包的绝对路径
     * @param suffix 要统计的文件的后缀
     */
    public static Integer countFiles(String path,String suffix){
       return getFileOrCodeCount(path, suffix,  "j");
    }

    /**
     * @MethodName: outFile
     * @date: 2020/11/27 14:43
     * @author 索半斤
     * @Description: 将代码输出到txt文件中（自动换行，默认目录）
     * @param path 要扫描的包的绝对路径
     * @param suffix 要统计的文件的后缀
     * @param name 输出的文件名称
     */
    public static void outFile(String path,String suffix,String name){
        File file = new File(path);
        BufferedWriter bufferedWriter;
        try {
            bufferedWriter = new BufferedWriter(new FileWriter(name+".txt"));
            traverseFiles(file,bufferedWriter,suffix);
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    /**
     * @MethodName: outFileToPath
     * @date: 2020/11/27 14:45
     * @author 索半斤
     * @Description: 将代码写入到文本文件并输出到指定的目录
     * @param path 要扫描的包的绝对路径
     * @param suffix 要统计的文件的后缀
     * @param Filepath 要输出的文件夹绝对路径
     */
    public static void outFileToPath(String path,String suffix,String Filepath){
        boolean b = Filepath.endsWith(".txt");
        if (!b)return;
        File file = new File(path);
        BufferedWriter bufferedWriter;
        try {
            bufferedWriter = new BufferedWriter(new FileWriter(Filepath));
            traverseFiles(file,bufferedWriter,suffix);
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    /**
     * @MethodName: getFileOrCodeCount
     * @date: 2020/11/27 14:46
     * @author 索半斤
     * @Description: 获取代码行数或者文件数量
     */
    private static Integer getFileOrCodeCount(String path,String suffix,String type){
        File file = new File(path);
        try {
            traverseFiles(file,suffix);
            if (type.equals("i"))return i;
            else if (type.equals("j"))return j;
            else return null;
        } catch (IOException e) {
            e.printStackTrace();
            return null;
        }
    }



    private static void traverseFiles(File file, BufferedWriter bufferedWriter,String suffix) throws IOException {
        if(!file.exists()){
            return;
        }

        if(!file.isDirectory()){
            String filename = file.getName();
            if(filename.endsWith("."+suffix)){
                BufferedReader bufferedReader = new BufferedReader(new FileReader(file));
                String string =null;
                while ((string = bufferedReader.readLine()) != null) {
                    //if (StringUtils.isEmpty(string))continue; 处理空行
                    bufferedWriter.write(string);
                    bufferedWriter.newLine();
                    bufferedWriter.flush();
                }
            }else
                return;
        }

        File[] files =file.listFiles();
        if (files == null || files.length == 0) {
            return;
        }

        for(File file2 : files){
            traverseFiles(file2,bufferedWriter,suffix);
        }
    }

    private static void traverseFiles(File file,String suffix) throws IOException {
        if(!file.exists()){
            return;
        }
        if(!file.isDirectory()){
            String filename = file.getName();
            if(filename.endsWith("."+suffix)){
                j++;
                BufferedReader bufferedReader = new BufferedReader(new FileReader(file));
                String string =null;
                while ((string = bufferedReader.readLine()) != null) {
                    i++;//读取行数
                }
            }else
                return;
        }

        File[] files =file.listFiles();
        if (files == null || files.length == 0) {
            return;
        }

        for(File file2 : files){
            traverseFiles(file2,suffix);
        }
    }

}