文章來(lái)自：https://www.jianshu.com/p/fddcc1a6b2d8

1. 繼承AbstractRoutingDataSource

AbstractRoutingDataSource 是spring提供的一個(gè)多數(shù)據(jù)源抽象類(lèi)。spring會(huì)在使用事務(wù)的地方來(lái)調(diào)用此類(lèi)的determineCurrentLookupKey()方法來(lái)獲取數(shù)據(jù)源的key值。我們繼承此抽象類(lèi)并實(shí)現(xiàn)此方法：

package com.ctitc.collect.manage.datasource;import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource; /*** * @author zongbo * 實(shí)現(xiàn)spring多路由配置，由spring調(diào)用 */ public class DataSourceRouter extends AbstractRoutingDataSource { // 獲取數(shù)據(jù)源名稱(chēng) protected Object determineCurrentLookupKey() { return HandleDataSource.getDataSource(); } }

2. 線(xiàn)程內(nèi)部數(shù)據(jù)源處理類(lèi)

DataSourceRouter 類(lèi)中通過(guò)HandleDataSource.getDataSource()獲取數(shù)據(jù)源的key值。此方法應(yīng)該和線(xiàn)程綁定。

package com.ctitc.collect.manage.datasource; /*** 線(xiàn)程相關(guān)的數(shù)據(jù)源處理類(lèi)* @author zongbo**/ public class HandleDataSource { // 數(shù)據(jù)源名稱(chēng)線(xiàn)程池 private static final ThreadLocal<String> holder = new ThreadLocal<String>(); /** * 設(shè)置數(shù)據(jù)源 * @param datasource 數(shù)據(jù)源名稱(chēng) */ public static void setDataSource(String datasource) { holder.set(datasource); } /** * 獲取數(shù)據(jù)源 * @return 數(shù)據(jù)源名稱(chēng) */ public static String getDataSource() { return holder.get(); } /** * 清空數(shù)據(jù)源 */ public static void clearDataSource() { holder.remove(); } }

3. 自定義數(shù)據(jù)源注解類(lèi)

對(duì)于spring來(lái)說(shuō)，注解即簡(jiǎn)單方便且可讀性也高。所以，我們也通過(guò)注解在service的方法前指定所用的數(shù)據(jù)源。我們先定義自己的注解類(lèi)，其中value為數(shù)據(jù)源的key值。

package com.ctitc.collect.manage.datasource; import java.lang.annotation.ElementType; import java.lang.annotation.Retention; import java.lang.annotation.RetentionPolicy; import java.lang.annotation.Target; /** * 數(shù)據(jù)源注解類(lèi) * @author zongbo * */ @Target({ ElementType.TYPE, ElementType.METHOD }) @Retention(RetentionPolicy.RUNTIME) public @interface DataSource { String value(); }

4. AOP 攔截service并切換數(shù)據(jù)源

指定注解以后，我們可以通過(guò)AOP攔截所有service方法，在方法執(zhí)行之前獲取方法上的注解：即數(shù)據(jù)源的key值。

package com.ctitc.collect.manage.datasource;import java.lang.reflect.Method; import java.text.MessageFormat;import org.aspectj.lang.JoinPoint; import org.aspectj.lang.annotation.After; import org.aspectj.lang.annotation.Aspect; import org.aspectj.lang.annotation.Before; import org.aspectj.lang.annotation.Pointcut; import org.aspectj.lang.reflect.MethodSignature; import org.slf4j.Logger; import org.slf4j.LoggerFactory; import org.springframework.context.annotation.EnableAspectJAutoProxy; import org.springframework.core.annotation.Order; import org.springframework.stereotype.Component; import org.springframework.util.StringUtils; /** * 切換數(shù)據(jù)源(不同方法調(diào)用不同數(shù)據(jù)源) */ @Aspect @Component @Order(1) //請(qǐng)注意：這里order一定要小于tx:annotation-driven的order，即先執(zhí)行DataSourceAspect切面，再執(zhí)行事務(wù)切面，才能獲取到最終的數(shù)據(jù)源 @EnableAspectJAutoProxy(proxyTargetClass = true) public class DataSourceAspect { static Logger logger = LoggerFactory.getLogger(DataSourceAspect.class); /** * 切入點(diǎn) service包及子孫包下的所有類(lèi) */ @Pointcut("execution(* com.ctitc.collect.service..*.*(..))") public void aspect() { } /** * 配置前置通知,使用在方法aspect()上注冊(cè)的切入點(diǎn) */ @Before("aspect()") public void before(JoinPoint point) { Class<?> target = point.getTarget().getClass(); MethodSignature signature = (MethodSignature) point.getSignature(); Method method = signature.getMethod() ; DataSource dataSource = null ; //從類(lèi)初始化 dataSource = this.getDataSource(target, method) ; //從接口初始化 if(dataSource == null){ for (Class<?> clazz : target.getInterfaces()) { dataSource = getDataSource(clazz, method); if(dataSource != null){ break ;//從某個(gè)接口中一旦發(fā)現(xiàn)注解，不再循環(huán) } } } if(dataSource != null && !StringUtils.isEmpty(dataSource.value()) ){ HandleDataSource.setDataSource(dataSource.value()); } } @After("aspect()") public void after(JoinPoint point) { //使用完記得清空 HandleDataSource.setDataSource(null); } /** * 獲取方法或類(lèi)的注解對(duì)象DataSource * @param target 類(lèi)class * @param method 方法 * @return DataSource */ public DataSource getDataSource(Class<?> target, Method method){ try { //1.優(yōu)先方法注解 Class<?>[] types = method.getParameterTypes(); Method m = target.getMethod(method.getName(), types); if (m != null && m.isAnnotationPresent(DataSource.class)) { return m.getAnnotation(DataSource.class); } //2.其次類(lèi)注解 if (target.isAnnotationPresent(DataSource.class)) { return target.getAnnotation(DataSource.class); } } catch (Exception e) { e.printStackTrace(); logger.error(MessageFormat.format("通過(guò)注解切換數(shù)據(jù)源時(shí)發(fā)生異常[class={0},method={1}]：" , target.getName(), method.getName()),e) ; } return null ; } }

5. 數(shù)據(jù)源配置

假設(shè)我有兩個(gè)庫(kù)：業(yè)務(wù)庫(kù)和訂單庫(kù)。先要配置這兩個(gè)數(shù)據(jù)源

• 業(yè)務(wù)數(shù)據(jù)源

<bean id="busiDataSource" class="com.alibaba.druid.pool.DruidDataSource" init-method="init" destroy-method="close"> <description>業(yè)務(wù)數(shù)據(jù)源</description> <!-- 數(shù)據(jù)庫(kù)基本信息配置 --> <property name="driverClassName" value="${busi.driverClassName}" /> <property name="url" value="${busi.url}" /> <property name="username" value="${busi.username}" /> <property name="password" value="${busi.password}" /> <!-- 初始化連接數(shù)量 --> <property name="initialSize" value="${druid.initialSize}" /> <!-- 最大并發(fā)連接數(shù) --> <property name="maxActive" value="${druid.maxActive}" /> <!-- 最小空閑連接數(shù) --> <property name="minIdle" value="${druid.minIdle}" /> <!-- 配置獲取連接等待超時(shí)的時(shí)間 --> <property name="maxWait" value="${druid.maxWait}" /> <!-- 超過(guò)時(shí)間限制是否回收 --> <property name="removeAbandoned" value="${druid.removeAbandoned}" /> <!-- 超過(guò)時(shí)間限制多長(zhǎng)； --> <property name="removeAbandonedTimeout" value="${druid.removeAbandonedTimeout}" /> <!-- 配置間隔多久才進(jìn)行一次檢測(cè)，檢測(cè)需要關(guān)閉的空閑連接，單位是毫秒 --> <property name="timeBetweenEvictionRunsMillis" value="${druid.timeBetweenEvictionRunsMillis}" /> <!-- 配置一個(gè)連接在池中最小生存的時(shí)間，單位是毫秒 --> <property name="minEvictableIdleTimeMillis" value="${druid.minEvictableIdleTimeMillis}" /> <!-- 用來(lái)檢測(cè)連接是否有效的sql，要求是一個(gè)查詢(xún)語(yǔ)句--> <property name="validationQuery" value="${druid.validationQuery}" /> <!-- 申請(qǐng)連接的時(shí)候檢測(cè) --> <property name="testWhileIdle" value="${druid.testWhileIdle}" /> <!-- 申請(qǐng)連接時(shí)執(zhí)行validationQuery檢測(cè)連接是否有效，配置為true會(huì)降低性能 --> <property name="testOnBorrow" value="${druid.testOnBorrow}" /> <!-- 歸還連接時(shí)執(zhí)行validationQuery檢測(cè)連接是否有效，配置為true會(huì)降低性能 --> <property name="testOnReturn" value="${druid.testOnReturn}" /> <!-- 打開(kāi)PSCache，并且指定每個(gè)連接上PSCache的大小 --> <property name="poolPreparedStatements" value="${druid.poolPreparedStatements}" /> <property name="maxPoolPreparedStatementPerConnectionSize" value="${druid.maxPoolPreparedStatementPerConnectionSize}" /> <!--屬性類(lèi)型是字符串，通過(guò)別名的方式配置擴(kuò)展插件，常用的插件有： 監(jiān)控統(tǒng)計(jì)用的filter:stat 日志用的filter:log4j 防御SQL注入的filter:wall --> <property name="filters" value="${druid.filters}" /> </bean>

• 訂單數(shù)據(jù)源

<bean id="orderDataSource" class="com.alibaba.druid.pool.DruidDataSource" parent="busiDataSource"> <description>訂單數(shù)據(jù)源</description> <property name="driverClassName" value="${order.driverClassName}" /> <property name="url" value="${order.url}" /> <property name="username" value="${order.username}" /> <property name="password" value="${order.password}" /> </bean>

• dataSource 則是剛剛實(shí)現(xiàn)的DataSourceRouter,且需要指定此類(lèi)的 targetDataSources屬性和 defaultTargetDataSource屬性。

targetDataSources :數(shù)據(jù)源列表，key-value形式，即上面配置的兩個(gè)數(shù)據(jù)源
defaultTargetDataSource:默認(rèn)數(shù)據(jù)源，如果未指定數(shù)據(jù)源 或者指定的數(shù)據(jù)源不存在的話(huà) 默認(rèn)使用這個(gè)數(shù)據(jù)源

<bean id="dataSource" class="com.ctitc.collect.manage.datasource.DataSourceRouter" lazy-init="true"> <description>多數(shù)據(jù)源路由</description> <property name="targetDataSources"> <map key-type="java.lang.String" value-type="javax.sql.DataSource"> <!-- write --> <entry key="busi" value-ref="busiDataSource" /> <entry key="order" value-ref="orderDataSource" /> </map> </property> <!-- 默認(rèn)數(shù)據(jù)源，如果未指定數(shù)據(jù)源 或者指定的數(shù)據(jù)源不存在的話(huà) 默認(rèn)使用這個(gè)數(shù)據(jù)源 --> <property name="defaultTargetDataSource" ref="busiDataSource" /> </bean>

6. AOP的順序問(wèn)題

由于我使用的注解式事務(wù)，和我們的AOP數(shù)據(jù)源切面有一個(gè)順序的關(guān)系。數(shù)據(jù)源切換必須先執(zhí)行，數(shù)據(jù)庫(kù)事務(wù)才能獲取到正確的數(shù)據(jù)源。所以要明確指定 注解式事務(wù)和 我們AOP數(shù)據(jù)源切面的先后順序。

• 我們數(shù)據(jù)源切換的AOP是通過(guò)注解來(lái)實(shí)現(xiàn)的，只需要在A(yíng)OP類(lèi)上加上一個(gè)order(1)注解即可，其中1代表順序號(hào)。
• 注解式事務(wù)的是通過(guò)xml配置啟動(dòng)

<tx:annotation-driven transaction-manager="transactionManager"proxy-target-class="true" order="2" />

?

?

7. 示例Demo

在每個(gè)service方法前使用@DataSource("數(shù)據(jù)源key")注解即可。

@Override@DataSource("busi")@Transactional(readOnly = true, propagation = Propagation.REQUIRED, rollbackFor = Exception.class) public List<BasVersion> test1() { // TODO Auto-generated method stub return coreMapper.getVersion(); } @Override @DataSource("order") @Transactional(readOnly = true, propagation = Propagation.REQUIRED, rollbackFor = Exception.class) public List<BasVersion> test2() { // TODO Auto-generated method stub return coreMapper.getVersion(); }

小禮物走一走，來(lái)簡(jiǎn)書(shū)關(guān)注我

作者：heichong
鏈接：https://www.jianshu.com/p/fddcc1a6b2d8
來(lái)源：簡(jiǎn)書(shū)
著作權(quán)歸作者所有。商業(yè)轉(zhuǎn)載請(qǐng)聯(lián)系作者獲得授權(quán)，非商業(yè)轉(zhuǎn)載請(qǐng)注明出處。

轉(zhuǎn)載于:https://www.cnblogs.com/xiaofengfeng/p/9039382.html

總結(jié)

以上是生活随笔為你收集整理的Spring MVC+Mybatis 多数据源配置的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問(wèn)題。

如果覺(jué)得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。