SpringBoot中使用LoadTimeWeaving技術實現AOP功能


1.1 LTW與不同的切面織入時機

AOP——面向切面編程,通過為目標類織入切面的方式,實現對目標類功能的增強。按切面被織如到目標類中的時間划分,主要有以下幾種:

  • 1.運行期織入
    這是最常見的,比如在運行期通過為目標類生成動態代理的方式實現AOP就屬於運行期織入,這也是Spring AOP中的默認實現,並且提供了兩種創建動態代理的方式:JDK自帶的針對接口的動態代理和使用CGLib動態創建子類的方式創建動態代理。

  • 2.編譯期織入
    使用特殊的編譯器在編譯期將切面織入目標類,這種比較少見,因為需要特殊的編譯器的支持。

  • 3.類加載期織入
    通過字節碼編輯技術在類加載期將切面織入目標類中,這是本篇介紹的重點。它的核心思想是:在目標類的class文件被JVM加載前,通過自定義類加載器或者類文件轉換器將橫切邏輯織入到目標類的class文件中,然后將修改后class文件交給JVM加載。這種織入方式可以簡稱為LTW(LoadTimeWeaving)。

1.2 JDK實現LTW的原理

可以使用JKD的代理功能讓代理器訪問到JVM的底層組件,借此向JVM注冊類文件轉換器,在類加載時對類文件的字節碼進行轉換。具體而言,java.lang.instrument包下定義了ClassFileTransformer接口,該接口的作用如下面的注釋所描述

* An agent provides an implementation of this interface in order
* to transform class files.
* The transformation occurs before the class is defined by the JVM.

可以通過實現該接口,並重寫如下抽象方法自定義類文件轉換規則

byte[]
transform(  ClassLoader         loader,
            String              className,
            Class<?>            classBeingRedefined,
            ProtectionDomain    protectionDomain,
            byte[]              classfileBuffer)
    throws IllegalClassFormatException;

classfileBuffer是原始類文件對應的字節碼數組,返回的byte[]為轉化后的字節碼數組,如果返回null,則表示不進行字節碼處理。

而java.lang.instrument包下的Instrumentation接口則可以將我們自定義的ClassTransFormer向JVM內部的組件進行注冊

void
addTransformer(ClassFileTransformer transformer);

在實際使用中,可以通過JVM的-javaagent代理參數在啟動時獲取JVM內部組件的引用,將ClassFileTransformer實例注冊到JVM中,JVM在加載Class文件時,會先調用這個ClassTransformer的transform()方法對Class文件的字節碼進行轉換,比如織入切面中定義的橫切邏輯,實現AOP功能。整個過程可以入下所示

1.3 如何在Spring中實現LTW

Spring中默認通過運行期生成動態代理的方式實現切面的織入,實現AOP功能,但是Spring也可以使用LTW技術來實現AOP,並且提供了細粒度的控制,支持在單個ClassLoader范圍內實施類文件轉換。

Spring中的org.springframework.instrument.classloading.LoadTimeWeaver接口定義了為類加載器添加ClassFileTransfomer的抽象

* Defines the contract for adding one or more
* {@link ClassFileTransformer ClassFileTransformers} to a {@link ClassLoader}.
*
public interface LoadTimeWeaver {

Spring的LTW支持AspectJ定義的切面,既可以是直接使用AspectJ語法定義的切面,也可以是使用@AspectJ注解,通過java類定義的切面。Spring LTW通過讀取classpath下META-INF/aop.xml文件,獲取切面類和要被切面織入的目標類的相關信息,通過LoadTimeWeaver在ClassLoader加載類文件時將切面織入目標類中,其工作原理如下所示

Spring中可以通過LoadTimeWeaver將Spring提供的ClassFileTransformer注冊到ClassLoader中。在類加載期,注冊的ClassFileTransformer讀取類路徑下META-INF/aop.xml文件中定義的切面類和目標類信息,在目標類的class文件真正被VM加載前織入切面信息,生成新的Class文件字節碼,然后交給VM加載。因而之后創建的目標類的實例,就已經實現了AOP功能。

2. Springboot中使用LTW實現AOP的例子

實現一個簡單的AOP需求,在方法調用前后打印出開始和結束的日志信息。

  • 相關的maven依賴和插件
<dependencies>
    <dependency>
        <groupId>org.springframework.boot</groupId>
        <artifactId>spring-boot-starter-aop</artifactId>
    </dependency>

    <dependency>
        <groupId>org.springframework.boot</groupId>
        <artifactId>spring-boot-starter-test</artifactId>
        <scope>test</scope>
    </dependency>

</dependencies>

<build>
    <plugins>
        <plugin>
            <groupId>org.apache.maven.plugins</groupId>
            <artifactId>maven-surefire-plugin</artifactId>
            <configuration>
                <argLine>
                    -javaagent:"${settings.localRepository}/org/aspectj/aspectjweaver/${aspectj.version}/aspectjweaver-${aspectj.version}.jar"
                    -javaagent:"${settings.localRepository}/org/springframework/spring-instrument/${spring.version}/spring-instrument-${spring.version}.jar"
                    <!-- -Dspring.profiles.active=test-->
                </argLine>
            </configuration>
        </plugin>
        <plugin>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-maven-plugin</artifactId>
            <configuration>
                <agent>
                    ${settings.localRepository}/org/aspectj/aspectjweaver/${aspectj.version}/aspectjweaver-${aspectj.version}.jar
                </agent>
                <agent>
                    ${settings.localRepository}/org/springframework/spring-instrument/${spring.version}/spring-instrument-${spring.version}.jar
                </agent>
            </configuration>
        </plugin>
    </plugins>

</build>

這里通過maven插件的方式為JVM設置代理,通過-javaagent參數指定織入器類包的路徑,這樣就可以在類加載期將切面織入,更多關於javaagent的知識可以參考javaagent

