devbox
程序自动化专家
这个屌丝很懒，什么也没留下！
关注作者
热门标签
热门文章
当前位置:   article > 正文

Spring框架下利用AbstractRoutingDataSource配置多数据源_abstractroutingdatasource 如何注入datasource

作者：程序自动化专家 | 2024-01-31 22:50:31

赞

踩

abstractroutingdatasource 如何注入datasource

首先看Druid数据库多数据源Demo

Spring的多数据源支持---AbstractRoutingDataSource，AbstractRoutingDataSource定义了抽象的determineCurrentLookupKey方法，子类实现此方法，来确定要使用的数据源

  1. public abstract class AbstractRoutingDataSource extends AbstractDataSource implements InitializingBean {
  2. 	 protected DataSource determineTargetDataSource() {
  3. 		Assert.notNull(this.resolvedDataSources, "DataSource router not initialized");
  4. 		Object lookupKey = determineCurrentLookupKey();
  5. 		DataSource dataSource = this.resolvedDataSources.get(lookupKey);
  6. 		if (dataSource == null && (this.lenientFallback || lookupKey == null)) {
  7. 			dataSource = this.resolvedDefaultDataSource;
  8. 		}
  9. 		if (dataSource == null) {
  10. 			throw new IllegalStateException("Cannot determine target DataSource for lookup key [" + lookupKey + "]");
  11. 		}
  12. 		return dataSource;
  13. 	}
  14.  
  15.         // 确定当前要使用的数据源
  16.         protected abstract Object determineCurrentLookupKey();
  17. }

Druid 实现多数据源支持，核心是Overwrite AbstractRoutingDataSource 的 determineCurrentLookupKey 方法

  1.   public class MultipleDataSource extends AbstractRoutingDataSource {
  2. 	public static final String DATA_SOURCE_A = "dataSourceTestA";  // A库
  3. 	public static final String DATA_SOURCE_B = "dataSourceTestB";  // B库
  4.  
  5. 	private static final ThreadLocal<String> dataSourceKey = new ThreadLocal<String>();
  6.  
  7. 	public static void setDataSourceKey(String dataSource) {
  8. 		dataSourceKey.set(dataSource);
  9. 	}
  10.  
  11. 	@Override
  12. 	protected Object determineCurrentLookupKey() {
  13. 		return dataSourceKey.get();
  14. 	}
  15.  
  16. 	public static void clear(){
  17. 		dataSourceKey.remove();
  18. 	}
  19. }

AbstractRoutingDataSourceApplication实战：

DataSourceLeonProperties

  1. package center.leon.oak.druid.abstract_routing_data_source.properties;
  2.  
  3. import lombok.Data;
  4. import lombok.experimental.Accessors;
  5. import org.springframework.boot.context.properties.ConfigurationProperties;
  6. import org.springframework.stereotype.Component;
  7.  
  8. import java.util.List;
  9.  
  10. /**
  11.  * @program: leon
  12.  * @description:
  13.  * @author: Leon
  14.  * @create: 2021-07-22 13:56
  15.  **/
  16. @Data
  17. @Component
  18. @ConfigurationProperties(value = "spring.datasource.leon")
  19. public class DataSourceLeonProperties {
  20.  
  21.     /**
  22.      * 默认数据源
  23.      */
  24.     private DataSourceLeon defaultDataSourceLeon;
  25.     /**
  26.      * 数据源列表
  27.      */
  28.     private List<DataSourceLeon> dataSourceLeonList;
  29.  
  30.  
  31.     /**
  32.      * 自定义数据源
  33.      */
  34.     @Data
  35.     @Accessors(chain = true)
  36.     public static class DataSourceLeon {
  37.         /**
  38.          * 数据源自定义名称
  39.          */
  40.         private String dataSourceName;
  41.         /**
  42.          * 数据源驱动
  43.          */
  44.         private String driverClassName;
  45.         /**
  46.          * 数据源URL
  47.          */
  48.         private String url;
  49.         /**
  50.          *
  51.          */
  52.         private String username;
  53.         /**
  54.          *
  55.          */
  56.         private String password;
  57.     }
  58. }

