spring 中多数据源的配置

黎明的到来，黑暗将消失殆尽。

我喜欢冷的天气，因为他可以让我毫无顾忌的多睡一会。今天早上再微信公众好看到这样一句话，挺励志的，就拿出来跟大家share下。

I am a slow walker,but I never walk backwards.（我希望大家不是用他来安慰自己的，而是让自己更有动力向前行）

今天跟大家分享的知识是spring中的动态多数据源的修改，直接点就是用户可以任意的切换数据库。

一.首先建立一个可以修改数据源名字的一个类:DatasourceContextHolder

给大家贴代码：

public class DatasourceContextHolder {private static final ThreadLocal<String> contextHolder = new ThreadLocal<String>();public static void setDatatsource(String datasoruce) {contextHolder.set(datasoruce);}public static String getDatasource() {return (String) contextHolder.get();}public static void clearDatasource() {contextHolder.remove();}}

大家可能对

private static final ThreadLocal<String> contextHolder = new ThreadLocal<String>();<pre name="code" class="java">/**      * Retrieve the current target DataSource. Determines the      * {@link #determineCurrentLookupKey() current lookup key}, performs      * a lookup in the {@link #setTargetDataSources targetDataSources} map,      * falls back to the specified      * {@link #setDefaultTargetDataSource default target DataSource} if necessary.      * @see #determineCurrentLookupKey()      */      protected DataSource determineTargetDataSource() {          Assert.notNull(this.resolvedDataSources, "DataSource router not initialized");          Object lookupKey = determineCurrentLookupKey();          DataSource dataSource = this.resolvedDataSources.get(lookupKey);          if (dataSource == null && (this.lenientFallback || lookupKey == null)) {              dataSource = this.resolvedDefaultDataSource;          }          if (dataSource == null) {              throw new IllegalStateException("Cannot determine target DataSource for lookup key [" + lookupKey + "]");          }          return dataSource;      }

不是很明白他的具体作用，我在这里说明下：我们再这个类里声明了一个本地的线程对象，他主要是可以保存当前调用该类的一个ｓｔｒｉｎｇ，他是一个局部的变量，这样我们在不同的用户切换数据源的时候就不会有冲突。

上面我们已经实现了ｄａｔａｂａｓｅ的一个上下文的环境，现在我们要考虑的问题就是如何根据我们修改的值把他ｓｅｔ到ｄｂ的ｃｏｎｎｅｃｔｉｏｎ呢？下面我们就介绍这部分内容。

二．内部修改ＤＢＳｏｕｒｃｅ的一个实现

首先上代码，然后解释：

