1. 程式人生 > >springboot aop簡單示例

springboot aop簡單示例

AOP為Aspect Oriented Programming的縮寫,意為:面向切面程式設計,通過預編譯方式和執行期動態代理實現程式功能的統一維護的一種技術。AOP是Spring框架中的一個重要內容,它通過對既有程式定義一個切入點,然後在其前後切入不同的執行內容,比如常見的有:開啟資料庫連線/關閉資料庫連線、開啟事務/關閉事務、記錄日誌等。基於AOP不會破壞原來程式邏輯,因此它可以很好的對業務邏輯的各個部分進行隔離,從而使得業務邏輯各部分之間的耦合度降低,提高程式的可重用性,同時提高了開發的效率。

下面主要講兩個內容,一個是如何在Spring Boot中引入Aop功能,二是如何使用Aop做切面去統一處理Web請求的日誌。

準備工作

因為需要對web請求做切面來記錄日誌,所以先引入web模組,並建立一個簡單的hello請求的處理。

  • pom.xml中引入web模組
<dependency>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter-web</artifactId>
</dependency>
  • 實現一個簡單請求處理:通過傳入name引數,返回“hello xxx”的功能。
@RestController
public
class HelloController {
@RequestMapping(value = "/hello", method = RequestMethod.GET) @ResponseBody public String hello(@RequestParam String name) { return "Hello " + name; } }

下面,我們可以對上面的/hello請求,進行切面日誌記錄。

引入AOP依賴

在Spring Boot中引入AOP就跟引入其他模組一樣,非常簡單,只需要在pom.xml中加入如下依賴:

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

在完成了引入AOP依賴包後,一般來說並不需要去做其他配置。也許在Spring中使用過註解配置方式的人會問是否需要在程式主類中增加@EnableAspectJAutoProxy來啟用,實際並不需要。

可以看下面關於AOP的預設配置屬性,其中spring.aop.auto屬性預設是開啟的,也就是說只要引入了AOP依賴後,預設已經增加了@EnableAspectJAutoProxy

# AOP
spring.aop.auto=true # Add @EnableAspectJAutoProxy.
spring.aop.proxy-target-class=false # Whether subclass-based (CGLIB) proxies are to be created (true) as
 opposed to standard Java interface-based proxies (false).

而當我們需要使用CGLIB來實現AOP的時候,需要配置spring.aop.proxy-target-class=true,不然預設使用的是標準Java的實現。

實現Web層的日誌切面

實現AOP的切面主要有以下幾個要素:

  • 使用@Aspect註解將一個java類定義為切面類
  • 使用@Pointcut定義一個切入點,可以是一個規則表示式,比如下例中某個package下的所有函式,也可以是一個註解等。
  • 根據需要在切入點不同位置的切入內容
    • 使用@Before在切入點開始處切入內容
    • 使用@After在切入點結尾處切入內容
    • 使用@AfterReturning在切入點return內容之後切入內容(可以用來對處理返回值做一些加工處理)
    • 使用@Around在切入點前後切入內容,並自己控制何時執行切入點自身的內容
    • 使用@AfterThrowing用來處理當切入內容部分丟擲異常之後的處理邏輯
@Aspect
@Component
public class WebLogAspect {

    private Logger logger = Logger.getLogger(getClass());

    @Pointcut("execution(public * com.didispace.web..*.*(..))")
    public void webLog(){}

    @Before("webLog()")
    public void doBefore(JoinPoint joinPoint) throws Throwable {
        // 接收到請求,記錄請求內容
        ServletRequestAttributes attributes = (ServletRequestAttributes) RequestContextHolder.getRequestAttributes();
        HttpServletRequest request = attributes.getRequest();

        // 記錄下請求內容
        logger.info("URL : " + request.getRequestURL().toString());
        logger.info("HTTP_METHOD : " + request.getMethod());
        logger.info("IP : " + request.getRemoteAddr());
        logger.info("CLASS_METHOD : " + joinPoint.getSignature().getDeclaringTypeName() + "." + joinPoint.getSignature().getName());
        logger.info("ARGS : " + Arrays.toString(joinPoint.getArgs()));

    }

