Javassist初体验

       最初接触javassist是在研究dubbo源码的时候，那会对其的理解还停留在动态生成字节码的位置，可以做动态代理之类的动态化处理。最近由于项目需要扫描springMVC中controller层的注解，同时还需要知道方法入参的名字。基于java反射拿到注解简单，但是要拿到入参名字却需要jdk8才提供支持。但是当前项目整体环境基于jdk7，因此又想起javassist，算是get一个新技能。

基本的代码如下：

private String[] getParamName(Method method) throws Exception {
    CtClass cc = pool.get(method.getDeclaringClass().getName());

    CtClass[] params = new CtClass[method.getParameterTypes().length];
    for(int i = 0; i < method.getParameterTypes().length; i++) {
        params[i] = pool.getCtClass(method.getParameterTypes()[i].getName());
    }

    CtMethod cm = cc.getDeclaredMethod(method.getName(), params);

    MethodInfo methodInfo = cm.getMethodInfo();
    CodeAttribute codeAttribute = methodInfo.getCodeAttribute();

    LocalVariableAttribute attr = (LocalVariableAttribute) codeAttribute.getAttribute(LocalVariableAttribute.tag);
    String[] paramNames = new String[cm.getParameterTypes().length];
    int pos = Modifier.isStatic(cm.getModifiers()) ? 0 : 1;
    for(int i = 0; i < attr.tableLength(); i++) {
        if(attr.index(i) >= pos && attr.index(i) < paramNames.length + pos) {
            paramNames[attr.index(i) - pos] = attr.variableName(i);
        }
    }

    return paramNames;
}

 

关键的代码在于下面那个for循环。

先看看Java虚拟机规范的描述：

Java虚拟机使用局部变量表来完成方法调用时的参数传递，当一个方法被调用的时候，它的参数将会传递至从0开始的连续的局部变量表位置上。特别地，当一个实例方法被调用的时候，第0个局部变量一定是用来存储被调用的实例方法所在的对象的引用（即Java语言中的“this”关键字）。后续的其他参数将会传递至从1开始的连续的局部变量表位置上。

 

但此描述中第 0 个局部变量是指 slot 排号为 0，而不是位置为 0。

所以正确的做法就是遍历本地变量表，根据 slot 值确认是不是方法参数，体现在代码中的 API 就是 attr.index(i) 将会返回 slot。

pos 变量主要用于处理实例方法前面的 this 变量，静态方法 pos 将为 0。

解决方案引用自：http://lzxz1234.github.io/java/2014/07/25/Get-Method-Parameter-Names-With-Javassist/

ps:

目前遇到一个问题，在调试的时候偶然发现，attr.index()下标不一定按照自然增序添加。比如，0,1,2,3,5,6等。类似这样的顺序。此时，只有从小到大的顺序可以描述参数位置，因此，上述后面的部分代码修正为如下：

TreeMap<Integer, Integer> map = new TreeMap<>();
for(int i = 0; i < attr.tableLength(); i++) {
    map.put(attr.index(i), i);
}
int index = 0;
boolean flag = false;
for(Integer key : map.keySet()) {
    if (!flag) {
        flag = true;
        continue;
    }

    if (index < paramNames.length) {
        paramNames[index++] = attr.variableName(map.get(key));
    } else {
        break;
    }
}