java.lang.instrument学习 （二），javassist实例练习

学习了instrument之后试着自己写点东西，上一篇的例子中使用的是asm，毕竟是面向字节码的，api还是比较复杂的。其实有时候的需求很简单，无非就是看下类里的方法啊之类的。javassist是基于源码级别的API比基于字节码的ASM简单。下面就是对使用到的类打印出所有方法的Transformer：

public class TestTransformer implements ClassFileTransformer {
    @Override
    public byte[] transform(ClassLoader classLoader, String className, Class<?> aClass, ProtectionDomain protectionDomain, byte[] bytes) throws IllegalClassFormatException {
        //javassist的包名是用点分割的，需要转换下
        String compareClass = className.replace('/', '.');
        //通过包名获取类文件
        CtClass clazz = null;
        try {
            clazz = ClassPool.getDefault().get(compareClass);
            CtMethod[] methods = clazz.getMethods();
            for(CtMethod method:methods){
                System.out.println(className + ",has method:" + method.getName());
            }
        } catch (NotFoundException e) {
            e.printStackTrace();
        }
        return bytes;
    }
}

比用asm的简单多了。

再说下IDEA上面怎么打jar包和调试这样的程序。首先点击Project structure。点左边的Artifacts，然后点“+”选jar，再选Empty。然后对jar包取个名字，在Output Layout里加入你想要打进去的资源。如果定义了MF文件，设置下MF文件路径。

点击ok之后就可以在Build菜单中选择Build Artifacts 就ok了。jar包的目录在上面配置的Output directory里。如果你的测试代码放在同一个工程里，可以在新建一个Application应用，然后在VM options中配置jar包的路径 -javaagent:/Users/ykdsg/my_workspace/class_instrument/out/artifacts/hackX/hackX.jar 。这样就ok了，debug的时候，断点甚至能进入Transformer，IDEA还是相当牛逼的。