手动AOP的实现的几种方式

静态AOP

在编译期,切面直接以字节 码的形式编译到目标字节 码文件中。

AspectJ属于静态AOP,是在编译时进行增强,会在编译的时候将AOP逻辑织入到代码中,需要专有的编译器和织入器。

  • 优点:被织入的类性能不受影响。
  • 缺点:不够灵活

动态AOP

JDK动态代理

在运行期,目标类加载后,为接口动态生成代理类,将切面植入到代理类中。

Java从1.3引入动态代理。实现原理是为被代理的业务接口生成代理类,将AOP逻辑写入到代理类中,在运行时动态织入AOP,使用反射执行织入的逻辑。 
主要实现方式依赖java.lang.reflect包下的InvocationHandler和Proxy类。

  • 优点:Java标准库原生支持,使用简单,无需引用额外的包。相对于静态AOP更灵活。
  • 缺点:带代理的类必须是接口,灵活性受到一些限制;使用反射会影响一些性能。

考虑对如下接口在调用前增加日志打印功能

public interface BarService {
    void doSomething();
}

业务类,需要代理的类

public class BarServiceImp implements BarService {
    @Override
    public void doSomething() {
        System.out.println("执行原有业务逻辑...");
    }
}

织入AOP逻辑,打印日志

public class LogPrintHandler implements InvocationHandler {
    private Object target;

    public LogPrintHandler(Object target) {
        this.target = target;
    }

    public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {
        System.out.println("执行AOP织入的代码!!!");
        Object ret = method.invoke(this.target, args);
        return ret;
    }
}

执行

public class App {
    public static void main(String[] args) {
        BarService service = (BarService) Proxy.newProxyInstance(
                Thread.currentThread().getContextClassLoader(), new Class[]{BarService.class}, new LogPrintHandler(new BarServiceImp()));
        service.doSomething();
    }
}

输出

执行AOP织入的代码!!!
执行原有业务逻辑...

动态代码字节生成

在运行期,目标类加载后,动态构建字节码文件生成目标类的子类,将切面逻辑加入到子类中。

CGLib是动态代码字节生成的实现,它封装字节码生成工具Asm,原理是在运行期间目标字节码加载后,生成目标类的子类,将切面逻辑加入到子类中,所以使用Cglib实现AOP不需要基于接口。

  • 优点:没有接口也可以织入,灵活性高。
  • 缺点:扩展类的实例方法为final时,则无法进行织入。

还是以之前的代码做演示

public class LogEnhancer {
    private static class MyCallback implements MethodInterceptor{
        @Override
        public Object intercept(Object obj, Method method, Object[] args, MethodProxy proxy) throws Throwable {
            System.out.println("执行AOP织入的代码!!!");
            Object ret = proxy.invokeSuper(obj, args);

            return ret;
        }
    }

    public static void main(String[] args) {
        Enhancer enhancer = new Enhancer();
        enhancer.setSuperclass(BarServiceImp.class);
        enhancer.setCallback(new MyCallback());

        BarService service = (BarService) enhancer.create();
        service.doSomething();
    }

}

输出

执行AOP织入的代码!!!
执行原有业务逻辑...

自定义类加载器

在运行前,目标加载前,将切面逻辑加到目标字节码中。

可以考虑javassist来实现。Javassist 是一个编辑字节码的框架,可以让你很简单地操作字节码。它可以在运行期定义或修改Class。使用Javassist实现AOP的原理是在字节码加载前直接修改需要切入的方法。

  • 优点:可以对绝大部分类织入。
  • 缺点:如果用到了其他类加载器,则这些类将不被织入
    public static Class enhanceClass() throws Exception {
        ClassPool classPool = ClassPool.getDefault();

        //获取目标类
        CtClass targetClass = classPool.getCtClass("com.ydoing.spring.core.aop.BarServiceImp");

        //获取织入的连接点
        CtMethod doSomething = targetClass.getDeclaredMethod("doSomething");

        //织入增强代码,这里会重新加载字节码
        doSomething.insertBefore("System.out.println(\"执行AOP织入的代码!!!\");");

        return targetClass.toClass();
    }

    public static void main(String[] args) throws Exception {
        Class clazz = enhanceClass();

        BarService service = (BarService) clazz.newInstance();
        service.doSomething();
    }

输出

执行AOP织入的代码!!!
执行原有业务逻辑...

