如何動態修改 spring aop 切面資訊？讓自動紀錄檔輸出框架更好用
業務背景
很久以前開源了一款 auto-log 自動紀錄檔列印框架。
其中對於 spring 專案，預設實現了基於 aop 切面的紀錄檔輸出。
但是發現一個問題，如果切面定義為全切範圍過大，於是 v0.2 版本就是基於註解 @AutoLog 實現的。
只有指定註解的類或者方法才會生效，但是這樣使用起來很不方便。
如何才能動態指定 pointcut，讓使用者使用時可以自定義切面範圍呢？
自定義註解切面原理
常規 aop 方式
@Aspect
@Component
@EnableAspectJAutoProxy
@Deprecated
public class AutoLogAop {
@Pointcut("@within(com.github.houbb.auto.log.annotation.AutoLog)" +
"|| @annotation(com.github.houbb.auto.log.annotation.AutoLog)")
public void autoLogPointcut() {
}
/**
* 執行核心方法
*
* 相當於 MethodInterceptor
*
* @param point 切點
* @return 結果
* @throws Throwable 異常資訊
* @since 0.0.3
*/
@Around("autoLogPointcut()")
public Object around(ProceedingJoinPoint point) throws Throwable {
// 紀錄檔增強邏輯
}
}
發現這裡的 @Pointcut 註解屬性是一個常數，無法方便地動態修改。
於是去查資料，找到了另一種更加靈活的方式。
可以指定 pointcut 的方式
我們通過 @Value 獲取屬性設定的切面值，給定預設值。這樣使用者就可以很方便的自定義。
/**
* 動態設定的切面
* 自動紀錄檔輸出 aop
* @author binbin.hou
* @since 0.3.0
*/
@Configuration
@Aspect
//@EnableAspectJAutoProxy
public class AutoLogDynamicPointcut {
/**
* 切面設定，直接和 spring 的設定對應 ${}，可以從 properties 或者設定中心讀取。更加靈活
*/
@Value("${auto.log.pointcut:@within(com.github.houbb.auto.log.annotation.AutoLog)||@annotation(com.github.houbb.auto.log.annotation.AutoLog)}")
private String pointcut;
@Bean("autoLogPointcutAdvisor")
public AspectJExpressionPointcutAdvisor autoLogPointcutAdvisor() {
AspectJExpressionPointcutAdvisor advisor = new AspectJExpressionPointcutAdvisor();
advisor.setExpression(pointcut);
advisor.setAdvice(new AutoLogAdvice());
return advisor;
}
}
當然，這裡的 Advice 和以前的 aop 不同，需要重新進行實現。
AutoLogAdvice
只需要實現 MethodInterceptor 介面即可。
/**
* 切面攔截器
*
* @author binbin.hou
* @since 0.3.0
*/
public class AutoLogAdvice implements MethodInterceptor {
@Override
public Object invoke(MethodInvocation methodInvocation) throws Throwable {
// 增強邏輯
}
}
介紹完了原理，我們一起來看下改進後的紀錄檔列印元件的效果。
spring 整合使用
完整範例參考 SpringServiceTest
maven 引入
<dependency>
<groupId>com.github.houbb</groupId>
<artifactId>auto-log-spring</artifactId>
<version>0.3.0</version>
</dependency>
註解宣告
使用 @EnableAutoLog 啟用自動紀錄檔輸出
@Configurable
@ComponentScan(basePackages = "com.github.houbb.auto.log.test.service")
@EnableAutoLog
public class SpringConfig {
}
測試程式碼
@ContextConfiguration(classes = SpringConfig.class)
@RunWith(SpringJUnit4ClassRunner.class)
public class SpringServiceTest {
@Autowired
private UserService userService;
@Test
public void queryLogTest() {
userService.queryLog("1");
}
}
- 輸出結果
資訊: public java.lang.String com.github.houbb.auto.log.test.service.impl.UserServiceImpl.queryLog(java.lang.String) param is [1]
五月 30, 2020 12:17:51 下午 com.github.houbb.auto.log.core.support.interceptor.AutoLogMethodInterceptor info
資訊: public java.lang.String com.github.houbb.auto.log.test.service.impl.UserServiceImpl.queryLog(java.lang.String) result is result-1
五月 30, 2020 12:17:51 下午 org.springframework.context.support.GenericApplicationContext doClose
切面自定義
原理解釋
spring aop 的切面讀取自 @Value("${auto.log.pointcut}")，預設為值 @within(com.github.houbb.auto.log.annotation.AutoLog)||@annotation(com.github.houbb.auto.log.annotation.AutoLog)
也就是預設是讀取被 @AutoLog 指定的方法或者類。
當然，這並不夠方便，我們希望可以想平時寫 aop 註解一樣，指定 spring aop 的掃描範圍，直接在 spring 中指定一下 auto.log.pointcut 的屬性值即可。
測試例子
我們在組態檔 autoLogConfig.properties 中自定義下包掃描的範圍：
auto.log.pointcut=execution(* com.github.houbb.auto.log.test.dynamic.service.MyAddressService.*(..))
自定義測試 service
package com.github.houbb.auto.log.test.dynamic.service;
import org.springframework.stereotype.Service;
@Service
public class MyAddressService {
public String queryAddress(String id) {
return "address-" + id;
}
}
自定義 spring 設定，指定我們定義的組態檔。springboot 啥的，可以直接放在 application.properties 中指定，此處僅作為演示。
@Configurable
@ComponentScan(basePackages = "com.github.houbb.auto.log.test.dynamic.service")
@EnableAutoLog
@PropertySource("classpath:autoLogConfig.properties")
public class SpringDynamicConfig {
}
測試
@ContextConfiguration(classes = SpringDynamicConfig.class)
@RunWith(SpringJUnit4ClassRunner.class)
public class SpringDynamicServiceTest {
@Autowired
private MyAddressService myAddressService;
@Autowired
private MyUserService myUserService;
@Test
public void queryUserTest() {
// 不會被紀錄檔攔截
myUserService.queryUser("1");
}
@Test
public void queryAddressTest() {
// 會被紀錄檔攔截
myAddressService.queryAddress("1");
}
}
開源地址
為了便於大家學習，專案已開源。
小結
這個專案很長一段時間拘泥於註解的方式，我個人用起來也不是很方便。
最近才想到了改進的方法，人還是要不斷學習進步。
關於紀錄檔最近還學到了 aspect 的編譯時增強，和基於 agent 的執行時增強，這 2 種方式都很有趣，有機會會做學習記錄。