Spring Boot 聲明式事務結合相關攔截器

我這項目的讀寫分離方式在使用ThreadLocal實現的讀寫分離在遷移後的偶發錯誤里提了，我不再說一次了，這次是有要求讀寫分離與事務部分要完全脫離配置文件，程序員折騰了很久，於是我就查了一下，由於我還是比較喜歡使用xml的方式，所以就隨便。。。（過程省略吧），然而，似乎是一定要聲明式的方式，所以，無奈之下就只好乾了。

首先，在之前的博客里提到過，我們的讀寫分離方式要求我們自己的AOP攔截器必須在事務攔截器之前執行，在配置文件的方式下很容易，在aop的配置里設置一下Order就好了。然而，Spring Boot的@EnableTransactionManagement註解中已經把這部分固定了，官方文檔似乎說它是和@Transactional配合使用的，總之幾乎沒有留下什麼插手的餘地（如果大家有好辦法，希望能告訴我一下）：

public TransactionInterceptor transactionInterceptor {
TransactionInterceptor interceptor = new TransactionInterceptor;
interceptor.setTransactionAttributeSource(transactionAttributeSource);
if (this.txManager != null) {
interceptor.setTransactionManager(this.txManager);
}
return interceptor;
}

雖然這裡可以通過上圖的方式加點什麼，但這個事務體系本身是非常獨立的，而且在啟動過程中就已經確定下來了，既然要調節它和自定義的aop的執行順序，我想只能先統一他們的執行策略。之前自定義的aop是在啟動過程中就被加入到一個攔截器的調用鏈中：

由於Spring Boot很多東西並沒有留什麼擴展的餘地（就好像前面那個new），如果完全不用配置文件，使用Spring Boot的方式，雖然沒有什麼順眼的方法，其實也還是能做的，先提個建議，能不用的情況下，最好不要用。方法是自定義一個事務攔截器，拋棄@EnableTransactionManagement，測試代碼如下，不要在意命名，圖方便直接在原來的上面改的：

@Bean(name = "newDataSourceAop")
public NewDataSourceAop newDataSourceAop{
return new NewDataSourceAop;
}

@Bean(name = "tInterceptor")
public TInterceptor tInterceptor{
return new TInterceptor;
}

/**
* 代理
* @return
*/
@Bean
public BeanNameAutoProxyCreator transactionAutoProxy {
BeanNameAutoProxyCreator autoProxy = new BeanNameAutoProxyCreator;
autoProxy.setProxyTargetClass(true);// 這個屬性為true時，表示被代理的是目標類本身而不是目標類的介面
autoProxy.setBeanNames("*ServiceImpl");
autoProxy.setInterceptorNames("newDataSourceAop", "tInterceptor");
return autoProxy;
}

注意，攔截器的順序依賴於名字字元串傳入的先後順序，@Order什麼的是完全沒用的，因為保存這些攔截器的是一個字元串數組。自定義的事務AOP Advice：

public class TInterceptor extends TransactionAspectSupport implements MethodInterceptor, Serializable {

public TInterceptor {
setTransactionAttributes(getAttrs);
}

private Properties getAttrs{
Properties attributes = new Properties;
// 新增
attributes.setProperty("create*", "PROPAGATION_REQUIRED,ISOLATION_DEFAULT");
// 修改
attributes.setProperty("update*", "PROPAGATION_REQUIRED,ISOLATION_DEFAULT");
// 刪除
attributes.setProperty("delete*", "PROPAGATION_REQUIRED,ISOLATION_DEFAULT");
//查詢
attributes.setProperty("query*", "PROPAGATION_REQUIRED,ISOLATION_DEFAULT");
return attributes;
}

@Override
public Object invoke(MethodInvocation invocation) throws Throwable {
Class targetClass = (invocation.getThis != null ? AopUtils.getTargetClass(invocation.getThis) : null);

// Adapt to TransactionAspectSupport"s invokeWithinTransaction...
return invokeWithinTransaction(invocation.getMethod, targetClass, -> invocation.proceed);
}
}

自定義的讀寫分離Advice：

@EnableConfigurationProperties(ReadDBPathProperties.class)
public class NewDataSourceAop implements MethodBeforeAdvice {
private static final Logger log = LoggerFactory.getLogger(NewDataSourceAop.class);

@Autowired
private ReadDBPathProperties readDBPathProperties;

@Override
public void before(Method method, Object[] args, Object target) throws Throwable {
String clazzName = method.getDeclaringClass.getSimpleName;
String runner = clazzName + "." + method.getName;
this.chooseDataSource(runner);
}

private void chooseDataSource(String runner){
runner += ",";
String read = readDBPathProperties.getReadPath+",";
log.info("case : read path, vo : readPath = {}", read);
int index = read.indexOf(runner);
if (index == -1){
log.info("case : choose write DB, runner : {}, tid={}", runner, Thread.currentThread.getId);
HandleDataSource.putDataSource("write");
return;
}
log.info("case : choose read DB, runner : {}, tid={}", runner, Thread.currentThread.getId);
HandleDataSource.putDataSource("read");
}
}

最後再特別說一下，關於這個功能的單元測試，這個單元測試有一點意思，如果是直接運行測試方法，啟動過程和執行過程在同一個線程是測試不出效果的，因為啟動過程中載入Bean的時候會對下圖中的resources初始化，寫入默認的數據源：

就會導致後續執行的讀寫分離攔截器失效，只要保證執行線程和啟動線程不在同一線程就好。

==========================================================

咱最近用的github：https://github.com/saaavsaaa

喜歡這篇文章嗎？立刻分享出去讓更多人知道吧！