SpringMVC框架中配置多数据源（测试库与正式库分离）

目的

为保护正式库数据相对安全，我们将系统中的数据库进行了分离。
开发者在windows环境开发时连接测试库，产品发布到Linux服务器上时连接正式库。

策略

鉴于开发环境与产品服务器存在系统差别，可以利用系统名称标识 os.name 来区分数据源。

步骤

• 1、在jdbc.properties中添加以不同名称标识的路径参数，如：

driver=com.mysql.jdbc.Driver#测试库url-dev=jdbc:mysql://192.168.1.11:3306/db_xx?useUnicode=true&characterEncoding=utf8&autoReconnect=trueusername-dev=userpassword-dev=password#正式库url-pro=jdbc:mysql://192.168.1.201:3306/db_xx?useUnicode=true&characterEncoding=utf8&autoReconnect=trueusername-pro=userpassword-pro=password#定义初始连接数  initialSize=0#定义最大连接数  maxActive=20#定义最大空闲  maxIdle=20#定义最小空闲  minIdle=1#定义最长等待时间  maxWait=60000

• 2、自定义AbstractRoutingDataSource，路径选择器

import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource;public class DynamicDataSource extends AbstractRoutingDataSource {    private static String dataSourceRef = "dataSource-dev"; // 数据源配置    static {        try {            String os = System.getProperties().getProperty("os.name");            if (os == null || os.length() == 0) {                os = "win";            }            if (os.toLowerCase().startsWith("win")) {                // 测试库                dataSourceRef = "dataSource-dev";            } else {                // 正式库                dataSourceRef = "dataSource-pro";            }        } catch (Exception e) {            e.printStackTrace();        }    }    @Override    protected Object determineCurrentLookupKey() {        return dataSourceRef;    }}

• 3、配置spring-mabatis.xml数据源部分

<!-- 本系统数据库配置 -->    <bean id="dataSource-base" class="org.apache.commons.dbcp.BasicDataSource"        destroy-method="close">        <property name="driverClassName" value="${driver}" />        <!-- 初始化连接大小 -->        <property name="initialSize" value="${initialSize}"></property>        <!-- 连接池最大数量 -->        <property name="maxActive" value="${maxActive}"></property>        <!-- 连接池最大空闲 -->        <property name="maxIdle" value="${maxIdle}"></property>        <!-- 连接池最小空闲 -->        <property name="minIdle" value="${minIdle}"></property>        <!-- 获取连接最大等待时间 -->        <property name="maxWait" value="${maxWait}"></property>    </bean>    <!-- 测试库 -->    <bean id="dataSource-dev" parent="dataSource-base">        <property name="url" value="${url-dev}" />        <property name="username" value="${username-dev}" />        <property name="password" value="${password-dev}" />    </bean>    <!-- 正式库 -->    <bean id="dataSource-pro"  parent="dataSource-base">        <property name="url" value="${url-pro}" />        <property name="username" value="${username-pro}" />        <property name="password" value="${password-pro}" />    </bean>    <!-- 使用自定义的路径选择器 -->    <bean id="dataSource" class="com.xxx.database.DynamicDataSource">        <property name="defaultTargetDataSource" ref="dataSource-dev"/>        <property name="targetDataSources">            <map key-type="java.lang.String">                <entry key="dataSource-dev" value-ref="dataSource-dev"></entry>                <entry key="dataSource-pro" value-ref="dataSource-pro"></entry>            </map>        </property>    </bean>    <!-- spring和MyBatis整合 -->    <bean id="sqlSessionFactory" class="org.mybatis.spring.SqlSessionFactoryBean">        <!-- 数据源 -->        <property name="dataSource" ref="dataSource" />        <property name="configLocation" value="classpath:sqlMapConfig.xml"></property>        <!-- 自动扫描mapping.xml文件 -->        <property name="mapperLocations" value="classpath:com/xxx/mapper/*.xml"></property>    </bean>    <!-- DAO接口所在包名，Spring会自动查找其下的类 -->    <bean class="org.mybatis.spring.mapper.MapperScannerConfigurer">        <property name="basePackage" value="com.xxx.dao" />        <property name="sqlSessionFactoryBeanName" value="sqlSessionFactory"></property>    </bean>    <!-- (事务管理)transaction manager -->    <bean id="transactionManager"        class="org.springframework.jdbc.datasource.DataSourceTransactionManager">        <property name="dataSource" ref="dataSource" />    </bean>

小结

在当前业务特点下，我们通过增加AbstractRoutingDataSource实例实现了数据源的动态变更，巧妙的解决了多数据源的需求。

因为数据源在系统初始化时（static方法）就已经指定，因此不会出现切换数据源引发的异常。

我们看下org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource.determineCurrentLookupKey()方法源码注释。

/**     * Determine the current lookup key. This will typically be     * implemented to check a thread-bound transaction context.     * <p>Allows for arbitrary keys. The returned key needs     * to match the stored lookup key type, as resolved by the     * {@link #resolveSpecifiedLookupKey} method.     */    protected abstract Object determineCurrentLookupKey();

大概意思是决定当前连接的数据源（lookupKey），通过本方法返回需要使用的数据源ID即可动态修改连接配置。