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生成的動態代理類