Enhancer允许为非接口类型创建一个Java代理。Enhancer动态创建了给定类型的子类但是拦截了所有的方法。和Proxy不一样的是,不管是接口还是类他都能正常工作。

来个场景模拟一下AOP

[java]  view plain  copy
  1. package cglib.enhancer;  
  2.   
  3. public class Hello {  
  4.     public String sayHello(boolean throwException) throws Exception {  
  5.         System.out.println("hello everyone!");  
  6.         if(throwException)  
  7.             throw new Exception("test exception");  
  8.         return "123";  
  9.     }  
  10. }  

ProxyFactory用于创建增强代理实现了方法拦截

[java]  view plain  copy
  1. package cglib.enhancer;  
  2.   
  3. import java.lang.reflect.Method;  
  4.   
  5. import net.sf.cglib.proxy.Enhancer;  
  6. import net.sf.cglib.proxy.MethodInterceptor;  
  7. import net.sf.cglib.proxy.MethodProxy;  
  8.   
  9. public class ProxyFactory implements MethodInterceptor {  
  10.      //要代理的原始对象  
  11.     private Object obj;  
  12.     public Object createProxy(Object target) {  
  13.         this.obj = target;  
  14.         Enhancer enhancer = new Enhancer();  
  15.         enhancer.setSuperclass(this.obj.getClass());// 设置代理目标  
  16.         enhancer.setCallback(this);// 设置回调  
  17.         enhancer.setClassLoader(target.getClass().getClassLoader());  
  18.         return enhancer.create();  
  19.     }  
  20.   
  21.     @Override  
  22.     public Object intercept(Object obj, Method method, Object[] args, MethodProxy proxy) throws Throwable {  
  23.         Object result = null;  
  24.         try {  
  25.             // 前置通知  
  26.             before();  
  27.             result = proxy.invokeSuper(obj, args);  
  28.             // 后置通知  
  29.             after();  
  30.         } catch (Exception e) {  
  31.             exception();  
  32.         }finally{  
  33.             beforeReturning();  
  34.         }  
  35.         return result;  
  36.     }  
  37.   
  38.   
  39.     private void before() {  
  40.         System.out.println("before method invoke");  
  41.     }  
  42.     private void after() {  
  43.         System.out.println("after method invoke");  
  44.     }  
  45.     private void exception() {  
  46.         System.out.println("method invoke exception");  
  47.     }  
  48.     private void beforeReturning() {  
  49.         System.out.println("before returning");  
  50.     }  
  51. }  
场景类:

[java]  view plain  copy
  1. package cglib.enhancer;  
  2.   
  3. public class EnhancerTest {  
  4.     public static void main(String[] args) throws Exception {  
  5.         Hello hello = new Hello();  
  6.         ProxyFactory cglibProxy = new ProxyFactory();  
  7.         Hello proxy = (Hello) cglibProxy.createProxy(hello);  
  8.         String result=proxy.sayHello(true);  
  9.         System.out.println(result);  
  10.     }  
  11. }  

运行结果:

[java]  view plain  copy
  1. before method invoke  
  2. hello everyone!  
  3. method invoke exception  
  4. before returning  
  5. null  
当然把true改为false之后结果变为:

[java]  view plain  copy
  1. before method invoke  
  2. hello everyone!  
  3. after method invoke  
  4. before returning  
  5. 123  

接下来我们了解下 Enhancer可以使用的Callback接口。

[java]  view plain  copy
  1. package net.sf.cglib.proxy;  
  2.   
  3. /** 
  4.  * All callback interfaces used by {@link Enhancer} extend this interface. 
  5.  * @see MethodInterceptor 
  6.  * @see NoOp 
  7.  * @see LazyLoader 
  8.  * @see Dispatcher 
  9.  * @see InvocationHandler 
  10.  * @see FixedValue 
  11.  */  
  12. public interface Callback  
  13. {  
  14. }  
Callback接口是一个空接口,没有任何契约方法,它只表示这是一个回调。接下来看它的子接口:

接口都很简单。

先来一个基础POJO类:

[java]  view plain  copy
  1. public class SampleClass {  
  2.   public String test(String input) {  
  3.     return "Hello world!";  
  4.   }  
  5. }  

FixedValue 

使用FixedValue可以很容易的替换掉方法的返回值。

