首页 > 代码库 > 代码统计

代码统计

http://www.oschina.net/code/snippet_566026_17466

 

import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.util.ArrayList;
import java.util.List;
import java.util.regex.Matcher;
import java.util.regex.Pattern;
 
/**
 * 代码条数统计
 * @author user
 */
public class CodeCount{
 
    /**
     * 正则表达式主要是匹配多行注释
     */
    private static Pattern pattern = Pattern.compile("/\\*([\\s|\\S])+?\\*/");
     
    // 基本文件目录
    private File baseDir;
    // 拥有的文件列表
    private List<String> fileList = new ArrayList<String>();
    // 文件数量
    private int fileCount = 0;
    // 代码行数
    private int lineCount = 0;
    // 匹配表达式
    private String matchRegex = "[a-zA-Z0-9_-]*.java";
 
    public CodeCount(File baseDir) {
        this.baseDir = baseDir;
    }
     
    public CodeCount(File baseDir, String matchRegex) {
        this.baseDir = baseDir;
        this.matchRegex = matchRegex;
    }
     
    /**
     * 统计入口
     */
    public void count() {
        if(!baseDir.isHidden()){
            if (baseDir.isDirectory()) {
                for (File child : baseDir.listFiles()) {
                    CodeCount count = new CodeCount(child, matchRegex);
                    count.count();
                    fileCount += count.fileCount;
                    lineCount += count.lineCount;
                    fileList.addAll(count.fileList);
                }
            } else {
                if(baseDir.getName().matches(matchRegex)){
                    try {
                        fileList.add(baseDir.getPath());
                        fileCount ++;
                        String content = getFileContent(baseDir);
                        int line = getCodeCount(content);
                        lineCount += line;
                    } catch (IOException e) {
                        e.printStackTrace();
                    }
                }
            }
        }
    }
 
    /**
     * 获取文件f中的文本内容
     *
     * @param f
     * @return
     * @throws IOException
     */
    public String getFileContent(File f) throws IOException {
        StringBuffer sb = new StringBuffer();
        FileInputStream input = new FileInputStream(f);
        byte[] b = new byte[1024];
        int count = 0;
        while ((count = input.read(b)) != -1) {
            sb.append(new String(b, 0, count));
        }
        input.close();
        return sb.toString();
    }
 
    /**
     * 统计文本中的代码行数 1.用正则表达式出去块注释即多行注释 2.再将字符串按照回车符分割成字符串数组 3.除去空白行或者单行注释行
     *
     * @param content
     *            输入文本
     * @return 文本中代码行数
     */
    public int getCodeCount(String content) {
        int rowCount = 0;
        Matcher matcher = pattern.matcher(content);
        content = matcher.replaceAll(" ");
        String ss[] = content.split("\n");
        for (String s : ss) {
            if (s.trim().length() > 0 && !s.trim().startsWith("//"))
                rowCount++;
        }
        return rowCount;
    }
 
    /**
     * 打印输出结果
     */
    public void print(){
        // 统计结果输出
        System.out.println(baseDir + "中所有文件:");
        for (String file : fileList) {
            System.out.println(file);
        }
        System.out.println();
        System.out.println("文件总数:" + fileCount);
        System.out.println("总代码行数" + lineCount);
    }
     
    public File getBaseDir() {
        return baseDir;
    }
 
    public void setBaseDir(File baseDir) {
        this.baseDir = baseDir;
    }
 
    public List<String> getFileList() {
        return fileList;
    }
 
    public void setFileList(List<String> fileList) {
        this.fileList = fileList;
    }
 
    public int getFileCount() {
        return fileCount;
    }
 
    public void setFileCount(int fileCount) {
        this.fileCount = fileCount;
    }
 
    public int getLineCount() {
        return lineCount;
    }
 
    public void setLineCount(int lineCount) {
        this.lineCount = lineCount;
    }
 
    public String getMatchRegex() {
        return matchRegex;
    }
 
    public void setMatchRegex(String matchRegex) {
        this.matchRegex = matchRegex;
    }
 
    public static void main(String[] args) throws Exception {
        String path = "S:\\快盘\\workspace\\commons-godway";
        CodeCount code = new CodeCount(new File(path));
        code.count();
        code.print();
    }
}

代码统计