統計java程式碼行數工具

瓜瓜東西發表於2014-05-20
package com.wanju.project001.zonghe.test;


import java.io.BufferedReader;
import java.io.File;
import java.io.FileNotFoundException;
import java.io.FileReader;
import java.io.IOException;
import java.util.ArrayList;


public class CodeCounter {


    //有效行數
    private static long normalLines = 0;


    //註釋行數
    private static long commentLines = 0;


    //空白行
    private static long whiteLines = 0;


    private ArrayList<File> al = new ArrayList<File>();


    /**
     * 遞迴找到該目錄下的所有java原始檔
     * @param homeDir
     */
    public void findJavaSource(File homeDir) {
        if (homeDir.isFile()) {
            al.add(homeDir);
            return;
        }
        File[] f = homeDir.listFiles();
        for (File child : f) {
            if (child.isFile() && child.getName().matches(".*\\.java$")) {
                al.add(child);
            } else if (child.isDirectory()) {
                findJavaSource(child);
            }
        }
    }


    public static void main(String[] str) {
        CodeCounter cc = new CodeCounter();
        //測試時別忘了該目錄地址
        File homeDir = new File("D:\\workspace_ptc\\wanju_debug\\src");
        cc.count(homeDir);
    }


    public void count(File homeDir) {
        this.findJavaSource(homeDir);
        for (File child : al) {
            parse(child);
        }
        System.out.println("normalLines:" + normalLines);
        System.out.println("commentLines:" + commentLines);
        System.out.println("whiteLines:" + whiteLines);
    }


    private static void parse(File f) {
        BufferedReader br = null;
        boolean comment = false;
        try {
            br = new BufferedReader(new FileReader(f));
            String line = "";
            while ((line = br.readLine()) != null) {
                line = line.trim();
                if (line.matches("^[\\s&&[^\\n]]*$")) {
                    whiteLines++;
                } else if (line.startsWith("/*") && !line.endsWith("*/")) {
                    commentLines++;
                    comment = true;
                } else if (line.startsWith("/*") && line.endsWith("*/")) {
                    commentLines++;
                } else if (true == comment) {
                    commentLines++;
                    if (line.endsWith("*/")) {
                        comment = false;
                    }
                } else if (line.startsWith("//")) {
                    commentLines++;
                } else {
                    normalLines++;
                }
            }
        } catch (FileNotFoundException e) {
            e.printStackTrace();
        } catch (IOException e) {
            e.printStackTrace();
        } finally {
            if (br != null) {
                try {
                    br.close();
                    br = null;
                } catch (IOException e) {
                    e.printStackTrace();
                }
            }
        }
    }


}

相關文章