MultipleDataSource

  1. package center.leon.oak.druid.abstract_routing_data_source.config;
  2.  
  3. import center.leon.oak.druid.abstract_routing_data_source.properties.DataSourceLeonProperties;
  4. import com.alibaba.druid.pool.DruidDataSource;
  5. import com.alibaba.druid.spring.boot.autoconfigure.DruidDataSourceBuilder;
  6. import org.springframework.beans.factory.InitializingBean;
  7. import org.springframework.beans.factory.annotation.Autowired;
  8. import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource;
  9. import org.springframework.stereotype.Component;
  10.  
  11. import java.util.HashMap;
  12. import java.util.Map;
  13.  
  14. /**
  15.  * @program: leon
  16.  * @description:
  17.  * @author: Leon
  18.  * @create: 2021-07-22 11:31
  19.  **/
  20. @Component
  21. public class MultipleDataSource extends AbstractRoutingDataSource implements InitializingBean {
  22.  
  23.     ThreadLocal<String> data_source_name = new ThreadLocal<>();
  24.  
  25.     @Autowired
  26.     private DataSourceLeonProperties datasourceLeonProperties;
  27.  
  28.     @Override
  29.     public void afterPropertiesSet() {
  30.         init();
  31.     }
  32.  
  33.     private void init() {
  34.         // 设置默认数据源
  35.         if (null != datasourceLeonProperties.getDefaultDataSourceLeon()) {
  36.             DataSourceLeonProperties.DataSourceLeon dataSourceLeon = datasourceLeonProperties.getDefaultDataSourceLeon();
  37.             DruidDataSource druidDataSource = DruidDataSourceBuilder.create().build();
  38.             druidDataSource.setDriverClassName(dataSourceLeon.getDriverClassName());
  39.             druidDataSource.setUrl(dataSourceLeon.getUrl());
  40.             druidDataSource.setUsername(dataSourceLeon.getUsername());
  41.             druidDataSource.setPassword(dataSourceLeon.getPassword());
  42.             super.setDefaultTargetDataSource(druidDataSource);
  43.         }
  44.  
  45.         // 设置多数据源
  46.         Map<Object, Object> targetDataSources = new HashMap<>();
  47.         for (DataSourceLeonProperties.DataSourceLeon dataSourceLeon : datasourceLeonProperties.getDataSourceLeonList()) {
  48.             DruidDataSource druidDataSource = DruidDataSourceBuilder.create().build();
  49.             druidDataSource.setDriverClassName(dataSourceLeon.getDriverClassName());
  50.             druidDataSource.setUrl(dataSourceLeon.getUrl());
  51.             druidDataSource.setUsername(dataSourceLeon.getUsername());
  52.             druidDataSource.setPassword(dataSourceLeon.getPassword());
  53.  
  54.             targetDataSources.put(dataSourceLeon.getDataSourceName(), druidDataSource);
  55.         }
  56.         super.setTargetDataSources(targetDataSources);
  57.  
  58.         // 调用父类 afterPropertiesSet 方法
  59.         // 将 targetDataSources 数据 填充到 resolvedDataSources 中
  60.         // determineCurrentLookupKey 返回的 lookupKey 本质上是在 resolvedDataSources 中查找 dataSource 返回
  61.         // 将 defaultTargetDataSource 数据 填充到 resolvedDefaultDataSource 中
  62.         // resolvedDataSources 中 查找不到合适的 dataSource 时返回。
  63.         super.afterPropertiesSet();
  64.     }
  65.  
  66.     @Override
  67.     protected Object determineCurrentLookupKey() {
  68.         return data_source_name.get();
  69.     }
  70.  
  71.     public void setDb_tb_name(String dataSourceName) {
  72.         data_source_name.set(dataSourceName);
  73.     }
  74.  
  75.     public void clearDb_tb_name() {
  76.         data_source_name.remove();
  77.     }
  78. }

