Spring AOP实现原理（二）实现一个简单的AOP

Spring AOP 简单示例

在了解AOP以后，咱们以注解的方式写一个Spring AOP的示例，这种例子网上不少，本文只贴一些关键性的代码

package com.aop.biz;

class BizA {
    public void doSomething() {
        //do something...
    }   
}

class BizB {
    public void doSomething() {
        //do something...
    }    
}
复制代码

定义两个业务类

import org.aspectj.lang.annotation.AfterReturning;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Before;

@Aspect
public class LogAspect {
    
    @Before("execution(** com.aop.biz.*.*(..))")
    public void before() {
        System.out.println("before");
    }

    @AfterReturning("execution(** com.aop.biz.*.*(..))")
    public void after() {
        System.out.println("after");
    }
}
复制代码

实现一个切面（@Aspect），包括一个前置处理（@Before），一个后置处理（@AfterReturning）和一个切点

（execution(** com.aop.biz.*.*(..))）
复制代码

匹配biz包下的全部方法。

固然也能够进一步抽取，将切点抽成一个方法myPointcut，其它的Advice引入这个方法

@Aspect
public class LogAspect {

    @Pointcut("execution(** com.aop.LogAspect.doSomething(..))")
    public void myPointcut() {

    }
  
    @Before("mypointcut()")
    public void before() {
        System.out.println("before");
    }

    @AfterReturning("mypointcut()")
    public void after() {
        System.out.println("after");
    }
}
复制代码

以上就是一个简单的示例，为了更进一步理解Spring AOP，下章节中咱们本身尝试实现一个AOP

本身实现一个AOP

定义注解

Aspect 注解类

@Retention(RetentionPolicy.RUNTIME)
@Target(ElementType.TYPE)
public @interface Aspect {
    public String value() default "";
}
复制代码

Before注解类（仅定义Before一种advice）

@Retention(RetentionPolicy.RUNTIME)
@Target(ElementType.METHOD)
public @interface Before {
    String value();
}
复制代码

Pointcut注解类

@Retention(RetentionPolicy.RUNTIME)
@Target(ElementType.METHOD)
public @interface Pointcut {
    String value() default "";
}
复制代码

定义一个日志切面

@Aspect
public class MyLogAspect {
    @Pointcut("com.aop.biz.*.*")
    public void pointcut() {
    }

    @Before("pointcut")
    public void beforeAdvice(Method method,Object object) {
        System.out.println("-------before------");
        System.out.println(object.getClass().getName());
    }
}
复制代码

实现AOP

final ClassPath cp = ClassPath.from(AspectLoader.class.getClassLoader());
final ImmutableSet<ClassPath.ClassInfo> allClasses = cp.getAllClasses();

//1.获取具备@Aspect注解的类
Class aspectClass = getAspectClass(allClasses.asList());

//2-1.获取before
Before before = getBefore(aspectClass);

//2-2.获取before相关的方法
Method beforeMethod = getBeforeMethod(aspectClass);

//2-3.获取切点
String methodName = before.value().replace("()", "");
Pointcut pointcut = getPointcut(aspectClass, methodName);

//3.构建advice和pointcut关系
//...

//4.定位切点目标类
List<Class> targets = getTargetClasses(allClasses.asList(),pointcut.value());

//5-1.生成aspect实例
Object aspectObject = aspectClass.newInstance();

//5-2.经过动态代理，将advice添加到代理类的interceptor中，并最终把代理对象放入容器中以供调用
for (Class c : targets) {
    Enhancer enhancer = new Enhancer();
    enhancer.setSuperclass(c);
    enhancer.setCallback(new MethodInterceptor(){
        @Override
        public Object intercept(Object obj, Method method, Object[] args, MethodProxy proxy) throws Throwable {
            // TODO Auto-generated method stub
            beforeMethod.invoke(aspectObject,method,obj);
            proxy.invokeSuper(obj, args);
            return obj;
        }
    });

    Object o = enhancer.create();
    this.beanContainer.put(c.getSimpleName(),o);
}
复制代码

前面的1到4步都是根据Java反射的特性实现的，这里再也不赘述。关键是第5步，基于cglib实现了对象的代理，将before类型的advice动态加到了目标方法调用以前。

调用

public static void main(final String[] args) throws Exception {
    AspectLoader aspectLoader = new AspectLoader();
    aspectLoader.init();
    BizA bizA = (BizA) aspectLoader.beanContainer.get(BizA.class.getSimpleName());
    bizA.doSomething();

    BizB bizB = (BizB) aspectLoader.beanContainer.get(BizB.class.getSimpleName());
    bizB.doSomething();
}
复制代码

调用结果：

-------before------
com.aop.biz.BizA$$EnhancerByCGLIB$$6fba1eda
-------before------
com.aop.biz.BizB$$EnhancerByCGLIB$$7b80fb85
复制代码

注：咱们能够看到，本来BizA类，实际上执行的时候变成了BizA$$EnhancerByCGLIB$$xxx，这个其实就是cglib生成的动态代理类