Eclipse AST 实现一个类信息统计小程序

这个是对Eclipse AST 获取与访问的介绍，http://blog.csdn.net/LoveLion/article/details/19050155  (我的老师的博客)

本文的工具类完全来自于该文章，其他的类有这篇文章的影子

首先是 工具类，不多讲，请看老师的介绍

package com.kyc.rec;

import java.io.BufferedInputStream;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.IOException;

import org.eclipse.jdt.core.dom.AST;
import org.eclipse.jdt.core.dom.ASTParser;
import org.eclipse.jdt.core.dom.CompilationUnit;

public class JdtAstUtil {
    /**
    * get compilation unit of source code
    * @param javaFilePath 
    * @return CompilationUnit
    */
    public static CompilationUnit getCompilationUnit(String javaFilePath){
        byte[] input = null;
		try {
		    BufferedInputStream bufferedInputStream = new BufferedInputStream(new FileInputStream(javaFilePath));
		    input = new byte[bufferedInputStream.available()];
	            bufferedInputStream.read(input);
	            bufferedInputStream.close();
		} catch (FileNotFoundException e) {
			e.printStackTrace();
		} catch (IOException e) {
			e.printStackTrace();
		}

        
	ASTParser astParser = ASTParser.newParser(AST.JLS3);
        astParser.setSource(new String(input).toCharArray());
        astParser.setKind(ASTParser.K_COMPILATION_UNIT);

        CompilationUnit result = (CompilationUnit) (astParser.createAST(null));
        
        return result;
    }
}

接下来是DTO用来存储类中方法的相关信息

package com.kyc.rec;


public class MethodDTO {
	
	private String name;//方法名
	private int codeLength;//方法的代码长度
	private int numOfParam;//方法参数个数
	
	public MethodDTO(String name,int codeLength,int numOfParam){
		setName(name);
		setCodeLength(codeLength);
		setNumOfParam(numOfParam);
		
	}
	public MethodDTO(){
		
	}

	public String getName() {
		return name;
	}

	public void setName(String name) {
		this.name = name;
	}

	public int getCodeLength() {
		return codeLength;
	}

	public void setCodeLength(int codeLength) {
		this.codeLength = codeLength;
	}

	public int getNumOfParam() {
		return numOfParam;
	}

	public void setNumOfParam(int numOfParam) {
		this.numOfParam = numOfParam;
	}


}

下面开始访问，其他访问都不难，只是在获取参数最多的方法的名字时，对我来说有点小麻烦，，因为我对AST的了解还很肤浅，不知道有没有更先进的技术可以使我的方法变得简单

package com.kyc.rec;

import java.util.ArrayList;


import org.eclipse.jdt.core.dom.ASTVisitor;
import org.eclipse.jdt.core.dom.CompilationUnit;
import org.eclipse.jdt.core.dom.FieldDeclaration;
import org.eclipse.jdt.core.dom.MethodDeclaration;
import org.eclipse.jdt.core.dom.SingleVariableDeclaration;
import org.eclipse.jdt.core.dom.TypeDeclaration;


public class DemoVisitor extends ASTVisitor {
	/*类中方法的个数、
	 * 属性的个数、
	 * 源代码行数、
	 * 代码行最多的方法名以及
	 * 代码行数、
	 * 参数个数最多的方法名
	 * 及其参数的个数等信息*/
	public int numOfMethod=0;//方法个数
	public int numOffield=0;//属性个数
	public int codeLength=0;//代码总行数
	
	public int index=-1;//用于获取参数的时候，指向上一个方法；
	public int[] listOfParam=new int[100];
	private int numOfParam=0;

	private ArrayList<MethodDTO> arrayList =new ArrayList<MethodDTO>();//存储类中方法的相关信息
	
	public void print(){
		System.out.println("属性个数为："+numOfMethod);
		System.out.println("方法个数为："+numOffield);
		System.out.println("代码行数为："+codeLength);
		System.out.println("代码行最多的方法:"+findLongestCodeLength().getName()+
				"行数为："+findLongestCodeLength().getCodeLength());
		System.out.println("参数最多的方法："+findMaxNumOfParam().getName()+
				"方法数为："+findMaxNumOfParam().getNumOfParam());
//		for(int i=0;i<arrayList.size();i++){
//			System.out.println(arrayList.get(i).getName()+" - "+arrayList.get(i).getNumOfParam());
//		}
	}