MonkeyDO

  1. package center.leon.oak.druid.abstract_routing_data_source.dto;
  2.  
  3. import com.baomidou.mybatisplus.annotation.TableName;
  4. import lombok.Data;
  5. import lombok.experimental.Accessors;
  6.  
  7. /**
  8.  * @program: leon
  9.  * @description:
  10.  * @author: Leon
  11.  * @create: 2021-07-22 12:05
  12.  **/
  13. @Data
  14. @TableName(value = "monkey")
  15. @Accessors(chain = true)
  16. public class MonkeyDO {
  17.     private Long id;
  18.     private String name;
  19.     private Long version;
  20. }

MonkeyMapper

  1. package center.leon.oak.druid.abstract_routing_data_source.mapper;
  2.  
  3. import center.leon.oak.druid.abstract_routing_data_source.dto.MonkeyDO;
  4. import com.baomidou.mybatisplus.core.mapper.BaseMapper;
  5. import org.apache.ibatis.annotations.Mapper;
  6.  
  7. /**
  8.  * @program: leon
  9.  * @description:
  10.  * @author: Leon
  11.  * @create: 2021-07-22 12:04
  12.  **/
  13. @Mapper
  14. public interface MonkeyMapper extends BaseMapper<MonkeyDO> {
  15. }

MonkeyService

  1. package center.leon.oak.druid.abstract_routing_data_source.service;
  2.  
  3. import center.leon.oak.druid.abstract_routing_data_source.dto.MonkeyDO;
  4.  
  5. /**
  6.  * @program: leon
  7.  * @description:
  8.  * @author: Leon
  9.  * @create: 2021-07-22 12:00
  10.  **/
  11. public interface MonkeyService {
  12.  
  13.     MonkeyDO get(Long id);
  14.  
  15.     MonkeyDO getInDataSource(Long id, String dataSourceName);
  16. }

MonkeyServiceImpl

  1. package center.leon.oak.druid.abstract_routing_data_source.service.impl;
  2.  
  3. import center.leon.oak.druid.abstract_routing_data_source.config.MultipleDataSource;
  4. import center.leon.oak.druid.abstract_routing_data_source.dto.MonkeyDO;
  5. import center.leon.oak.druid.abstract_routing_data_source.mapper.MonkeyMapper;
  6. import center.leon.oak.druid.abstract_routing_data_source.service.MonkeyService;
  7. import org.springframework.beans.factory.annotation.Autowired;
  8. import org.springframework.stereotype.Service;
  9.  
  10. /**
  11.  * @program: leon
  12.  * @description:
  13.  * @author: Leon
  14.  * @create: 2021-07-22 12:03
  15.  **/
  16. @Service
  17. public class MonkeyServiceImpl implements MonkeyService {
  18.  
  19.     @Autowired
  20.     private MonkeyMapper monkeyMapper;
  21.     @Autowired
  22.     private MultipleDataSource multipleDataSource;
  23.  
  24.     @Override
  25.     public MonkeyDO get(Long id) {
  26.         return monkeyMapper.selectById(id);
  27.     }
  28.  
  29.     @Override
  30.     public MonkeyDO getInDataSource(Long id, String dataSourceName) {
  31.         try {
  32.             multipleDataSource.setDb_tb_name(dataSourceName);
  33.             return monkeyMapper.selectById(id);
  34.         } finally {
  35.             multipleDataSource.clearDb_tb_name();
  36.         }
  37.     }
  38. }

PrintMonkey

  1. package center.leon.oak.druid.abstract_routing_data_source.task;
  2.  
  3. import center.leon.oak.druid.abstract_routing_data_source.dto.MonkeyDO;
  4. import center.leon.oak.druid.abstract_routing_data_source.service.MonkeyService;
  5. import lombok.extern.slf4j.Slf4j;
  6. import org.springframework.beans.factory.annotation.Autowired;
  7. import org.springframework.scheduling.annotation.Scheduled;
  8. import org.springframework.stereotype.Component;
  9.  
  10. /**
  11.  * @program: leon
  12.  * @description:
  13.  * @author: Leon
  14.  * @create: 2021-07-22 21:28
  15.  **/
  16. @Slf4j
  17. @Component
  18. public class PrintMonkey {
  19.  
  20.     @Autowired
  21.     private MonkeyService monkeyService;
  22.  
  23.     @Scheduled(fixedRate = 6 * 1000)
  24.     public void printMonkey_localhost() {
  25.         MonkeyDO monkeyDO = monkeyService.getInDataSource(1L, "localhost");
  26.         log.info("threadName : {}, monkey : {}", Thread.currentThread().getName(), monkeyDO);
  27.     }
  28.  
  29.     @Scheduled(fixedRate = 6 * 1000)
  30.     public void printMonkey_www_leon_center() {
  31.         MonkeyDO monkeyDO = monkeyService.getInDataSource(1L, "www_leon_center");
  32.         log.info("threadName : {}, monkey : {}", Thread.currentThread().getName(), monkeyDO);
  33.     }
  34. }

