java 动态AOP

一、實現機制：

在运行期，所有类加载器加载字节码前，前进行拦截。並將代碼植入。可以对所有类进行织入。

二、實現方式：

1. 實現ClassFileTransformer 接口

2. 添加以下方法（必須）：

public static void premain(String options, Instrumentation ins) {   
     //注册我自己的字节码转换器   
    ins.addTransformer(new MyClassFileTransformer());   
}  

實例：

package com.aop;

import java.io.IOException;
import java.lang.instrument.ClassFileTransformer;
import java.lang.instrument.IllegalClassFormatException;
import java.lang.instrument.Instrumentation;
import java.security.ProtectionDomain;

import javassist.CannotCompileException;
import javassist.ClassPool;
import javassist.CtClass;
import javassist.CtMethod;
import javassist.NotFoundException;

public class AopTransformer implements ClassFileTransformer {

    /**
     * 字节码加载到虚拟机前会进入这个方法
*/
    @Override
    public byte[] transform(ClassLoader loader, String className,
            Class<?> classBeingRedefined, ProtectionDomain protectionDomain,
            byte[] classfileBuffer) throws IllegalClassFormatException {

        // javassist的包名是用点分割的，需要转换下
        if (className.indexOf("/") != -1) {
            className = className.replaceAll("/", ".");
        }
        
        try {
            // 通过包名获取类文件
            CtClass cc = ClassPool.getDefault().get(className);

            // 获得指定方法名的方法
            CtMethod m = cc.getDeclaredMethod("sayhello");

            // 在方法执行前插入代码
            m.insertBefore("{System.out.println(\"在HelloTest.sayhello之前執行\");}");
            m.insertAfter("{System.out.println(\"在HelloTest.sayhello之後執行\");}");
            m = cc.getDeclaredMethod("sayGoodBye");

            // 在方法执行前插入代码
            m.setBody("{System.out.println(\"修改HelloTest.sayGoodBye的方法體\");}");
            return cc.toBytecode();
        } catch (NotFoundException e) {
        } catch (CannotCompileException e) {
            e.printStackTrace();
        } catch (IOException e) {
            // 忽略异常处理
        }
        return null;
    }

    /**
     * 在main函数执行前，执行的函数
     * 
     * @param options
     * @param ins
*/
    public static void premain(String options, Instrumentation ins) {
        // 注册我自己的字节码转换器
        ins.addTransformer(new AopTransformer());
    }
}

package com.test;

public class HelloTest {
    public void sayhello() {
        System.out.println("HelloTest sayhello");
    }
    public void sayGoodBye() {
        System.out.println("HelloTest sayGoodBye");
    }

    public static void main(String[] args) {
        HelloTest ht = new HelloTest();
        ht.sayhello();
        ht.sayGoodBye();
    }
}

三、執行

1. 需要告诉JVM在启动main函数之前，需要先执行premain函数。首先需要将premain函数所在的类打成jar包。并修改该jar包里的META-INF\MANIFEST.MF 文件，MANIFEST.MF 文件內容如下：

Manifest-Version: 1.0
Premain-Class: com.aop.AopTransformer
Can-Redefine-Classes: true
Can-Retransform-Classes: true
Can-Set-Native-Method-Prefix: true

2. 將aop.jar放到同一目錄

3. 使用java命令執行main方法：

java -javaagent:.\aop.jar HelloTest

4. 執行結果比較

如果沒有添加aop執行結果如下：

HelloTest sayhello
HelloTest sayGoodBye

添加aop執行結果如下：

在HelloTest.sayhello之前執行
HelloTest sayhello
在HelloTest.sayhello之後執行
修改HelloTest.sayGoodBye的方法體