完成menu单测
This commit is contained in:
63
src/test/java/cn/iocoder/dashboard/util/AopTargetUtils.java
Normal file
63
src/test/java/cn/iocoder/dashboard/util/AopTargetUtils.java
Normal file
@@ -0,0 +1,63 @@
|
||||
package cn.iocoder.dashboard.util;
|
||||
|
||||
import org.springframework.aop.framework.AdvisedSupport;
|
||||
import org.springframework.aop.framework.AopProxy;
|
||||
import org.springframework.aop.support.AopUtils;
|
||||
|
||||
import java.lang.reflect.Field;
|
||||
|
||||
/**
|
||||
* http://www.bubuko.com/infodetail-3471885.html
|
||||
*/
|
||||
public class AopTargetUtils {
|
||||
|
||||
|
||||
/**
|
||||
* 获取 目标对象
|
||||
*
|
||||
* @param proxy 代理对象
|
||||
* @return
|
||||
* @throws Exception
|
||||
*/
|
||||
public static Object getTarget(Object proxy) throws Exception {
|
||||
|
||||
if (!AopUtils.isAopProxy(proxy)) {
|
||||
return proxy; //不是代理对象
|
||||
}
|
||||
|
||||
if (AopUtils.isJdkDynamicProxy(proxy)) {
|
||||
return getJdkDynamicProxyTargetObject(proxy);
|
||||
} else { //cglib
|
||||
return getCglibProxyTargetObject(proxy);
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
|
||||
private static Object getCglibProxyTargetObject(Object proxy) throws Exception {
|
||||
Field h = proxy.getClass().getDeclaredField("CGLIB$CALLBACK_0");
|
||||
h.setAccessible(true);
|
||||
Object dynamicAdvisedInterceptor = h.get(proxy);
|
||||
|
||||
Field advised = dynamicAdvisedInterceptor.getClass().getDeclaredField("advised");
|
||||
advised.setAccessible(true);
|
||||
|
||||
Object target = ((AdvisedSupport) advised.get(dynamicAdvisedInterceptor)).getTargetSource().getTarget();
|
||||
|
||||
return target;
|
||||
}
|
||||
|
||||
private static Object getJdkDynamicProxyTargetObject(Object proxy) throws Exception {
|
||||
Field h = proxy.getClass().getSuperclass().getDeclaredField("h");
|
||||
h.setAccessible(true);
|
||||
AopProxy aopProxy = (AopProxy) h.get(proxy);
|
||||
|
||||
Field advised = aopProxy.getClass().getDeclaredField("advised");
|
||||
advised.setAccessible(true);
|
||||
|
||||
Object target = ((AdvisedSupport) advised.get(aopProxy)).getTargetSource().getTarget();
|
||||
|
||||
return target;
|
||||
}
|
||||
|
||||
}
|
||||
Reference in New Issue
Block a user