1. 首先在配置文件中配置多个dataSource
2. 扩展Spring的AbstractRoutingDataSource抽象类,实现动态数据源。
AbstractRoutingDataSource中的抽象方法determineCurrentLookupKey是实现数据源的route的核心.这里对该方法进行Override。
public class DynamicDataSource extends AbstractRoutingDataSource{
@Override
protected Object determineCurrentLookupKey() {
return DBContextHolder.getDBType();
}
}
上下文DbContextHolder为一线程安全的ThreadLocal,具体代码如下:
public class DBContextHolder{
public static final String DATA_SOURCE_FROM = "dataSourceFrom";
public static final String DATA_SOURCE_TO = "dataSourceTo";
private static final ThreadLocal contextHolder = new ThreadLocal();
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映射。
4. 使用动态数据源
DynamicDataSource是继承与AbstractRoutingDataSource,而AbstractRoutingDataSource又是继承于org.springframework.jdbc.datasource.AbstractDataSource,AbstractDataSource实现了统一的DataSource接口,所以DynamicDataSource同样可以当一个DataSource使用。
5. 动态数据源的管理
如何选择控制每个业务中需要的具体数据源,可是使用手动控制:
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层执行完之后统一调用
DBContextHolder.clearDBType();
清空数据源信息。
当然,在上面配置里面有个参数defaultTargetDataSource为默认数据源,就是不设置数据源的话,就是用这个数据源。