業(yè)務(wù)背景

很久以前開(kāi)源了一款 auto-log[1] 自動(dòng)日志打印框架。

其中對(duì)于 spring 項(xiàng)目，默認(rèn)實(shí)現(xiàn)了基于 aop 切面的日志輸出。

但是發(fā)現(xiàn)一個(gè)問(wèn)題，如果切面定義為全切范圍過(guò)大，于是 v0.2 版本就是基于注解 @AutoLog 實(shí)現(xiàn)的。

只有指定注解的類(lèi)或者方法才會(huì)生效，但是這樣使用起來(lái)很不方便。

如何才能動(dòng)態(tài)指定 pointcut，讓用戶(hù)使用時(shí)可以自定義切面范圍呢？

自定義注解切面原理

常規(guī) 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() {
    }
    /**
     * 執(zhí)行核心方法
     *
     * 相當(dāng)于 MethodInterceptor
     *
     * @param point 切點(diǎn)
     * @return 結(jié)果
     * @throws Throwable 異常信息
     * @since 0.0.3
     */
    @Around("autoLogPointcut()")
    public Object around(ProceedingJoinPoint point) throws Throwable {
        // 日志增強(qiáng)邏輯
    }
}

發(fā)現(xiàn)這里的 @Pointcut 注解屬性是一個(gè)常量，無(wú)法方便地動(dòng)態(tài)修改。

于是去查資料，找到了另一種更加靈活的方式。

可以指定 pointcut 的方式

我們通過(guò) @Value 獲取屬性配置的切面值，給定默認(rèn)值。這樣用戶(hù)就可以很方便的自定義。

/**
 * 動(dòng)態(tài)配置的切面
 * 自動(dòng)日志輸出 aop
 * @author binbin.hou
 * @since 0.3.0
 */
@Configuration
@Aspect
//@EnableAspectJAutoProxy
public class AutoLogDynamicPointcut {
    /**
     * 切面設(shè)置，直接和 spring 的配置對(duì)應(yīng) ${}，可以從 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;
    }
}

當(dāng)然，這里的 Advice 和以前的 aop 不同，需要重新進(jìn)行實(shí)現(xiàn)。

AutoLogAdvice

只需要實(shí)現(xiàn) MethodInterceptor 接口即可。

/**
 * 切面攔截器
 *
 * @author binbin.hou
 * @since 0.3.0
 */
public class AutoLogAdvice implements MethodInterceptor {
    @Override
    public Object invoke(MethodInvocation methodInvocation) throws Throwable {
        // 增強(qiáng)邏輯
    }
}

介紹完了原理，我們一起來(lái)看下改進(jìn)后的日志打印組件的效果。

spring 整合使用

完整示例參考 SpringServiceTest[2]。

maven 引入

<dependency>
    <groupId>com.github.houbb</groupId>
    <artifactId>auto-log-spring</artifactId>
    <version>0.3.0</version>
</dependency>

注解聲明

使用 @EnableAutoLog 啟用自動(dòng)日志輸出。

@Configurable
@ComponentScan(basePackages = "com.github.houbb.auto.log.test.service")
@EnableAutoLog
public class SpringConfig {
}

測(cè)試代碼

@ContextConfiguration(classes = SpringConfig.class)
@RunWith(SpringJUnit4ClassRunner.class)
public class SpringServiceTest {
    @Autowired
    private UserService userService;
    @Test
    public void queryLogTest() {
        userService.queryLog("1");
    }
}

輸出結(jié)果

信息: 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}")，默認(rèn)為值 @within(com.github.houbb.auto.log.annotation.AutoLog)||@annotation(com.github.houbb.auto.log.annotation.AutoLog)。

也就是默認(rèn)是讀取被 @AutoLog 指定的方法或者類(lèi)。

當(dāng)然，這并不夠方便，我們希望可以想平時(shí)寫(xiě) aop 注解一樣，指定 spring aop 的掃描范圍，直接在 spring 中指定一下 auto.log.pointcut 的屬性值即可。

測(cè)試?yán)?/h3>

完整測(cè)試代碼[3]

我們?cè)谂渲梦募?nbsp;autoLogConfig.properties 中自定義下包掃描的范圍：

auto.log.pointcut=execution(* com.github.houbb.auto.log.test.dynamic.service.MyAddressService.*(..))

自定義測(cè)試 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 {
}

測(cè)試

@ContextConfiguration(classes = SpringDynamicConfig.class)
@RunWith(SpringJUnit4ClassRunner.class)
public class SpringDynamicServiceTest {
    @Autowired
    private MyAddressService myAddressService;
    @Autowired
    private MyUserService myUserService;
    @Test
    public void queryUserTest() {
        // 不會(huì)被日志攔截
        myUserService.queryUser("1");
    }
    @Test
    public void queryAddressTest() {
        // 會(huì)被日志攔截
        myAddressService.queryAddress("1");
    }
}

開(kāi)源地址

為了便于大家學(xué)習(xí)，項(xiàng)目已開(kāi)源。

Github: https://github.com/houbb/auto-log

Gitee: https://gitee.com/houbinbin/auto-log

小結(jié)

這個(gè)項(xiàng)目很長(zhǎng)一段時(shí)間拘泥于注解的方式，我個(gè)人用起來(lái)也不是很方便。

最近才想到了改進(jìn)的方法，人還是要不斷學(xué)習(xí)進(jìn)步。

關(guān)于日志最近還學(xué)到了 aspect 的編譯時(shí)增強(qiáng)，和基于 agent 的運(yùn)行時(shí)增強(qiáng)，這 2 種方式都很有趣，有機(jī)會(huì)會(huì)做學(xué)習(xí)記錄。

References

[1] auto-log: https://github.com/houbb/auto-log。

[2] SpringServiceTest: https://github.com/houbb/auto-log/tree/master/auto-log-test/src/test/java/com/github/houbb/auto/log/spring/SpringServiceTest.java。

[3] 完整測(cè)試代碼: https://github.com/houbb/auto-log/blob/master/auto-log-test/src/test/java/com/github/houbb/auto/log/dynamic/SpringDynamicServiceTest.java。