springboot实现分库分表时动态切换数据库（源）

冷不防 2024-03-27 16:49 57阅读 0赞

#### springboot实现分库分表时动态切换数据库 ####

在分库分表中，当主库挂了，需要将新插入的数据插入到从库中，可以通过动态地切换数据源来实现。在Spring Boot中，可以使用AbstractRoutingDataSource实现动态数据源切换。  
具体实现步骤如下：

###### 1、继承AbstractRoutingDataSource类，重写determineCurrentLookupKey()方法，根据具体的业务规则动态切换数据源，如下所示： ######

public class DynamicDataSource extends AbstractRoutingDataSource {
        
        @Override
        protected Object determineCurrentLookupKey() {
        
            return DataSourceContextHolder.getDataSource();
        }
    }

###### 2、定义一个数据源上下文，用于保存当前使用的数据源的名称，如下所示： ######

public class DataSourceContextHolder {
        
        private static final ThreadLocal<String> contextHolder = new ThreadLocal<>();
    	//写入数据源
        public static void setDataSource(String dataSource) {
        
            contextHolder.set(dataSource);
        }
    	//获取数据源
        public static String getDataSource() {
        
            return contextHolder.get();
        }
    	//移除数据源
        public static void clearDataSource() {
        
            contextHolder.remove();
        }
    }

###### 3、在代码中，根据具体业务规则，动态地设置当前使用的数据源的名称，如下所示： ######

// 在需要切换数据源的地方，设置当前使用的数据源的名称
    DataSourceContextHolder.setDataSource("secondaryDataSource");
    
    // 执行需要访问数据库的操作
    ...
    
    // 操作完成后，清除当前使用的数据源的名称
    DataSourceContextHolder.clearDataSource();

###### 4、在配置文件中，定义数据源和动态数据源的配置，如下所示： ######

spring.datasource.primary.url=jdbc:mysql://localhost:3306/db1
    spring.datasource.primary.username=root
    spring.datasource.primary.password=123456
    spring.datasource.primary.driver-class-name=com.mysql.jdbc.Driver
    
    spring.datasource.secondary.url=jdbc:mysql://localhost:3306/db2
    spring.datasource.secondary.username=root
    spring.datasource.secondary.password=123456
    spring.datasource.secondary.driver-class-name=com.mysql.jdbc.Driver

###### 5、在代码中，使用@Primary注解标记主数据源的DataSource对象，使用@Qualifier注解指定从数据源的名称，如下所示： ######

@Configuration
    public class DataSourceConfig {
        
        @Primary
        @Bean(name = "primaryDataSource")
        @ConfigurationProperties(prefix = "spring.datasource.primary")
        public DataSource primaryDataSource() {
        
            return DataSourceBuilder.create().build();
        }
    
        @Bean(name = "secondaryDataSource")
        @ConfigurationProperties(prefix = "spring.datasource.secondary")
        public DataSource secondaryDataSource() {
        
            return DataSourceBuilder.create().build();
        }
    
        @Bean(name = "dataSource")
        public DynamicDataSource dataSource(@Qualifier("primaryDataSource") DataSource primaryDataSource,
                                             @Qualifier("secondaryDataSource") DataSource secondaryDataSource) {
        
            Map<Object, Object> targetDataSources = new HashMap<>();
            targetDataSources.put("primaryDataSource", primaryDataSource);
            targetDataSources.put("secondaryDataSource", secondaryDataSource);
    
            DynamicDataSource dataSource = new DynamicDataSource();
            dataSource.setTargetDataSources(targetDataSources);
            dataSource.setDefaultTargetDataSource(primaryDataSource);
    
            return dataSource;
        }
    }

以上就是在主库挂了时，动态地切换数据源的实现方法。需要注意的是，具体的切换方法。

#### 注意： ####

上面的代码中需要动态设置当前需要的数据源  
![1f6a2e54e1794b389202a870f74e832a.png][]  
但是我们往往需要设置一个默认的数据源，以下是对DynamicDataSource 的改造；

public class DynamicDataSource extends AbstractRoutingDataSource {
        
        // 设置默认数据源
        public void setDefaultDataSource(DataSource defaultDataSource) {
        
            super.setDefaultTargetDataSource(defaultDataSource);
        }
        
        @Override
        protected Object determineCurrentLookupKey() {
        
            // 返回当前线程绑定的数据源名称，如果为空则使用默认数据源
            return StringUtils.isNotBlank(DataSourceContextHolder.getDataSource()) ?
                   DataSourceContextHolder.getDataSource() : getDefaultTargetDataSource();
        }
    }

这样当我调用需要的数据源执行清除方法后就切换回默认的数据源；

[1f6a2e54e1794b389202a870f74e832a.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/03/27/5df6a363fd7f4b6daa973704d2c6a9a0.png