快捷導(dǎo)航

Spring?AOP手寫動態(tài)代理代碼實例

更新時間：2024年01月11日 09:57:58 作者：好奇的7號

這篇文章主要介紹了Spring?AOP手寫動態(tài)代理代碼實例,AOP我們知道,是在不修改源代碼的情況下,為代碼添加一些新功能的技術(shù),通過動態(tài)代理,可以在不修改原始類代碼的前提下,對方法進(jìn)行攔截和增強(qiáng),需要的朋友可以參考下

Spring AOP動態(tài)代理

基礎(chǔ)知識

AOP我們知道，是在不修改源代碼的情況下，為代碼添加一些新功能的技術(shù)。通過動態(tài)代理，可以在不修改原始類代碼的前提下，對方法進(jìn)行攔截和增強(qiáng)。動態(tài)代理常用于在不改變原有業(yè)務(wù)邏輯的情況下，對方法的調(diào)用進(jìn)行額外的處理，比如日志記錄、性能監(jiān)控、事務(wù)管理等。它實現(xiàn)了面向切面編程（AOP）的核心概念。

1、動態(tài)代理

AOP其實就是先通過“一些辦法”拿到“代理對象”，再對這個代理對象添加新方法。

動態(tài)代理的兩種辦法：

1.1 JDK代理：基于接口的動態(tài)代理技術(shù)

方法：通過目標(biāo)接口，創(chuàng)建接口實現(xiàn)類，動態(tài)地在運行時內(nèi)存里生成代理對象，其擁有目標(biāo)對象的原方法實現(xiàn)，最后通過代理對象來增加新功能。

看文字有些抽象，現(xiàn)在手動用代碼實現(xiàn)一個JDK代理的AOP技術(shù)，根據(jù)jdk1.8手冊找到Proxy類，通過newInstance方法實現(xiàn)動態(tài)代理對象的創(chuàng)建。

“Proxy提供了創(chuàng)建動態(tài)代理類和實例的靜態(tài)方法，它也是由這些方法創(chuàng)建的所有動態(tài)代理類的超類。”

（1）準(zhǔn)備接口和實現(xiàn)類

//=========================接口============================
public interface UserDao {
    public int add(int a,int b);
    public String update(String id);
}
//=========================實現(xiàn)類============================
public class UserDaoImpl implements UserDao{
    @Override
    public int add(int a, int b) {
        System.out.println("add.......");
        return a+b;
    }
    @Override
    public String update(String id) {
        System.out.println("update===========");
        return id;
    }
}

（2）創(chuàng)建UserDao的代理對象，調(diào)用增強(qiáng)方法需要注意：InvocationHandler接口里有invoke方法必須實現(xiàn)，而調(diào)用代理對象的任何方法，實質(zhì)都是在執(zhí)行invoke方法。

public class JDKProxy {
    public static void main(String[] args) {
/**
newProxyInstance的三個內(nèi)容：類加載器、要實現(xiàn)的接口的class、
和InvocationHandler（這是個接口，里面寫的增強(qiáng)類的具體邏輯）
*/
        Class[] interfaces = {UserDao.class};
        UserDaoImpl userDao = new UserDaoImpl();
        UserDao dao = (UserDao) Proxy.newProxyInstance(
                JDKProxy.class.getClassLoader(),
                interfaces,
                new UserDaoProxy(userDao));//生成的“dao”就是代理對象實例。
        //也可以直接用匿名內(nèi)部類的方法來寫(new InvocationHandler)
        int add = dao.add(1, 2);//使用代理對象dao調(diào)用add的時候，無需改變add源碼，卻增強(qiáng)了功能。
        System.out.println("add=" + add);
    }
}
//創(chuàng)建代理對象代碼
class UserDaoProxy implements InvocationHandler {
    //把創(chuàng)建的是誰的代理對象，把誰傳遞過來
    //有參數(shù)構(gòu)造傳遞
    private Object obj;//要創(chuàng)建obj的代理對象
    public UserDaoProxy(Object obj) {
        this.obj = obj;
    }
    //增強(qiáng)的邏輯：InvocationHandler接口里有invoke方法必須實現(xiàn)
//調(diào)用代理對象的任何方法，實質(zhì)都是在執(zhí)行invoke方法。
    @Override
    public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {
        //方法之前（增強(qiáng)）
        System.out.println("方法之前執(zhí)行...."+method.getName()+" :傳遞的參數(shù)..." + Arrays.toString(args));
        //被增強(qiáng)的方法執(zhí)行（原模塊功能不變）
        Object res = method.invoke(obj, args);//res=3
        //方法之后（增強(qiáng)）
        System.out.println("方法之后執(zhí)行...."+obj);
        return res;
    }
}

但是有個遺留問題，invoke函數(shù)究竟是如何被調(diào)用的，往下看了源碼也沒有分析出來，個人推測是這樣的：

因為傳入的obj其實就是要動態(tài)代理的原對象，但我們不知道具體執(zhí)行什么方法，此時dao還未調(diào)用add方法。但調(diào)用了以后，invoke方法卻能夠自動調(diào)用、增強(qiáng)方法，內(nèi)部應(yīng)該是采用反射的方法，拿到“add”這個方法名稱，通過getMethod反射出方法的類對象，再用method.invoke(具體對象)來實現(xiàn)方法的調(diào)用，從而調(diào)用add，并完成前后的增強(qiáng)。