	@Override
	public boolean visit(FieldDeclaration node) {
		numOffield++;
		return true;
	}

	@Override
	public boolean visit(MethodDeclaration node) {
		if(index!=-1){
			arrayList.get(index).setNumOfParam(numOfParam);
			numOfParam=0;
		}
		index++;
		//System.out.println("Method:\t" + node.getName()+" "+node.getLength());
		numOfMethod++;
		CountLength cl=new CountLength();
		cl.countLength(node.toString(), "\n");
		addToArrayList(new MethodDTO(node.getName().toString(), cl.getCodeLength(),0));
		
		return true;
	}

	@Override
	public boolean visit(TypeDeclaration node) {
		//该方法可访问类名
		//System.out.println("Class:\t" + node.getName());		
		return true;
	}
	@Override
	public boolean visit(CompilationUnit node){
		//该方法可获得代码的总行数
		CountLength cl=new CountLength();
		cl.countLength(node.toString(), "\n");
		codeLength=cl.getCodeLength();
		return true;
	}
	
	public void addToArrayList(MethodDTO meDto) {
		arrayList.add(meDto);
	}
	
	public MethodDTO findLongestCodeLength(){
		//获取代码行数最多的方法
		int maxLength=0;
		MethodDTO  methodDTO=new MethodDTO();
		for(int i=0;i<arrayList.size();i++){
			if(arrayList.get(i).getCodeLength()>maxLength){
				maxLength=arrayList.get(i).getCodeLength();
			}
		}
		for(int i=0;i<arrayList.size();i++){
			if(arrayList.get(i).getCodeLength()==maxLength){
				methodDTO=arrayList.get(i);
				break;
			}
		}
		return methodDTO;
	}
	public MethodDTO findMaxNumOfParam(){
		//获取参数最多的方法
		//这个方法与上一个方法存在明显的代码的坏味道（重复代码），可以对此进行优化
		int maxNum=0;
		MethodDTO  methodDTO=new MethodDTO();
		for(int i=0;i<arrayList.size();i++){
			if(arrayList.get(i).getNumOfParam()>maxNum){
				maxNum=arrayList.get(i).getNumOfParam();
			}
		}
		for(int i=0;i<arrayList.size();i++){
			if(arrayList.get(i).getNumOfParam()==maxNum){
				methodDTO=arrayList.get(i);
				break;
			}
		}
		return methodDTO;
	}
	@Override
	public boolean visit(SingleVariableDeclaration node){
		//本方法在执行visit(MethodDeclaration node)后执行 Method中的参数次；
		
		numOfParam++;
		return true;

	}
	public void dotheEndParam(){
		arrayList.get(index).setNumOfParam(numOfParam);
	}

}

统计行数类

package com.kyc.rec;

public class CountLength {
	//此类用于获取字符串中换行字符的个数，str2代表子字符串
	private int codeLength=0;
	public int getCodeLength() {
		return codeLength;
	}
	public void setCodeLength(int codeLength) {
		this.codeLength = codeLength;
	}
	public int countLength(String str1, String str2) {  
        if (str1.indexOf(str2) == -1) {  
            return 0;  
        } else if (str1.indexOf(str2) != -1) {  
        	codeLength++;  
        	countLength(str1.substring(str1.indexOf(str2) +  
                   str2.length()), str2);  
               return codeLength ; 
        }
            return 0;  
    }  
}

test  类

package com.kyc.rec;

import org.eclipse.jdt.core.dom.CompilationUnit;



public class DemoVisitorTest {
	
	public DemoVisitorTest(String path) {
		CompilationUnit comp = JdtAstUtil.getCompilationUnit(path);
		
		DemoVisitor visitor = new DemoVisitor();
		
		comp.accept(visitor);
		visitor.dotheEndParam();
		visitor.print();
	}
	
	public static void main(String[] args) {
		new DemoVisitorTest("DemoVisitor.java");
		
		
	}
}

结果如下：

jar包可私信