import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource;public class CustomerRoutingDataSource extends AbstractRoutingDataSource{@Overrideprotected Object determineCurrentLookupKey() {// TODO Auto-generated method stubreturn DatasourceContextHolder.getDatasource();}}

上面是建立动态数据源类，我们必须要继承ＡｂｓｔｒａｃｔＲｏｕｔｉｎｇＤａｔａＳｏｕｒｃｅ，并且要实现determineCurrentLookupKey方法。这是为什么呢，我们一步一步来，我先从概念上说明下： 在Spring 2.0.1中引入了AbstractRoutingDataSource, 该类充当了DataSource的路由中介, 能有在运行时, 根据某种key值来动态切换到真正的DataSource上。

而在我们继承的AbstractRoutingDataSource类中，他同样继承了AbstractDataSource（ｊａｖａ．ｓｑｌ．ｄａｔａｓｏｕｒｃｅ里面的一个类），我们发现在他的ｇｅｔｃｏｎｎｅｃｔｉｏｎ方法中返回了一个值determineTargetDataSource（），然后我们继续深入发现了好东西，我展示代码给大家看：

/**      * Retrieve the current target DataSource. Determines the      * {@link #determineCurrentLookupKey() current lookup key}, performs      * a lookup in the {@link #setTargetDataSources targetDataSources} map,      * falls back to the specified      * {@link #setDefaultTargetDataSource default target DataSource} if necessary.      * @see #determineCurrentLookupKey()      */      protected DataSource determineTargetDataSource() {          Assert.notNull(this.resolvedDataSources, "DataSource router not initialized");          Object lookupKey = determineCurrentLookupKey();          DataSource dataSource = this.resolvedDataSources.get(lookupKey);          if (dataSource == null && (this.lenientFallback || lookupKey == null)) {              dataSource = this.resolvedDefaultDataSource;          }          if (dataSource == null) {              throw new IllegalStateException("Cannot determine target DataSource for lookup key [" + lookupKey + "]");          }          return dataSource;      }

在这里ｄｅｔｅｒｍｉｎｅＴａｒｇｅｔＤａｔａＳｏｕｒｃｅ（）中，我们的ｄｅｔｅｒｍｉｎｅＣｕｒｒｅｎｔＬｏｏｋｕｐＫｅｙ（）会返回一个ｄａｔａＳｏｕｒｃｅ的一个名字，然后

DataSource dataSource = this.resolvedDataSources.get(lookupKey); 

这句话会根据所给的ｌｏｏｋｕｐｋｅｙ从ｒｅｓｏｌｖｅｄＤａｔａＳｏｕｒｃｅｓ中ｍａｐ中找对应的ｋｅｙ，如果有就会配置相应的数据源，没有的没有就使用默认的数据源，下面我先把ｍａｐ中的部分内容展示给大家看：（ｅｐ：ｌｏｏｋｕｐＫｅｙ＝＂ｂ２ｂａｐｐｓｔｇ＂，ｍａｐ－－－＞ｋｅｙ＝＂ｂ２ｂａｐｐｓｔｇ＂）

<beans:bean id="dataSource" class="com.ｘｘｘ．ｘｘｘ.ｘｘｘ.ｘｘｘ.CustomerRoutingDataSource"><beans:property name="targetDataSources"><beans:map><beans:entry key="ｔｅｓｔ１" value-ref="Ｔｅｓｔ１" /><beans:entry key="ｔｅｓｔ２" value-ref="Ｔｅｓｔ２" /><beans:entry key="ｔｅｓｔ３" value-ref="Ｔｅｓｔ３" /></beans:map></beans:property><beans:property name="defaultTargetDataSource" ref="Ｔｅｓｔ１" /></beans:bean>

接下来的工作我们就比较简单了，就是在ｓｐｒｉｎｇ配置文件中配置你的多数据源。

三．编写ｓｐｒｉｎｇ中的多数据源

这里给大家放一个模板就可以了，然后你只需修改你自己的ｄａｔａｓｏｕｒｃｅ就可以了：

<beans:bean id="Ｔｅｓｔ１" class="org.springframework.jdbc.datasource.DriverManagerDataSource" abstract="true"><beans:property name="driverClassName" value="oracle.jdbc.driver.OracleDriver" /><beans:property name="username" value="ｘｘｘｘｘｘ" /><beans:property name="password" value="ｘｘｘｘｘｘ" /></beans:bean><beans:bean id="Ｔｅｓｔ２" class="org.springframework.jdbc.datasource.DriverManagerDataSource" parent="Ｔｅｓｔ２"><beans:property name="url" value="oracle.jdbc.driver.OracleDriver" /></beans:bean><beans:bean id="Ｔｅｓｔ３" class="org.springframework.jdbc.datasource.DriverManagerDataSource" parent="Ｔｅｓｔ３"><beans:property name="url" value="oracle.jdbc.driver.OracleDriver" /></beans:bean><pre name="code" class="html">　　　　　　　<beans:bean id="dataSource" class="com.ｘｘｘ．ｘｘｘ.ｘｘｘ.ｘｘｘ.CustomerRoutingDataSource"><beans:property name="targetDataSources"><beans:map><beans:entry key="ｔｅｓｔ１" value-ref="Ｔｅｓｔ１" /><beans:entry key="ｔｅｓｔ２" value-ref="Ｔｅｓｔ２" /><beans:entry key="ｔｅｓｔ３" value-ref="Ｔｅｓｔ３" /></beans:map></beans:property><beans:property name="defaultTargetDataSource" ref="Ｔｅｓｔ１" /></beans:bean>

　　　　　　　<beans:bean id="sqlSessionFactory" class="org.mybatis.spring.SqlSessionFactoryBean">
        　　　<beans:property name="dataSource" ref="dataSource" />
       　　　 <beans:property name="mapperLocations" value="classpath*:com/ＸＸＸＸ/ＸＸＸ/ＸＸＸ/ＸＸＸ/*.xml" />
    　　　　</beans:bean>

以上就全部的配置过程，下面你就可以在你的ｃｏｎｔｒｏｌｌｅｒ或者ａｃｔｉｏｎ中应用下面这句话就可以完成数据源的动态性。

DatasourceContextHolder.setDatatsource(＂ｔｅｓｔ１　ｏｒ　ｔｅｓｔ２　ｏｒ　ｔｅｓｔ３＂)；

最后让大家轻松下：

　　　一只小狗爬上你的餐桌，向一只烧鸡爬去，你大怒道：你敢对那只烧鸡怎样，我就敢对你怎样，结果小狗舔了一下鸡屁股，你昏倒，小狗乐道：小样看谁狠。