SpringBoot+Mybatis 实现动态数据源切换方案

前端之家收集整理的这篇文章主要介绍了SpringBoot+Mybatis 实现动态数据源切换方案前端之家小编觉得挺不错的,现在分享给大家,也给大家做个参考。

背景

最近让我做一个大数据的系统,分析了一下,麻烦的地方就是多数据源切换抽取数据。考虑到可以跨服务器跨数据库抽数,再整理数据,就配置了这个动态数据源的解决方案。在此分享给大家。

实现方案

数据库配置文件

我们项目使用的是yml形式的配置文件,采用的是hikari的数据库连接池。第一步我们自然是配置多个数据库源头。
我们找到spring的datasource,在下方配置三个数据源。

  1. spring:
  2. application:
  3. name: dynamicDatasource
  4. datasource:
  5. test1:
  6. driver-class-name: com.MysqL.jdbc.Driver
  7. url: jdbc:MysqL://127.0.0.1:3306/test1?serverTimezone=GMT%2B8&characterEncoding=utf-8&useSSL=false
  8. username: root
  9. password: 123456
  10. test2:
  11. driver-class-name: com.MysqL.jdbc.Driver
  12. url: jdbc:MysqL://127.0.0.1:3306/test2?serverTimezone=GMT%2B8&characterEncoding=utf-8&useSSL=false
  13. username: root
  14. password: 123456
  15. test3:
  16. driver-class-name: com.MysqL.jdbc.Driver
  17. url: jdbc:MysqL://127.0.0.1:3306/test3?serverTimezone=GMT%2B8&characterEncoding=utf-8&useSSL=false
  18. username: root
  19. password: 123456
  20. hikari:
  21. leak-detection-threshold: 2000

定义数据源实体类

我们可以建立个datasourceBean文件夹专门管理数据源的实体类。
我们这里要建立三个实体类。分别对应test1,test2,test3

  1. @Configuration
  2. public class Test1DataSourceBean {
  3. @Value("${spring.datasource.test1.driver-class-name}")
  4. private String test1Driver;
  5. @Value("${spring.datasource.test1.url}")
  6. private String test1Url;
  7. @Value("${spring.datasource.test1.username}")
  8. private String test1Username;
  9. @Value("${spring.datasource.test1.password}")
  10. private String test1Password;
  11. @Bean(name="test1DataSource")
  12. public DataSource test1DataSource() throws Exception{
  13. HikariDataSource dataSource = new HikariDataSource();
  14. dataSource.setDriverClassName(test1Driver);
  15. dataSource.setJdbcUrl(test1Url);
  16. dataSource.setUsername(test1Username);
  17. dataSource.setPassword(test1Password);
  18. return dataSource;
  19. }
  20. }
  1. @Configuration
  2. public class Test2DataSourceBean {
  3. @Value("${spring.datasource.test2.driver-class-name}")
  4. private String test2Driver;
  5. @Value("${spring.datasource.test2.url}")
  6. private String test2Url;
  7. @Value("${spring.datasource.test2.username}")
  8. private String test2Username;
  9. @Value("${spring.datasource.test2.password}")
  10. private String test2Password;
  11. @Bean(name="test2DataSource")
  12. public DataSource test2DataSource() throws Exception{
  13. HikariDataSource dataSource = new HikariDataSource();
  14. dataSource.setDriverClassName(test2Driver);
  15. dataSource.setJdbcUrl(test2Url);
  16. dataSource.setUsername(test2Username);
  17. dataSource.setPassword(test2Password);
  18. return dataSource;
  19. }
  20. }
  1. @Configuration
  2. public class Test3DataSourceBean {
  3. @Value("${spring.datasource.test3.driver-class-name}")
  4. private String test3Driver;
  5. @Value("${spring.datasource.test3.url}")
  6. private String test3Url;
  7. @Value("${spring.datasource.test3.username}")
  8. private String test3Username;
  9. @Value("${spring.datasource.test3.password}")
  10. private String test3Password;
  11. @Bean(name="test3DataSource")
  12. public DataSource test3DataSource() throws Exception{
  13. HikariDataSource dataSource = new HikariDataSource();
  14. dataSource.setDriverClassName(test3Driver);
  15. dataSource.setJdbcUrl(test3Url);
  16. dataSource.setUsername(test3Username);
  17. dataSource.setPassword(test3Password);
  18. return dataSource;
  19. }
  20. }

定义一个枚举类管理数据源

  1. public enum DatabaseType {
  2. test1("test1","test1"),test2("test2","test2"),test3("test3","test3");
  3. private String name;
  4. private String value;
  5. DatabaseType(String name,String value){
  6. this.name = name;
  7. this.value = value;
  8. }
  9. public String getName(){
  10. return name;
  11. }
  12. public String getValue(){
  13. return value;
  14. }
  15. }