    @AfterReturning(returning = "ret", pointcut = "webLog()")
    public void doAfterReturning(Object ret) throws Throwable {
        // 處理完請求,返回內容
        logger.info("RESPONSE : " + ret);
    }

}

可以看上面的例子,通過@Pointcut定義的切入點為com.didispace.web包下的所有函式(對web層所有請求處理做切入點),然後通過@Before實現,對請求內容的日誌記錄(本文只是說明過程,可以根據需要調整內容),最後通過@AfterReturning記錄請求返回的物件。

通過執行程式並訪問:http://localhost:8080/hello?name=didi,可以獲得下面的日誌輸出

2016-05-19 13:42:13,156  INFO WebLogAspect:41 - URL : http://localhost:8080/hello
2016-05-19 13:42:13,156  INFO WebLogAspect:42 - HTTP_METHOD : http://localhost:8080/hello
2016-05-19 13:42:13,157  INFO WebLogAspect:43 - IP : 0:0:0:0:0:0:0:1
2016-05-19 13:42:13,160  INFO WebLogAspect:44 - CLASS_METHOD : com.didispace.web.HelloController.hello
2016-05-19 13:42:13,160  INFO WebLogAspect:45 - ARGS : [didi]
2016-05-19 13:42:13,170  INFO WebLogAspect:52 - RESPONSE:Hello didi

優化:AOP切面中的同步問題

在WebLogAspect切面中,分別通過doBefore和doAfterReturning兩個獨立函式實現了切點頭部和切點返回後執行的內容,若我們想統計請求的處理時間,就需要在doBefore處記錄時間,並在doAfterReturning處通過當前時間與開始處記錄的時間計算得到請求處理的消耗時間。

那麼我們是否可以在WebLogAspect切面中定義一個成員變數來給doBefore和doAfterReturning一起訪問呢?是否會有同步問題呢?

的確,直接在這裡定義基本型別會有同步問題,所以我們可以引入ThreadLocal物件,像下面這樣進行記錄:

@Aspect
@Component
public class WebLogAspect {

    private Logger logger = Logger.getLogger(getClass());

    ThreadLocal<Long> startTime = new ThreadLocal<>();

    @Pointcut("execution(public * com.didispace.web..*.*(..))")
    public void webLog(){}

    @Before("webLog()")
    public void doBefore(JoinPoint joinPoint) throws Throwable {
        startTime.set(System.currentTimeMillis());

        // 省略日誌記錄內容
    }

    @AfterReturning(returning = "ret", pointcut = "webLog()")
    public void doAfterReturning(Object ret) throws Throwable {
        // 處理完請求,返回內容
        logger.info("RESPONSE : " + ret);
        logger.info("SPEND TIME : " + (System.currentTimeMillis() - startTime.get()));
    }


}

優化:AOP切面的優先順序

由於通過AOP實現,程式得到了很好的解耦,但是也會帶來一些問題,比如:我們可能會對Web層做多個切面,校驗使用者,校驗頭資訊等等,這個時候經常會碰到切面的處理順序問題。

所以,我們需要定義每個切面的優先順序,我們需要@Order(i)註解來標識切面的優先順序。i的值越小,優先順序越高。假設我們還有一個切面是CheckNameAspect用來校驗name必須為didi,我們為其設定@Order(10),而上文中WebLogAspect設定為@Order(5),所以WebLogAspect有更高的優先順序,這個時候執行順序是這樣的:

  • @Before中優先執行@Order(5)的內容,再執行@Order(10)的內容
  • @After@AfterReturning中優先執行@Order(10)的內容,再執行@Order(5)的內容

所以我們可以這樣子總結:

  • 在切入點前的操作,按order的值由小到大執行
  • 在切入點後的操作,按order的值由大到小執行

文章來源:http://blog.didispace.com/springbootaoplog/