  • 織入目標類
/**
 * @author: takumiCX
 * @create: 2018-12-19
 **/
@Component
public class LtwBean {


    public void test(){
        System.out.println("process.......");
    }

}

只有一個test()方法,通過@Componet注解向容器注冊。

  • 切面類
/**
 * @author: takumiCX
 * @create: 2018-12-19
 **/
@Aspect
public class LogMethodInvokeAspect {

    @Pointcut("execution(public * com.takumiCX.ltw.*.*(..))")
    public void pointCut(){

    }


    @Around("pointCut()")
    public void advise(ProceedingJoinPoint pjp) throws Throwable {


        Signature signature = pjp.getSignature();
        System.out.println(signature+" start..... ");
        pjp.proceed();
        System.out.println(signature+" end......");
    }

}

@Aspect注解表示這是一個切面類

  • 配置類
/**
 * @author: takumiCX
 * @create: 2018-12-19
 **/
@Configuration
@ComponentScan("com.takumiCX.ltw")
@EnableLoadTimeWeaving(aspectjWeaving=AUTODETECT)
public class CustomLtwConfig{
}

通過@@EnableLoadTimeWeaving開啟LTW功能,可以通過屬性aspectjWeaving指定LTW的開啟策略

  1. ENABLED
    開啟LTW
  2. DISABLED
    不開啟LTW
  3. AUTODETECT
    如果類路徑下能讀取到META-INF/aop.xml文件,則開啟LTW,否則關閉
  • 在META-INF文件夾下編寫aop.xml文件

aop.xml文件內容

<?xml version="1.0" encoding="UTF-8"?>
<!DOCTYPE aspectj PUBLIC "-//AspectJ//DTD//EN" "http://www.eclipse.org/aspectj/dtd/aspectj.dtd">
<aspectj>

    <!--要織入切面的目標類-->
    <weaver>
        <include within="com.takumiCX.ltw..*" />
    </weaver>
    <!--切面類-->
    <aspects>
        <aspect name="com.takumiCX.ltw.aspect.LogMethodInvokeAspect" />
    </aspects>
</aspectj>

這樣我們的Spring容器就能加載該文件讀取到描述目標類和切面類的相關信息,容器在加載目標類的class文件到jvm之前,會將切面類中定義的增強邏輯織入到class文件中,真正加載到jvm中的是織入切面后的class文件,因而通過該class文件創建出的目標類的實例,不需要經過動態代理就能實現AOP相關功能。

  • 測試類
/**
 * @author: takumiCX
 * @create: 2018-12-20
 **/
@RunWith(SpringRunner.class)
@SpringBootTest(classes ={CustomLtwConfig.class})
public class LTWTest {

    @Autowired
    private LtwBean ltwBean;


    @Test
    public void testLTW() throws InterruptedException {

        ltwBean.test();

    }

}

最后的結果如下

方法調用前后分別記錄的開始和結束的日志信息,說明我們的切面成功的織入到了目標類。但是這里可能有一個疑問,這真的是LTW(Load TimeWeaving)通過在類加載期織入切面起到的作用嗎?有沒有可能是LTW沒起作用,是Spring AOP默認通過運行期生成動態代理的方式實現的AOP。
我們的LogMethodInvokeAspect切面類上並沒有加@Component注解向容器注冊,並且配置類CustomLtwConfig上也沒有加@EnableAspectJAutoProxy注解開啟Aspectj的運行時動態代理,所以這里基於動態代理的AOP並不會生效。

為了驗證我們的想法,將aop.xml文件刪除

重新運行測試代碼

AOP沒起到作用,說明剛才的AOP功能確實是通過LTW技術實現的。

當我們給切面類加上@Component注解,給配置類加上@EnableAspectJAutoProxy

/**
 * @author: takumiCX
 * @create: 2018-12-19
 **/
@Aspect
@Component
public class LogMethodInvokeAspect {
/**
 * @author: takumiCX
 * @create: 2018-12-19
 **/
@Configuration
@ComponentScan("com.takumiCX.ltw")
@EnableAspectJAutoProxy
@EnableLoadTimeWeaving(aspectjWeaving=AUTODETECT)
public class CustomLtwConfig{
}

再次運行測試類時,發現AOP又生效了,這時候類路徑下並沒有aop.xml,所以這時候AOP是Spring在運行期通過動態代理的方式實現的。

3. 參考資料

《精通Spring4.x企業應用開發實戰》
《spring揭秘》
https://sexycoding.iteye.com/blog/1062372


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM