动态多源数据库架构实现

原创

已于 2023-06-06 09:26:22 修改 · 726 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#动态数据源 #多数据源 #数据源切换 #分库引擎 #多库引擎

于 2019-12-05 17:50:47 首次发布

本文介绍了一个基于Spring框架的动态多源数据库访问架构。通过创建主数据库和多个动态数据库，利用AbstractRoutingDataSource抽象类和ApplicationContextAware接口，实现了数据源的动态管理和切换。文章详细讲解了实现步骤，包括配置Spring主配置文件、编写动态数据源管理代码、线程上下文管理以及业务代码的数据源切换。并提到了涉及的表结构和后续将上传到GitHub的代码计划。

动态多源数据库架构实现样例

这是一个基于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();
        }

        super