application-druid.properties

  1. spring.datasource.type=center.leon.oak.druid.abstract_routing_data_source.config.MultipleDataSource
  2.  
  3. spring.datasource.leon.data-source-leon-list[0].data-source-name=localhost
  4. spring.datasource.leon.data-source-leon-list[0].driver-class-name=com.mysql.cj.jdbc.Driver
  5. spring.datasource.leon.data-source-leon-list[0].url=jdbc:mysql://localhost:3306/zoo?useUnicode=true&characterEncoding=UTF-8&serverTimezone=GMT%2B8&useSSL=false
  6. spring.datasource.leon.data-source-leon-list[0].username=root
  7. spring.datasource.leon.data-source-leon-list[0].password=Liang123,
  8.  
  9. spring.datasource.leon.data-source-leon-list[1].data-source-name=www_leon_center
  10. spring.datasource.leon.data-source-leon-list[1].driver-class-name=com.mysql.cj.jdbc.Driver
  11. spring.datasource.leon.data-source-leon-list[1].url=jdbc:mysql://www.leon.center:3306/zoo?useUnicode=true&characterEncoding=UTF-8&serverTimezone=GMT%2B8&useSSL=false
  12. spring.datasource.leon.data-source-leon-list[1].username=lmy
  13. spring.datasource.leon.data-source-leon-list[1].password=Lmy123456,
  14.  
  15.  
  16. spring.task.scheduling.pool.size=2

AbstractRoutingDataSourceApplication

  1. package center.leon.oak.druid.abstract_routing_data_source;
  2.  
  3. import center.leon.oak.druid.abstract_routing_data_source.config.MultipleDataSource;
  4. import center.leon.oak.druid.abstract_routing_data_source.dto.MonkeyDO;
  5. import center.leon.oak.druid.abstract_routing_data_source.properties.DataSourceLeonProperties;
  6. import center.leon.oak.druid.abstract_routing_data_source.service.MonkeyService;
  7. import center.leon.oak.druid.abstract_routing_data_source.service.impl.MonkeyServiceImpl;
  8. import lombok.extern.slf4j.Slf4j;
  9. import org.springframework.boot.SpringApplication;
  10. import org.springframework.boot.autoconfigure.SpringBootApplication;
  11. import org.springframework.boot.context.properties.EnableConfigurationProperties;
  12. import org.springframework.context.ConfigurableApplicationContext;
  13. import org.springframework.scheduling.annotation.EnableScheduling;
  14.  
  15. /**
  16.  * @program: leon
  17.  * @description:
  18.  * @author: Leon
  19.  * @create: 2021-07-22 11:28
  20.  **/
  21. @Slf4j
  22. @EnableScheduling
  23. @EnableConfigurationProperties
  24. @SpringBootApplication
  25. public class AbstractRoutingDataSourceApplication {
  26.     public static void main(String[] args) {
  27.         ConfigurableApplicationContext run = SpringApplication.run(AbstractRoutingDataSourceApplication.class, args);
  28.         DataSourceLeonProperties dataSourceLeonProperties = run.getBean(DataSourceLeonProperties.class);
  29.         log.info("dataSourceLeonProperties : {}", dataSourceLeonProperties);
  30.         MonkeyService monkeyService = run.getBean(MonkeyServiceImpl.class);
  31.  
  32.         final MultipleDataSource multipleDataSource = run.getBean(MultipleDataSource.class);
  33.  
  34.         multipleDataSource.setDb_tb_name("localhost");
  35.         MonkeyDO monkey_1 = monkeyService.get(1L);
  36.         log.info("monkey_1 : {}", monkey_1);
  37.         multipleDataSource.clearDb_tb_name();
  38.  
  39.         multipleDataSource.setDb_tb_name("www_leon_center");
  40.         MonkeyDO monkey_2 = monkeyService.get(1L);
  41.         log.info("monkey_2 : {}", monkey_2);
  42.         multipleDataSource.clearDb_tb_name();
  43.     }
  44. }

