Spring动态添加使用数据源
单位项目有个需求,我描述一下,希望会的人 ,帮忙指导一下
公司注册时,系统给该公司创建一个数据库,同时使用该数据库创建数据源,动态加载到内存.
公司用户登录时,根据登陆信息,判断该公司对应的数据源,切换到使用对应的数据源.该登录的用户操作过程中产生的业务数据都保存在对应的数据库中.
下边是我的实现思路,参考http://www.iteye.com/topic/72486
1.相关的配置文件
2.用户注册时,产生一个spring配置文件,该文件就是一个数据源的配置信息,如图:
配置文件产生后,新建一个数据,最后将该配置文件加载到内存中,代码如图:
3.编写一个获得和设置上下文环境的类，主要负责改变上下文数据源的名称
4.编写一个动态数据源类
public class DynamicDataSource extends AbstractRoutingDataSource {
/*
* 该方法必须要重写 方法是为了根据数据库标示符取得当前的数据库
*/
@Override
public Object determineCurrentLookupKey() {
DataSourceType dataSourceType= DataSourceContextHolder.getDataSourceType();
return dataSourceType;
}
@Override
public void setDataSourceLookup(DataSourceLookup dataSourceLookup) {
super.setDataSourceLookup(dataSourceLookup);
}
@Override
public void setDefaultTargetDataSource(Object defaultTargetDataSource) {
super.setDefaultTargetDataSource(defaultTargetDataSource);
}
@Override
public void setTargetDataSources(Map targetDataSources) {
super.setTargetDataSources(targetDataSources);
}
}@Test
public void test() {
try {
List<UserinfoEntity> list = userInfoService.getList(UserinfoEntity.class);
System.out.println(list.size());
String dataSourceName = "qqq";
String url = "jdbc:mysql://127.0.0.1:3306/mos_saas_test";
String username = "root";
String password = "root";
ConfigFileCreator.createConfigFile(dataSourceName, url, username, password);
DynamicLoadBean dynamicBeanLoad =(DynamicLoadBean)SpringContextUtil.getBean("dynamicLoadBean");
dynamicBeanLoad.loadBean("classpath:mos/qqq.xml");
//dynamicBeanLoad.registBean(dataSourceName, "parentDataSource");
DruidDataSource dds = (DruidDataSource) SpringContextUtil.getBean(dataSourceName);
DynamicDataSource ds = (DynamicDataSource)SpringContextUtil.getBean("dataSource");
//Map targetDataSources = new HashMap();
//targetDataSources.put(DataSourceType.qqq, dds);
//
//ds.setTargetDataSources(targetDataSources);
//DataSourceLookup dataSourceLookup = new MapDataSourceLookup();
//dataSourceLookup.getDataSource("qqq");
//
//ds.setDataSourceLookup(dataSourceLookup);
DataSourceContextHolder.setDataSourceType(DataSourceType.qqq);
List<UserinfoEntity> list2 = userInfoService.getList(UserinfoEntity.class);
System.out.println(list2.size());
} catch (Exception e) {
// TODO Auto-generated catch block
e.printStackTrace();
}
}