Spring 的动态数据源实现

Spring 的动态数据源实现
1. 配置多个数据源

这里以两个c3p0数据库连接池的数据源作为实例。在Spring框架下使用c3p0的数据库需要加入c3p0-0.9.1.2.jar(现在最新的)这个支持包。这里以数据同步项目为例：

数据来源库的连接池数据源配置：
Xml代码
1. <bean id="dataSourceFrom" class="com.mchange.v2.c3p0.ComboPooledDataSource">
2. <property name="driverClass" value="${jdbc.driver}" />
3. <property name="jdbcUrl" value="${jdbc.from.url}" />
4. <property name="user" value="${jdbc.from.username}" />
5. <property name="password" value="${jdbc.from.password}" />
6. <property name="autoCommitOnClose" value="true" />
7. <property name="checkoutTimeout" value="${cpool.checkoutTimeout}" />
8. <property name="initialPoolSize" value="${cpool.minPoolSize}" />
9. <property name="minPoolSize" value="${cpool.minPoolSize}" />
10. <property name="maxPoolSize" value="${cpool.maxPoolSize}" />
11. <property name="maxIdleTime" value="${cpool.maxIdleTime}" />
12. <property name="acquireIncrement" value="${cpool.acquireIncrement}" />
13. <property name="maxIdleTimeExcessConnections" value="${cpool.maxIdleTimeExcessConnections}" />
14. </bean>
数据插入库的连接池数据源配置：
Xml代码

<bean id="dataSourceTo" class="com.mchange.v2.c3p0.ComboPooledDataSource">

    <property name="driverClass" value="${jdbc.driver}" />

    <property name="jdbcUrl" value="${jdbc.to.url}" />

    <property name="user" value="${jdbc.to.username}" />

    <property name="password" value="${jdbc.to.password}" />

    <property name="autoCommitOnClose" value="true" />

    <property name="checkoutTimeout" value="${cpool.checkoutTimeout}" />

    <property name="initialPoolSize" value="${cpool.minPoolSize}" />

    <property name="minPoolSize" value="${cpool.minPoolSize}" />

    <property name="maxPoolSize" value="${cpool.maxPoolSize}" />

    <property name="maxIdleTime" value="${cpool.maxIdleTime}" />

    <property name="acquireIncrement" value="${cpool.acquireIncrement}" />

    <property name="maxIdleTimeExcessConnections" value="${cpool.maxIdleTimeExcessConnections}" />

</bean>
注意：上面url，user，password等值是从classpath下的jdbc.properties中取得的。

通过Spring获取属性文件中的值，以供配置文件使用：
Xml代码

<bean class="org.springframework.beans.factory.config.PropertyPlaceholderConfigurer">

    <property name="locations" value="classpath:jdbc.properties" />

</bean>
2. 扩展Spring的AbstractRoutingDataSource抽象类，实现动态数据源。

AbstractRoutingDataSource中的抽象方法determineCurrentLookupKey是实现数据源的route的核心.这里对该方法进行Override。
Java代码

public class DynamicDataSource extends AbstractRoutingDataSource{



    @Override

    protected Object determineCurrentLookupKey() {

        return DBContextHolder.getDBType();

    }

}
上下文DbContextHolder为一线程安全的ThreadLocal，具体代码如下：
Java代码

public class DBContextHolder{

    public static final String DATA_SOURCE_FROM = "dataSourceFrom";

    public static final String DATA_SOURCE_TO = "dataSourceTo";



    private static final ThreadLocal<String> contextHolder = new ThreadLocal<String>();



    public static void setDBType(String dbType) {

        contextHolder.set(dbType);

    }



    public static String getDBType() {

        return contextHolder.get();

    }



    public static void clearDBType() {

        contextHolder.remove();

    }

}
3.配置动态数据源

将DynamicDataSource Bean加入到Spring的上下文xml配置文件中去，同时配置DynamicDataSource的targetDataSources(多数据源目标)属性的Map映射。
Xml代码

<bean id="dynamicDataSource" class="datasource.DynamicDataSource" >

    

    <property name="targetDataSources">

        <map>

            <entry value-ref="dataSourceFrom" key="dataSourceFrom"></entry>

            <entry value-ref="dataSourceTo" key="dataSourceTo"></entry>

        </map>

    </property>

    <property name="defaultTargetDataSource" ref="dataSourceFrom" />

</bean>
4.使用动态数据源

例子中DynamicDataSource是继承与AbstractRoutingDataSource，而AbstractRoutingDataSource又是继承于org.springframework.jdbc.datasource.AbstractDataSource，AbstractDataSource实现了统一的DataSource接口，所以DynamicDataSource同样可以当一个DataSource使用。

