springBoot配置多数据源

桃扇骨 2022-02-19 14:59 349阅读 0赞

> 1, 配置文件中配置两个数据源

spring:
        datasource:
            type: com.alibaba.druid.pool.DruidDataSource
            driverClassName: com.mysql.jdbc.Driver
            druid:
                first:  #数据源1
                    url: jdbc:mysql://127.0.0.1:3306/guandaoxunjian?allowMultiQueries=true&useUnicode=true&characterEncoding=UTF-8
                    username: root
                    password: 123456
                second:  #数据源2
                    url: jdbc:mysql://127.0.0.1:3306/guandaoxunjian?allowMultiQueries=true&useUnicode=true&characterEncoding=UTF-8
                    username: root
                    password: 123456

> 2，自定义注解可作用于方法，接口，类中

@Target(ElementType.METHOD)
    @Retention(RetentionPolicy.RUNTIME)
    @Documented
    public @interface DataSource {
        String name() default "";
    }

3, 创建一个接口用于标明数据源的名称，结合自定义注解中的name 属性，结合注解通过名称来切换数据源

## @Target 注解 ##

功能：指明了修饰的这个注解的使用范围，即被描述的注解可以用在哪里。

ElementType的取值包含以下几种：

*  TYPE:类，接口或者枚举
 *  FIELD:域，包含枚举常量
 *  METHOD:方法
 *  PARAMETER:参数
 *  CONSTRUCTOR:构造方法
 *  LOCAL\_VARIABLE:局部变量
 *  ANNOTATION\_TYPE:注解类型
 *  PACKAGE:包

## @Retention 注解 ##

功能：指明修饰的注解的生存周期，即会保留到哪个阶段。

RetentionPolicy的取值包含以下三种：

*  SOURCE：源码级别保留，编译后即丢弃。
 *  CLASS:编译级别保留，编译后的class文件中存在，在jvm运行时丢弃，这是默认值。
 *  RUNTIME： 运行级别保留，编译后的class文件中存在，在jvm运行时保留，可以被反射调用。

## @Documented 注解 ##

功能：指明修饰的注解，可以被例如javadoc此类的工具文档化，只负责标记，没有成员取值。

## @Inherited注解 ##

@interface意思是声明一个注解，方法名对应参数名，返回值类型对应参数类型

/**
     * 增加多数据源
     */
    public interface DataSourceNames {
        String FIRST = "first";
        String SECOND = "second";
        String THIRD = "third";
    
    }

> 4, 配置数据源切面类通过 springAop 思想来设置切面类

/**
     * 多数据源，切面处理类
     */
    @Aspect
    @Component
    public class DataSourceAspect implements Ordered {
        protected Logger logger = LoggerFactory.getLogger(getClass());
    
        @Pointcut("@annotation(com.fs.common.annotation.DataSource)")
        public void dataSourcePointCut() {
    
        }
    
        @Around("dataSourcePointCut()")
        public Object around(ProceedingJoinPoint point) throws Throwable {
            MethodSignature signature = (MethodSignature) point.getSignature();
            Method method = signature.getMethod();
    
            DataSource ds = method.getAnnotation(DataSource.class);
            if(ds == null){
                DynamicDataSource.setDataSource(DataSourceNames.FIRST);
                logger.debug("set datasource is " + DataSourceNames.FIRST);
            }else {
                DynamicDataSource.setDataSource(ds.name());
                logger.debug("set datasource is " + ds.name());
            }
    
            try {
                return point.proceed();
            } finally {
                DynamicDataSource.clearDataSource();
                logger.debug("clean datasource");
            }
        }
    
        @Override
        public int getOrder() {
            return 1;
        }
    }

5，配置多数据源的 获取配置文件中的数据源信息

@Primary配置了数据源为主数据源，当没有配置自动切换的package时默认使用该数据源进行数据处理操作。

用到了阿里的 Druid数据库连接池

/**
     * 配置多数据源
     */
    @Configuration
    public class DynamicDataSourceConfig {
    
        @Bean
        @ConfigurationProperties("spring.datasource.druid.first")
        public DataSource firstDataSource(){
            return DruidDataSourceBuilder.create().build();
        }
    
        @Bean
        @ConfigurationProperties("spring.datasource.druid.second")
        public DataSource secondDataSource(){
            return DruidDataSourceBuilder.create().build();
        }
    
        @Bean
        @Primary
        public DynamicDataSource dataSource(DataSource firstDataSource, DataSource secondDataSource) {
            Map<Object, Object> targetDataSources = new HashMap<>();
            targetDataSources.put(DataSourceNames.FIRST, firstDataSource);
            targetDataSources.put(DataSourceNames.SECOND, secondDataSource);
            return new DynamicDataSource(firstDataSource, targetDataSources);
        }

6 设置动态数据源资源类

主要是两个参数的构造器 第一个默认的数据源，第二个是一个Map集合用来存放多个数据源

/**
     * 动态数据源
     */
    public class DynamicDataSource extends AbstractRoutingDataSource {
        private static final ThreadLocal<String> contextHolder = new ThreadLocal<>();
    
        public DynamicDataSource(DataSource defaultTargetDataSource, Map<Object, Object> targetDataSources) {
            super.setDefaultTargetDataSource(defaultTargetDataSource);
            super.setTargetDataSources(targetDataSources);
            super.afterPropertiesSet();
        }
    
        @Override
        protected Object determineCurrentLookupKey() {
            return getDataSource();
        }
    
        public static void setDataSource(String dataSource) {
            contextHolder.set(dataSource);
        }
    
        public static String getDataSource() {
            return contextHolder.get();
        }
    
        public static void clearDataSource() {
            contextHolder.remove();
        }
    
    }

开启多数据源 标注在接口或者实现类中

@DataSource(name = DataSourceNames.THIRD)
        public List<Position> selectListByLogId(Map<String, Object> map) {
            return positionMapper.selectListByLogId(map);
        }