資料庫讀寫分離是最基本的DB擴充套件方式，我們可以在應用中自己實現，決定什麼業務訪問寫庫，什麼業務訪問讀庫。
Spring提供了一套機制可以宣告式地實現讀寫分離，簡化我們的開發。下面把最主要的幾塊列出來。
1，AbstractRoutingDataSource，這是Spring提供的最核心的元件，由它來管理所有的寫庫、讀庫，並統一對外提供dataSource。
在這裡，DynamicDataSource繼承了AbstractRoutingDataSource，並在配置中指明瞭所有的讀寫庫。dynamicDs_galaxy就是統一對外提供的dataSource，
之後，注入DAO、transaction管理，都將使用dynamicDs_galaxy。

點選(此處)摺疊或開啟

<bean id="dynamicDs_galaxy" class="com.sogou.earth.api.common.db.DynamicDataSource">
<property name="targetDataSources">
<map key-type="java.lang.String">
<entry key="master" value-ref="ds_galaxy_master" />
<entry key="slave" value-ref="ds_galaxy_slave" />
</map>
</property>
<property name="defaultTargetDataSource" ref="ds_galaxy_master" />
</bean>

在程式碼裡實現determineCurrentLookupKey方法，該方法就一個作用：獲取key，以便根據key獲取哪個datasource。

點選(此處)摺疊或開啟

package com.sogou.earth.api.common.db;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource;
public final class DynamicDataSource extends AbstractRoutingDataSource {
private static final Logger logger = LoggerFactory.getLogger(DynamicDataSource.class);
protected Object determineCurrentLookupKey() {
Object object = DynamicDataSourceKeyHolder.getDataSourceKey();
logger.debug("determineCurrentLookupKey:" + object);
return object;
}
}

2，DynamicDataSourceInterceptor，這是個攔截器，最終會以advice的形式攔截所有需要訪問DB的業務，它的作用是給需要DB訪問的業務分配一個key，
這個key用來determineCurrentLookupKey，最終確定訪問哪個dataSource。

點選(此處)摺疊或開啟

<bean id="dynamicDsInterceptor_galaxy" class="com.sogou.earth.api.common.db.DynamicDataSourceInterceptor">
<property name="attributeSource">
<list>
<value>query*,slave</value>
<value>count*,slave</value>
<value>find*,slave</value>
<value>get*,slave</value>
<value>list*,slave</value>
<value>*,master</value>
</list>
</property>
</bean>

程式碼：根據業務方法名來分配key。

點選(此處)摺疊或開啟

package com.sogou.earth.api.common.db;
import java.util.ArrayList;
import java.util.List;
import org.aopalliance.intercept.MethodInterceptor;
import org.aopalliance.intercept.MethodInvocation;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.util.PatternMatchUtils;
/**
* 設定資料來源KEY的攔截器
*
*/
public class DynamicDataSourceInterceptor implements MethodInterceptor {
private static final Logger logger = LoggerFactory.getLogger(DynamicDataSourceInterceptor.class);
/**
* 方法和使用資料來源key的對應關係
*/
private List<String> attributeSource = new ArrayList<String>();
public Object invoke(MethodInvocation invocation) throws Throwable {
final String methodName = invocation.getMethod().getName();
String key = null;
for (String value : attributeSource) {
String mappedName = value.split(\",\")[0];
if (isMatch(methodName, mappedName)) {
key = value.split(\",\")[1];
break;
}
}
logger.debug(\"methodName:\" + methodName);
if (null != key) {
DynamicDataSourceKeyHolder.setKey(key);
}
return invocation.proceed();
}
private boolean isMatch(String methodName, String mappedName) {
return PatternMatchUtils.simpleMatch(mappedName, methodName);
}
public List<String> getAttributeSource() {
return attributeSource;
}
public void setAttributeSource(List<String> attributeSource) {
this.attributeSource = attributeSource;
}
}

DynamicDataSourceKeyHolder裡面持有ThreadLocal，是執行緒安全的。

點選(此處)摺疊或開啟

package com.sogou.earth.api.common.db;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
public class DynamicDataSourceKeyHolder {
private static final Logger logger = LoggerFactory.getLogger(DynamicDataSourceKeyHolder.class);
private static final ThreadLocal<String> dataSourceHolder = new ThreadLocal<String>();
public static void setKey(String key) {
dataSourceHolder.set(key);
}
public static String getDataSourceKey() {
return (String) dataSourceHolder.get();
}
}

Spring實現資料庫讀寫分離

相關文章