在Spring的JdbcTemplate使用动态数据源的配置示例：
Xml代码



<bean id="jdbcTemplate" class="org.springframework.jdbc.core.JdbcTemplate">

    <property name="dataSource">

        <ref bean="dynamicDataSource" />

    </property>

</bean>





<bean id="sqlBaseDAO" class="com.whty.dao.BaseDAOImpl">

    <property name="jdbcTemplate">

        <ref bean="jdbcTemplate" />

    </property>

</bean>
在ORM框架Hibernate中的使用配置示例：
Xml代码

<bean id="sessionFactory" class="org.springframework.orm.hibernate3.annotation.AnnotationSessionFactoryBean">

    

    <property name="dataSource" ref="dynamicDataSource" />

    <property name="configLocations" value="classpath:hibernate.cfg.xml" />

    <property name="hibernateProperties">

        <props>

            <prop key="hibernate.dialect">${hibernate.dialect}</prop>

    </property>

</bean>
5.事务管理

使用动态数据源的时候，可以看出和使用单数据源的时候相比，在使用配置上几乎没有差别，在进行性事务管理配置的时候也没有差别：

使用Spring的JdbcTemplate的事务管理配置示例：
Xml代码

<bean id="transactionManager" class="org.springframework.jdbc.datasource.DataSourceTransactionManager">

    <property name="dataSource" ref="dynamicDataSource" />

</bean>



<bean id="sqlBaseDAOProxy" class="org.springframework.transaction.interceptor.TransactionProxyFactoryBean">

    <property name="transactionManager" ref="transactionManager" />

    <property name="target" ref="sqlBaseDAO" />

    <property name="transactionAttributes">

        <props>

            <prop key="insert*">PROPAGATION_REQUIRED</prop>

            <prop key="*">PROPAGATION_REQUIRED,readOnly</prop>

        </props>

    </property>

</bean>
使用Hibernate时的事务管理配置示例：
Xml代码

<tx:annotation-driven transaction-manager="transactionManager"/>



<bean id="transactionManager" class="org.springframework.orm.hibernate3.HibernateTransactionManager">

    <property name="sessionFactory" ref="sessionFactory" />

</bean>
6.动态数据源的管理控制

如何选择控制每个业务中需要的具体数据源，可是使用手动控制：
Java代码

ApplicationContext context = new ClassPathXmlApplicationContext("applicationContext.xml");

BaseDAO dao = (BaseDAO) context.getBean("sqlBaseDAO", BaseDAOImpl.class);



try {

    DBContextHolder.setCustomerType(DBContextHolder.DATA_SOURCE_FROM);

    System.err.println(dao.select("select count(*) sum from TEST t ").get(0).get("SUM"));

    DBContextHolder.setCustomerType(DBContextHolder.DATA_SOURCE_TO);

    System.err.println(dao.select("select count(*) sum from TEST t ").get(0).get("SUM"));



} catch (Exception e) {

    e.printStackTrace();

}
也可以采用AOP的控制方式：
Java代码

@Aspect

public class DynamicDataSourceAspect {

    @Pointcut("execution (public service.impl..*.*(..))")

    public void serviceExecution(){}



    @Before("serviceExecution()")

    public void setDynamicDataSource(JoinPoint jp) {

        for(Object o : jp.getArgs()) {

            //处理具体的逻辑，根据具体的境况CustomerContextHolder.setCustomerType()选取DataSource

        }

    }

}
7.总结

通过扩展Spring的AbstractRoutingDataSource可以很好的实现多数据源的rout效果，而且对扩展更多的数据源有良好的伸缩性，只要增加数据源和修改DynamicDataSource的targetDataSources属性配置就好。在数据源选择控制上，可以采用手动控制(业务逻辑并不多的时候)，也可以很好的用AOP的@Aspect在Service的入口加入一个切面@Pointcut，在@Before里判断JoinPoint的类容选定特定的数据源。
相关阅读:
【CCF】高速公路 tarjan强连通缩点
 【hihocoder】欧拉路径并查集判连通
 【CCF】地铁修建改编Dijkstra
Android仿微信朋友圈图片展示实现
 android 禁止 recycler 滑动
 android中关闭软键盘
 java Math.pow 精度丢失问题
 Centos查看端口占用情况和开启端口命令
 centos 部署 php
php undefinde function json_decode()
原文地址：https://www.cnblogs.com/lionelwu-qianqian/p/5216273.html