Spring 容器AOP的實現原理——動態代理

參考：http://wiki.jikexueyuan.com/project/ssh-noob-learning/dynamic-proxy.html（from極客學院）

 

一、介紹

Spring的動態代理有兩種：一是JDK的動態代理；另一個是cglib動態代理（通過修改字節碼來實現代理）。

今天主要討論JDK動態代理的方式。

JDK的代理方式主要就是通過反射跟動態編譯來實現的，主要搭配InvocationHandler和Proxy來實現，下面的例子中使用了兩層代理（即代理上加了一層代理）。

 

二、實例

1. 公共接口

package com.tgb.proxy;

public interface UserMgr {

    void addUser();
    void delUser();
    
}

 

2. 實現類

package com.tgb.proxy;

public class UserMgrImpl implements UserMgr {

    @Override
    public void addUser() {
        System.out.println("添加用戶....");
    }

    @Override
    public void delUser() {
        System.out.println("刪除用戶.....");
    }

}

 

3. TransactionHandler（InvocationHandler的實現）

package com.tgb.proxy;

import java.lang.reflect.InvocationHandler;
import java.lang.reflect.Method;

public class TransactionHandler implements InvocationHandler {

    private Object target;
    
    public TransactionHandler(Object target){
        super();
        this.target = target;
    }

    @Override
    public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {
        
        System.out.println("開啟事務...");
        
        method.invoke(target);
        
        System.out.println("提交事務...");
        
        return null;
    }
    

}

 

4. TimeHandler（InvocationHandler的實現）

package com.tgb.proxy;

import java.lang.reflect.InvocationHandler;
import java.lang.reflect.Method;
import java.util.Calendar;

public class TimeHandler implements InvocationHandler {

    private Object target;

    public TimeHandler(Object target) {
        super();
        this.target = target;
    }

    @Override
    public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {

        Calendar calendar = Calendar.getInstance();

        System.out.println("start time:" + calendar.get(Calendar.HOUR_OF_DAY));

        method.invoke(target);

        System.out.println("end time:" + calendar.get(Calendar.HOUR_OF_DAY));

        return null;
    }

}

 

5. 測試類

package com.tgb.proxy;

import java.lang.reflect.InvocationHandler;
import java.lang.reflect.Proxy;

public class Client {

    public static void main(String[] args) {

        UserMgr userMgr = new UserMgrImpl();

        // 1.第一層代理----------通過動態代理，添加事務處理
        InvocationHandler handler = new TransactionHandler(userMgr);
        UserMgr userMgrProxy = (UserMgr) Proxy.newProxyInstance(userMgr.getClass().getClassLoader(),
                userMgr.getClass().getInterfaces(), handler);

        // 2.第二層代理----------通過動態代理，添加時間處理
        InvocationHandler handler2 = new TimeHandler(userMgrProxy);
        UserMgr userMgrProxy2 = (UserMgr) Proxy.newProxyInstance(userMgrProxy.getClass().getClassLoader(),
                userMgrProxy.getClass().getInterfaces(), handler2);

        userMgrProxy2.addUser();

        System.out.println("========================================");

        userMgrProxy2.delUser();

    }
}

 

輸出結果：

start time:Tue Aug 09 23:54:54 CST 2016
開啟事務...
添加用戶....
提交事務...
end time:Tue Aug 09 23:54:54 CST 2016
========================================
start time:Tue Aug 09 23:54:54 CST 2016
開啟事務...
刪除用戶.....
提交事務...
end time:Tue Aug 09 23:54:54 CST 2016