最初接触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。
解决方案引用自:
ps:
目前遇到一个问题,在调试的时候偶然发现,attr.index()下标不一定按照自然增序添加。比如,0,1,2,3,5,6等。类似这样的顺序。此时,只有从小到大的顺序可以描述参数位置,因此,上述后面的部分代码修正为如下:
TreeMapmap = 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; } }