springboot数据源切换

需求：对不同的数据库进行操作，如读写分离

思想：在对数据库进行操作时，程序会默认去找数据源，从数据源获取一个连接connection，要做的点是在程序调用之前把数据源换了（例如本身默认是写，换成读的数据源），spring boot提供了一个AbstractRoutingDataSource类，翻译过来是抽象路由数据源，这个类可以指定使用的数据源，继承这个抽象类之后需要实现两个方法，一个方法是用来添加数据源的（读写分离就两个数据源，有其他的也可以添加进去），设置默认数据源的，还有一个是指定使用的数据源的。方法添加数据源的时候是以键值对的形式添加到map中，当指定数据源的方法返回的值在map中存在时，就是用这个key对应的数据源，如果不存在，就会使用设置的默认数据源

这里使用aop的方式来进行数据源的动态切换，先看数据源配置（这里两个数据源是读写分离，写在本机上，读在虚拟机上）

application.xml:

server:
  port: 9222
spring:
  datasource:
    localmysql:
      username: root
      password: root
      url: jdbc:mysql://localhost:3306/mytest?useUnicode=true&characterEncoding=utf8
      driverClassName: com.mysql.cj.jdbc.Driver
    remomysql:
      username: root
      password: root
      url: jdbc:mysql://172.21.4.130:3306/mytest?useUnicode=true&characterEncoding=utf8
      driverClassName: com.mysql.cj.jdbc.Driver
    type: com.alibaba.druid.pool.DruidDataSource

数据源配置为bean，加入ioc，供后面切换：

DatabaseResourceConfig

package com.gzt.config;

import com.alibaba.druid.pool.DruidDataSource;
import org.springframework.boot.context.properties.ConfigurationProperties;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

import javax.sql.DataSource;

/**
 * @Description:
 * @Author: guozhengtao
 * @Date: 2022/1/29 11:48
 */
@Configuration
public class DatabaseResourceConfig {

    @ConfigurationProperties(prefix = "spring.datasource.localmysql")
    @Bean(value = "LocalMysql")
    public DataSource getlocalDataSource() {
        return new DruidDataSource();
    }

    @ConfigurationProperties(prefix = "spring.datasource.remomysql")
    @Bean(value = "RemoMysql")
    public DataSource getremoDataSource() {
        return new DruidDataSource();
    }
}

数据源key枚举：DatabaseSelectEnum

package com.gzt.config;

/**
 * @Description: 定义一个枚举变量，当枚举变量的值为write是，选择写入的数据库连接，当枚举变量的值为read时，返回读取的数据库连接
 * @Author: guozhengtao
 * @Date: 2022/1/29 13:30
 */
public enum DatabaseSelectEnum {
    /**
     * 选择写枚举
     * */
    Write("write"),
    /**
     * 选择读枚举
     * */
    Read("read");

    private final String select;

    DatabaseSelectEnum(String select) {
        this.select = select;
    }

    public String getSelect() {
        return select;
    }
}

数据源切换适配：DatabaseAdapt

package com.gzt.config;

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.context.annotation.Primary;
import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource;
import org.springframework.stereotype.Component;

import javax.sql.DataSource;
import java.util.HashMap;
import java.util.Map;

/**
 * @Description:
 * @Author: guozhengtao
 * @Date: 2022/1/29 13:19
 */
@Component
@Primary
public class DatabaseAdapt extends AbstractRoutingDataSource {
    /**
     * 默认为写的数据库（主数据库）
    */
    public static String flag = DatabaseSelectEnum.Write.getSelect();

    @Autowired
    @Qualifier("LocalMysql")
    DataSource localMysql;

    @Autowired
    @Qualifier("RemoMysql")
    DataSource remoMysql;

    /**
     * 指定数据源的key，这里数据源的key使用了enum
     * */
    @Override
    protected Object determineCurrentLookupKey() {
        return flag;
    }

    /**
     * 设置默认数据源和添加所有数据源
     */
    @Override
    public void afterPropertiesSet() {
        //初始化所有数据源
        Map<Object, Object> targetDataSources = new HashMap<>();
        targetDataSources.put(DatabaseSelectEnum.Write.getSelect(), localMysql);
        targetDataSources.put(DatabaseSelectEnum.Read.getSelect(), remoMysql);
        super.setTargetDataSources(targetDataSources);

        //设置默认数据源,当上面方法返回的key找不到时，使用这里设置的数据源
        super.setDefaultTargetDataSource(localMysql);

        //属性设置（把上面数据源的设置到父类）
        super.afterPropertiesSet();
    }
}

到现在为止，数据源相关操作设置完毕，下面切换数据源的两种方式，因为在DatabaseAdapt类中定义了一个变量flag，这个flag是数据源对应的key，有两种方式来进行设置：

在方法中调用mapper之前先使用
```
DatabaseAdapt.flag=DatabaseSelectEnum.Write.getSelect();
```
来切换数据源，但这种方式适合单个请求的，如果类里面有多个方法请求，每个方法里都写很麻烦，所以推荐注解的方式，在类上使用注解+aop进行增强

使用aop的方式在执行方法或者类之前进行切换，主要介绍：

注解：DatabaseResourcedefine

package com.gzt.annotation;

import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;

/**
 * @Description:
 * @Author: guozhengtao
 * @Date: 2022/1/29 14:37
 */

@Target({ElementType.TYPE, ElementType.METHOD})
@Retention(RetentionPolicy.RUNTIME)
public @interface DatabaseResourcedefine {
    String value() default "";
}

Aop：DatabaseResourceToggle

package com.gzt.aspect;

import com.gzt.annotation.DatabaseResourcedefine;
import com.gzt.config.DatabaseAdapt;
import org.aspectj.lang.JoinPoint;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Before;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.stereotype.Component;

/**
 * @Description:
 * @Author: guozhengtao
 * @Date: 2022/1/29 15:01
 */
@Aspect
@Component
public class DatabaseResourceToggle {
    final Logger logger = LoggerFactory.getLogger(getClass());

    //使用前置通知或者是环绕通知
    @Before("within(com.gzt.controller.*)&&@annotation(databaseResourcedefine)")
    public void before(JoinPoint joinPoint, DatabaseResourcedefine databaseResourcedefine) {
        DatabaseAdapt.flag=databaseResourcedefine.value();
        logger.info(databaseResourcedefine.value());
    }
}

目录结构如下：

主要代码部分：

package com.gzt.service;

import com.gzt.annotation.DatabaseResourcedefine;
import com.gzt.entity.User;
import com.gzt.mapper.TestMapper;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import java.util.List;

/**
 * @Description:
 * @Author: guozhengtao
 * @Date: 2022/1/30 13:03
 */
@Service
@DatabaseResourcedefine(value = "read")
public class implTestImpl implements TestService{
   final Logger logger = LoggerFactory.getLogger(getClass());
   @Autowired
   TestMapper testMapper;

   //@DatabaseResourcedefine(value = "read")
   @Override
   public List<User> getUser() {
      //DatabaseAdapt.flag= DatabaseSelectEnum.Write.getSelect();
      List<User> user = testMapper.getUser();
      logger.info(String.valueOf(user));
      return user;
   }

   //@DatabaseResourcedefine(value = "write")
   @Override
   public void setUserInfo(User user) {
      logger.info("获取到用户信息：{}",user);
      //DatabaseAdapt.flag= DatabaseSelectEnum.Write.getSelect();
      testMapper.setUserInfo(user.getName(), user.getAge(), user.getSex());
   }
}