定义一个线程安全的数据源容器

  1. public class DatabaseContextHolder {
  2. private static final ThreadLocal<DatabaseType> contextHolder = new ThreadLocal<>();
  3. public static void setDatabaseType(DatabaseType type){
  4. contextHolder.set(type);
  5. }
  6. public static DatabaseType getDatabaseType(){
  7. return contextHolder.get();
  8. }
  9. }

定义动态数据源

  1. public class DynamicDataSource extends AbstractRoutingDataSource{
  2. protected Object determineCurrentLookupKey() {
  3. return DatabaseContextHolder.getDatabaseType();
  4. }
  5. }

mybatis配置类

网上的很多文章配置出来都会产生数据源循环依赖的问题,这里解决了这个问题。

  1. @Configuration
  2. @MapperScan(basePackages="cn.test.jichi",sqlSessionFactoryRef="sessionFactory")
  3. public class MybatisConfig {
  4. /**
  5. * @Description:设置动态数据源
  6. */
  7. @Bean(name="dynamicDataSource")
  8. @Primary
  9. public DynamicDataSource DataSource(
  10. @Qualifier("test1DataSource") DataSource test1DataSource,@Qualifier("test2DataSource") DataSource test2DataSource,@Qualifier("test3DataSource") DataSource test3DataSource){
  11. Map<Object,Object> targetDataSource = new HashMap<>();
  12. targetDataSource.put(DatabaseType.test1,test1DataSource);
  13. targetDataSource.put(DatabaseType.test2,test2DataSource);
  14. targetDataSource.put(DatabaseType.test3,test3DataSource);
  15. DynamicDataSource dataSource = new DynamicDataSource();
  16. dataSource.setTargetDataSources(targetDataSource);
  17. dataSource.setDefaultTargetDataSource(test1DataSource);
  18. return dataSource;
  19. }
  20. /**
  21. * @Description:根据动态数据源创建sessionFactory
  22. */
  23. @Bean(name="sessionFactory")
  24. public sqlSessionFactory sessionFactory(
  25. @Qualifier("test1DataSource") DataSource test1DataSource,@Qualifier("test3DataSource") DataSource test3DataSource) throws Exception{
  26. sqlSessionfactorybean sessionfactorybean = new sqlSessionfactorybean();
  27. //构造方法解决动态数据源循环依赖问题。
  28. sessionfactorybean.setDataSource(this.DataSource(test1DataSource,test2DataSource,test3DataSource));
  29. return sessionfactorybean.getObject();
  30. }
  31. }
  1. package cn.chinaunicom.sdsi.utils;
  2. import cn.chinaunicom.sdsi.config.dynamicDataSourceConfig.DatabaseContextHolder;
  3. import cn.chinaunicom.sdsi.config.dynamicDataSourceConfig.DatabaseType;
  4. /**
  5. * @Description:封装数据源选择
  6. */
  7. public class DynamicDataSourceUtils {
  8. public static void chooseBasicDataSource(){
  9. DatabaseContextHolder.setDatabaseType(DatabaseType.basic);
  10. }
  11. public static void chooseBranchDataSource(){
  12. DatabaseContextHolder.setDatabaseType(DatabaseType.branch);
  13. }
  14. }

提供一个示例

  1. public void testDymnaicDatasource(){
  2. //不切换数据源默认是自己的。
  3. System.out.println("-----默认数据源");
  4. DemoEntity totalCount = demoMapper.getTotalCount();
  5. String nameCount1 = totalCount.getNameCount();
  6. String ageCount2 = totalCount.getAgeCount();
  7. System.out.println("nameCount:"+nameCount1);
  8. System.out.println("ageCount:"+ageCount2);
  9. //数据源切换为branch
  10. System.out.println("-----数据源为test2");
  11. DynamicDataSourceUtils.chooseBranchDataSource();
  12. Integer nameCount = demoMapper.getNameCount();
  13. Integer ageCount = demoMapper.getAgeCount();
  14. System.out.println("nameCount:"+nameCount);
  15. System.out.println("ageCount:"+ageCount);
  16. //数据源为basic
  17. System.out.println("-----数据源为test3");
  18. DynamicDataSourceUtils.chooseBasicDataSource();
  19. Integer ageCount1 = demoMapper.getAgeCount();
  20. System.out.println("ageCount:"+ageCount1);
  21. }

总结

至此实现了多数据源的动态切换。可以在同一个方法里面进行操作多个数据源。

猜你在找的Mybatis相关文章