动态多源数据库架构实现样例
这是一个基于Spring原始框架进行拓展的数据库访问框架
首先,你必须了解Spring、Mybatis等基本框架应用。另外,你需要学习抽象类AbstractRoutingDataSource和接口ApplicationContextAware。
步骤说明
想要完成动态数据源,需要以下几个步骤:
- 创建一个主数据库和几个动态数据库,其中主数据库里只存储路由表data_source(内容是存储其他数据库的配置信息),动态数据库存储其他所有的业务实现表,如:user等;
- 配置Spring主配置文件applicationContext.xml,如:数据源配置文件、主数据源bean、动态数据源bean、主数据源Session工厂、Mapper接口、事务管理、事务切入点等;
- 编写动态数据源管理代码:DynamicDatabaseSource extends AbstractRoutingDataSource implements ApplicationContextAware,并实现其中必要的函数:determineCurrentLookupKey、afterPropertiesSet、setApplicationContext,其中增加一个initDataSources函数,作用就是查询主数据源的路由表,并按照记录创建动态数据源的BeanDefinitionBuilder对象,并覆盖关于sqlSessionFactory、scannerConfigurer和txManager的相关配置;
- 增加一个线程上下文的静态类DatabaseSourceContextHolder,实现setDataSourceType、getDataSourceType、clearDataSourceType三个静态方法;
- 业务代码设置数据源切换,然后利用已经注入的Mapper接口,完成对指定的数据库访问。
代码段一(用来查询主数据源的路由表)
import ...
@Transactional
@Service("databaseService")
public class DatabaseSourceServiceImpl implements DatabaseSourceService {
private static Logger log = LoggerFactory.getLogger(DatabaseSourceServiceImpl.class);
@Autowired
private DatabaseSourceMapper dataSourceMapper;
public List<DatabaseSource> selectAllDataSource() {
return dataSourceMapper.selectAll();
}
}
代码段二(用来初始化动态数据源)
import ...
/**
* Created by rocker on 16-12-15.
*/
public class DynamicDatabaseSource extends AbstractRoutingDataSource implements ApplicationContextAware {
private static Logger log = LoggerFactory.getLogger(DynamicDatabaseSource.class);
private static ApplicationContext context = null;
@Override
protected Object determineCurrentLookupKey() {
String datasource = DatabaseSourceContextHolder.getDataSourceType();
log.info("data source is "+datasource);
return datasource;
}
@Override
public void afterPropertiesSet() {
try {
initDataSources();
} catch (Exception e) {
e.printStackTrace();