JAVA兩種代理模式

簡單設計動態代理，基本模擬spring的動態代理方式。

before afterReturning around afterException after這些通知方法都可以這塊模擬出來

spring的AOP： 

1.在容器中的對象如果實現了接口則采用JDK的動態代理。

2在容器中的對象沒有實現接口，則用（cglib）繼承的方式實現動態代理。

現在模擬spring的動態代理。

首先准備接口（UserService）和實現接口的目標對象（UserServiceImpl）。

 

public interface UserService {

void save(); 

}

public class UserServiceImpl implements UserService {

    public void save() {
        System.out.println("保存用戶");
    } 

}

 

1.動態代理

/**
 * 動態代理1
 * 
 * @author shihaibin
 * @param <T>
 *
 */
public class UserServiceProxyFactory implements InvocationHandler {

    Object impl;

    public <T> Object getProxy(Class<T> clz) {
        try {
            impl = clz.newInstance();
        } catch (InstantiationException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        } catch (IllegalAccessException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        }
        // 生成動態代理
        Object usProxy = Proxy.newProxyInstance(clz.getClassLoader(), clz.getInterfaces(), this);
        // 返回
        return usProxy;
    }

    /**
     * 參數：1.當前代理對象 2.當前方法 3.當前方法執行的時候的參數
     */
    public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {
        before();
        Object invoke = method.invoke(impl, args);
        after();
        return invoke;
    }
    //可以重寫
    public void before() {
        System.out.println("之后");
    }

    public void after() {
        System.out.println("之后");
    }
}

測試：

@Test
    public void find1() {
        // 簡單的aop
        UserServiceProxyFactory factory = new UserServiceProxyFactory();
        UserService userServiceProxy = (UserService) factory.getProxy(UserServiceImpl.class);
        userServiceProxy.save(); 
        System.out.println(userServiceProxy instanceof UserServiceImpl);
    }

 

2.cglib代理

/**
 * 動態代理2 cglib代理
 * 
 * @author shihaibin
 *
 */
public class UserServiceProxyFactory2 implements MethodInterceptor {

    public <T> Object getProxy(Class<T> clz) {
        Enhancer en = new Enhancer();// 幫我們生成代理對象
        en.setSuperclass(clz);// 設置對誰進行代理
        en.setCallback(this);//回調函數
        return en.create();// 創建代理對象;
    }

    /**
     * prxoyobj:被代理的原始對象 method：被代理的原始方法 arg：運行期的參數 methodProxy：產生的代理方法
     */
    public Object intercept(Object prxoyobj, Method method, Object[] arg, MethodProxy methodProxy) throws Throwable {
        // 打開事務
        System.out.println("打開事務！");
        // 調用原有方法
        Object invokeSuper = methodProxy.invokeSuper(prxoyobj, arg);
        // 提交事務
        System.out.println("提交事務！");
        return invokeSuper;
    }
}

 

測試：

@Test
    public void find2() {
        // 重寫aop前后方法
        UserServiceProxyFactory2 factoryContext = new UserServiceProxyFactory2();
        UserService userServiceProxy = (UserService) factoryContext.getProxy(UserServiceImpl.class);
        userServiceProxy.save();
        // 判斷代理對象是否屬於被代理對象類型
        // 代理對象繼承了被代理對象=>true
        System.out.println(userServiceProxy instanceof UserServiceImpl);// 判斷是否屬於被代理對象類型
    }