Console

  1. 2021-07-22 22:11:39.813  INFO 69282 --- [           main] d.a.AbstractRoutingDataSourceApplication : show Time ...
  2. 2021-07-22 22:11:39.813  INFO 69282 --- [           main] d.a.AbstractRoutingDataSourceApplication : dataSourceLeonProperties : DataSourceLeonProperties(defaultDataSourceLeon=null, dataSourceLeonList=[DataSourceLeonProperties.DataSourceLeon(dataSourceName=localhost, driverClassName=com.mysql.cj.jdbc.Driver, url=jdbc:mysql://localhost:3306/zoo?useUnicode=true&characterEncoding=UTF-8&serverTimezone=GMT%2B8&useSSL=false, username=root, password=Liang123,), DataSourceLeonProperties.DataSourceLeon(dataSourceName=www_leon_center, driverClassName=com.mysql.cj.jdbc.Driver, url=jdbc:mysql://www.leon.center:3306/zoo?useUnicode=true&characterEncoding=UTF-8&serverTimezone=GMT%2B8&useSSL=false, username=lmy, password=Lmy123456,)])
  3. 2021-07-22 22:11:39.864  INFO 69282 --- [   scheduling-2] com.alibaba.druid.pool.DruidDataSource   : {dataSource-2} inited
  4. 2021-07-22 22:11:39.864  INFO 69282 --- [           main] com.alibaba.druid.pool.DruidDataSource   : {dataSource-1} inited
  5. 2021-07-22 22:11:40.026  INFO 69282 --- [           main] d.a.AbstractRoutingDataSourceApplication : monkey_1 : MonkeyDO(id=1, name=write, version=2)
  6. 2021-07-22 22:11:40.026  INFO 69282 --- [   scheduling-1] c.l.o.d.a.task.PrintMonkey               : threadName : scheduling-1, monkey : MonkeyDO(id=1, name=write, version=2)
  7. 2021-07-22 22:11:40.067  INFO 69282 --- [   scheduling-2] c.l.o.d.a.task.PrintMonkey               : threadName : scheduling-2, monkey : MonkeyDO(id=1, name=monkey sun, version=500)
  8. 2021-07-22 22:11:40.075  INFO 69282 --- [           main] d.a.AbstractRoutingDataSourceApplication : monkey_2 : MonkeyDO(id=1, name=monkey sun, version=500)
  9. 2021-07-22 22:11:40.367  INFO 69282 --- [on(3)-127.0.0.1] o.a.c.c.C.[Tomcat].[localhost].[/]       : Initializing Spring DispatcherServlet 'dispatcherServlet'
  10. 2021-07-22 22:11:40.367  INFO 69282 --- [on(3)-127.0.0.1] o.s.web.servlet.DispatcherServlet        : Initializing Servlet 'dispatcherServlet'
  11. 2021-07-22 22:11:40.367 TRACE 69282 --- [on(3)-127.0.0.1] o.s.web.servlet.DispatcherServlet        : Detected org.springframework.web.multipart.support.StandardServletMultipartResolver@42d0e41
  12. 2021-07-22 22:11:40.367 TRACE 69282 --- [on(3)-127.0.0.1] o.s.web.servlet.DispatcherServlet        : Detected org.springframework.web.servlet.i18n.AcceptHeaderLocaleResolver@48f2054d
  13. 2021-07-22 22:11:40.367 TRACE 69282 --- [on(3)-127.0.0.1] o.s.web.servlet.DispatcherServlet        : Detected org.springframework.web.servlet.theme.FixedThemeResolver@6d6cd1e0
  14. 2021-07-22 22:11:40.368 TRACE 69282 --- [on(3)-127.0.0.1] o.s.web.servlet.DispatcherServlet        : Detected DefaultRequestToViewNameTranslator
  15. 2021-07-22 22:11:40.369 TRACE 69282 --- [on(3)-127.0.0.1] o.s.web.servlet.DispatcherServlet        : Detected SessionFlashMapManager
  16. 2021-07-22 22:11:40.369 DEBUG 69282 --- [on(3)-127.0.0.1] o.s.web.servlet.DispatcherServlet        : enableLoggingRequestDetails='false': request parameters and headers will be masked to prevent unsafe logging of potentially sensitive data
  17. 2021-07-22 22:11:40.369  INFO 69282 --- [on(3)-127.0.0.1] o.s.web.servlet.DispatcherServlet        : Completed initialization in 2 ms
  18. 2021-07-22 22:11:45.812  INFO 69282 --- [   scheduling-1] c.l.o.d.a.task.PrintMonkey               : threadName : scheduling-1, monkey : MonkeyDO(id=1, name=write, version=2)
  19. 2021-07-22 22:11:45.824  INFO 69282 --- [   scheduling-2] c.l.o.d.a.task.PrintMonkey               : threadName : scheduling-2, monkey : MonkeyDO(id=1, name=monkey sun, version=500)
  20. 2021-07-22 22:11:51.808  INFO 69282 --- [   scheduling-1] c.l.o.d.a.task.PrintMonkey               : threadName : scheduling-1, monkey : MonkeyDO(id=1, name=write, version=2)
  21. 2021-07-22 22:11:51.822  INFO 69282 --- [   scheduling-2] c.l.o.d.a.task.PrintMonkey               : threadName : scheduling-2, monkey : MonkeyDO(id=1, name=monkey sun, version=500)
  22. 2021-07-22 22:11:57.807  INFO 69282 --- [   scheduling-1] c.l.o.d.a.task.PrintMonkey               : threadName : scheduling-1, monkey : MonkeyDO(id=1, name=write, version=2)
  23. 2021-07-22 22:11:57.818  INFO 69282 --- [   scheduling-2] c.l.o.d.a.task.PrintMonkey               : threadName : scheduling-2, monkey : MonkeyDO(id=1, name=monkey sun, version=500)
  24. 2021-07-22 22:12:03.809  INFO 69282 --- [   scheduling-1] c.l.o.d.a.task.PrintMonkey               : threadName : scheduling-1, monkey : MonkeyDO(id=1, name=write, version=2)
  25. 2021-07-22 22:12:04.051  INFO 69282 --- [   scheduling-2] c.l.o.d.a.task.PrintMonkey               : threadName : scheduling-2, monkey : MonkeyDO(id=1, name=monkey sun, version=500)
  26. 2021-07-22 22:12:09.811  INFO 69282 --- [   scheduling-1] c.l.o.d.a.task.PrintMonkey               : threadName : scheduling-1, monkey : MonkeyDO(id=1, name=write, version=2)
  27. 2021-07-22 22:12:09.824  INFO 69282 --- [   scheduling-2] c.l.o.d.a.task.PrintMonkey               : threadName : scheduling-2, monkey : MonkeyDO(id=1, name=monkey sun, version=500)
  28. 2021-07-22 22:12:15.810  INFO 69282 --- [   scheduling-1] c.l.o.d.a.task.PrintMonkey               : threadName : scheduling-1, monkey : MonkeyDO(id=1, name=write, version=2)
  29. 2021-07-22 22:12:15.821  INFO 69282 --- [   scheduling-2] c.l.o.d.a.task.PrintMonkey               : threadName : scheduling-2, monkey : MonkeyDO(id=1, name=monkey sun, version=500)

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/article/detail/52374?site
推荐阅读
相关标签

Copyright © 2003-2013 www.wpsshop.cn 版权所有，并保留所有权利。

  

闽ICP备14008679号