1. 程式人生 > >Spring Boot系列——AOP配自定義註解的最佳實踐

Spring Boot系列——AOP配自定義註解的最佳實踐

AOP(Aspect Oriented Programming),即面向切面程式設計,是Spring框架的大殺器之一。

首先,我宣告下,我不是來系統介紹什麼是AOP,更不是照本宣科講解什麼是連線點、切面、通知和切入點這些讓人頭皮發麻的概念。

今天就來說說AOP的一些應用場景以及如何通過和其他特性的結合提升自己的靈活性。

AOP應用舉例

AOP的一大好處就是解耦。通過切面,我們可以將那些反覆出現的程式碼抽取出來,放在一個地方統一處理。

同時,抽出來的程式碼很多是與業務無關的,這樣可以方便開發者更加專注自己的業務邏輯的開發。

一個AOP的典型應用場景就是日誌列印。

下面是一個極端情況的Controller


@RestController
@RequestMapping("/")
public class HelloController {
    private static final Logger LOG = LoggerFactory.getLogger(HelloController.class);

    @GetMapping(value = "/index")
    public String index(HttpServletRequest request) {
        LOG.info("============列印日誌開始============");
        LOG.info("URL: " + request.getRequestURL().toString());
        LOG.info("============列印日誌結束============");
        return "hello jackie";
    }

    @GetMapping(value = "/test1")
    public String test1(HttpServletRequest request, String var1) {
        LOG.info("============列印日誌開始============");
        LOG.info("URL: " + request.getRequestURL().toString());
        LOG.info("============列印日誌結束============");
        return "test1";
    }

    @DemoAnnotation
    @GetMapping(value = "/test2")
    public String test2(HttpServletRequest request, String var1, String var2) {
        LOG.info("============列印日誌開始============");
        LOG.info("URL: " + request.getRequestURL().toString());
        LOG.info("============列印日誌結束============");
//        int i = 1/0;
        if (1<2)
            throw new IllegalArgumentException("exception");
        return "test2";
    }
}

HelloController中提供了三個Http介面,由於業務需要,所以每次進入某個方法的時候都需要列印請求的相關資訊。

當然,如果只是上面的例子,我們完全可以通過其他手段讓程式碼看著並不這麼糟糕。我們可以抽象一個列印方法,將相同的程式碼封裝在這個方法中,之後在各個方法中每次呼叫即可。

但是,這種處理方法似乎抽象的還不夠,因為我們在每個Http介面中還是要呼叫這個抽象的函式。而且,比較要命的是,這列印日誌的程式碼與其他業務程式碼顯得有些格格不入。

所以,這時候,我們想到了AOP。

如何使用AOP

在Spring Boot專案中,只需要如下幾步,就可以輕鬆上手AOP。

新增maven依賴


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

編寫切面類


@Aspect
@Component
public class DemoAspect {
    private static final Logger LOG = LoggerFactory.getLogger(DemoAspect.class);

    @Pointcut("execution(public * com.jackie.springbootdemo.controller.HelloController.test*(..))")
    public void addAdvice(){}

    @Before("addAdvice()")
    public void before(JoinPoint joinPoint){
        Object[] args = joinPoint.getArgs();
        HttpServletRequest requests = (HttpServletRequest) args[0];
        LOG.info("============列印日誌開始============");
        LOG.info("URL: " + requests.getRequestURL().toString());
        LOG.info("============列印日誌結束============");
//        LOG.info("before....");
    }
}

結果驗證對比

啟動SpringBootDemoApplication,訪問url:http://localhost:8080/test2?var1=1&var2=2

未使用切面功能列印日誌

使用切面功能列印日誌

從上面的結果展示發現,最終的效果是一樣的,但是使用切面更加簡潔,而且可複用。

如上訪問的是test2介面,如果訪問test1介面也可以走切面類實現列印日誌的需求,但是如果走index請求就不會列印日誌了。

這是為什麼呢?

AOP的侷限

在切面類DemoAspect中,我們看到了切入點的設定


@Pointcut("execution(public * com.jackie.springbootdemo.controller.HelloController.test*(..))")
public void addAdvice(){}

其中Pointcut後面的表示式是用於控制切面的有效影響範圍。

**表示式中,第一個表示返回任意型別,第二個表示任意方法名,後面的小括號表示任意引數值,這裡是以test為字首的,所以可以匹配上test1和test2方法。

注意,在第二個之前也可以再有個,即HelloController所在位置,表示任意類名,假如這裡是有兩個*.則表示包括包裡面的子包。**

好了,明白了表示式的含義,我們自然就看到了AOP的侷限性。

當我們要使用切面前,就要寫好表示式,但是專案一直在做,程式碼一直在加,那誰能保證後面接收程式碼的兄弟也正好知道這個test字首的意義這麼重大呢?

如果他非要用hello作為字首,那麼本應該匹配到的介面就匹配不上了,日誌也就不能正常列印了。

這時候,自定義註解,就能夠很好的解決這個問題。

自定義註解配合AOP

新建一個自定義註解


@Documented
@Retention(RetentionPolicy.RUNTIME)
@Target(ElementType.METHOD)
public @interface DemoAnnotation {

}

自定義註解的花樣也很多,比如可以在註解中宣告變數等,但這些不是我們這次討論的重點。

將該註解新增到Http介面test2方法上


    @DemoAnnotation
    @GetMapping(value = "/test2")
    public String test2(HttpServletRequest request, String var1, String var2) {
//        LOG.info("============列印日誌開始============");
//        LOG.info("URL: " + request.getRequestURL().toString());
//        LOG.info("============列印日誌結束============");
//        int i = 1/0;
        if (1<2)
            throw new IllegalArgumentException("exception");
        return "test2";
    }

在切面類中將切入點的表示式改為


@Pointcut("execution(public * com.jackie.springbootdemo.controller.*.*(..)) && @annotation(com.jackie.springbootdemo.annotation.DemoAnnotation)")
public void addAdvice(){}

這樣,我們不需要限制在controller類中是以test作為字首了,只要是在上面定義的類路徑下,並且掃描到註解DemoAnnotation就可以讓切面生效。

從結果可以看出,訪問http://localhost:8080/test1?var1=1並沒有經過切面處理,因為不滿足切入點中的表示式要求。

這樣做的好處在於,控制的粒度更細,也更加靈活,方便切面功能的實現和細分。

程式碼已提交至rome

如果您覺得閱讀本文對您有幫助,請點一下“推薦”按鈕,您的“推薦”將是我最大的寫作動力!如果您想持續關注我的文章,請掃描二維碼,關注JackieZheng的微信公眾號,我會將我的文章推送給您,並和您一起分享我日常閱讀過的優質文章。