[java]  view plain  copy
  1. @Test  
  2. public void testFixedValue() throws Exception {  
  3.   Enhancer enhancer = new Enhancer();  
  4.   enhancer.setSuperclass(SampleClass.class);  
  5.   enhancer.setCallback(new FixedValue() {  
  6.     @Override  
  7.     public Object loadObject() throws Exception {  
  8.       return "Hello cglib!";  
  9.     }  
  10.   });  
  11.   SampleClass proxy = (SampleClass) enhancer.create();  
  12.   assertEquals("Hello cglib!", proxy.test(null));  
  13. }  
在上面的例子中, enhancer返回一个SampleClass的子类实例,当所有的方法被调用的时候返回匿名FixedValue回调返回的值。当然了如果方法返回的不是String类型例如,hashcode()将抛出一个类型转换异常,因为"Hello cglib!"不能转换为java.lang.Number类型。

InvocationHandler

[java]  view plain  copy
  1. @Test  
  2. public void testInvocationHandler() throws Exception {  
  3.   Enhancer enhancer = new Enhancer();  
  4.   enhancer.setSuperclass(SampleClass.class);  
  5.   enhancer.setCallback(new InvocationHandler() {  
  6.     @Override  
  7.     public Object invoke(Object proxy, Method method, Object[] args)  
  8.         throws Throwable {  
  9.       if(method.getDeclaringClass() != Object.class && method.getReturnType() == String.class) {  
  10.         return "Hello cglib!";  
  11.       } else {  
  12.         throw new RuntimeException("Do not know what to do.");  
  13.       }  
  14.     }  
  15.   });  
  16.   SampleClass proxy = (SampleClass) enhancer.create();  
  17.   assertEquals("Hello cglib!", proxy.test(null));  
  18.   assertNotEquals("Hello cglib!", proxy.toString());  
  19. }  
这个回调允许你去回答调用的方法。但是你要小心使用这个回调,因为这个代理对象的所有的方法调用都会使用 InvocationHandler#invoke方法,这样的话可能导致死循环。为了避免这种情况你可以使用MethodInterceptor。

MethodInterceptor

[java]  view plain  copy
  1. @Test  
  2. public void testMethodInterceptor() throws Exception {  
  3.   Enhancer enhancer = new Enhancer();  
  4.   enhancer.setSuperclass(SampleClass.class);  
  5.   enhancer.setCallback(new MethodInterceptor() {  
  6.     @Override  
  7.     public Object intercept(Object obj, Method method, Object[] args, MethodProxy proxy)  
  8.         throws Throwable {  
  9.       if(method.getDeclaringClass() != Object.class && method.getReturnType() == String.class) {  
  10.         return "Hello cglib!";  
  11.       } else {  
  12.         return proxy.invokeSuper(obj, args);  
  13.       }  
  14.     }  
  15.   });  
  16.   SampleClass proxy = (SampleClass) enhancer.create();  
  17.   assertEquals("Hello cglib!", proxy.test(null));  
  18.   assertNotEquals("Hello cglib!", proxy.toString());  
  19.   proxy.hashCode(); // Does not throw an exception or result in an endless loop.  
  20. }  

唯一需要注意的就是proxy.invokeSuper和proxy.invoke的区别。invokeSuper是退出当前interceptor的处理,进入下一个callback处理,invoke则会继续回调该方法,如果传递给invoke的obj参数出错容易造成递归调用。

LazyLoader

尽管LazyLoader唯一的方法和FixedValue的一模一样,但是他们还是有一个根本上的区别。
相比于Dispatcher,lazyLoader在第一次获取了loadObject后,会进行缓存,后续的请求调用都会直接调用该缓存的属性。
这个是有道理的,如果你的对象是昂贵的在其创建,不知道对象什么时候被使用。请注意,一些增强的类的构造函数必须被代理对象和延迟加载的对象。因此,确保还有另一个廉价的(可能保护)构造函数获得或使用一个接口类型的代理。你可以选择提供参数的调用Enhancer#create(Object...)。

Dispatcher

Dispatcher和LazyLoader很想但是每次调用方法的时候不会存储加载的对象。这允许改变一个类的实现但不改变原来的对象。
再者说一些构造函数必须被代理和生成的对象调用。

ProxyRefDispatcher

这个类携带了代理对象的引用,通过它的签名调用。例如这允许委托方法调用代理的另一个方法。要清楚这很容易造成死循环,并且总是导致死循环如果同样的方法被ProxyRefDispatcher#loadObject(Object)调用。

NoOp

像名字所说的一样,这个类不做任何操作。相反地,它委托每个方法的调用给被增强类的方法实现。
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

ZhaoYingChao88

整理不易,还请多多支持,谢谢

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值