1. 首先在配置文件中配置多个dataSource
2. 扩展spring的AbstractRoutingDataSource抽象类，实现动态数据源。
AbstractRoutingDataSource中的抽象方法determineCurrentLookupKey是实现数据源的route的核心.这里对该方法进行Override。
|
1 2 3 4 5 6 7 |
public class DynamicDataSource extends AbstractRoutingDataSource{ @Override protected Object determineCurrentLookupKey() { return DBContextHolder.getDBType(); } } |
上下文DbContextHolder为一线程安全的ThreadLocal，具体代码如下：
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
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映射。
|
1 2 3 4 5 6 7 8 9 10 |
<bean id="dynamicDataSource" class="datasource.DynamicDataSource" > <!-- 通过key-value的形式来关联数据源 --> <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使用。
|
1 2 3 4 5 6 7 8 9 10 11 12 13 |
<!-- JdbcTemplate使用动态数据源的配置 --> <bean id="jdbcTemplate" class="org.springframework.jdbc.core.JdbcTemplate"> <property name="dataSource"> <ref bean="dynamicDataSource" /> </property> </bean> <!-- 对JdbcTemplate的应用封装类 --> <bean id="sqlBaseDAO" class="com.whty.dao.BaseDAOImpl"> <property name="jdbcTemplate"> <ref bean="jdbcTemplate" /> </property> </bean> |
5. 动态数据源的管理
如何选择控制每个业务中需要的具体数据源，可是使用手动控制：
|
1 2 3 4 5 6 7 8 9 10 11 12 |
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(); } |
如果在service层有比较统一的规则的话，也可以使用aop设置数据源使用。
6. 这里一般都是一个service一个数据源，所以最好使用aop在service层执行完之后统一调用
|
1 |
DBContextHolder.clearDBType(); |
清空数据源信息。
当然，在上面配置里面有个参数defaultTargetDataSource为默认数据源，就是不设置数据源的话，就是用这个数据源。
©原创文章，转载请注明来源： 赵伊凡's Blog
©本文链接地址: Spring配置多数据源
事务处理怎么控制？