spring:
 datasource:
  master:
   jdbcUrl: jdbc:mysql://192.168.xxx.xxx:xxxx/db1?.........
   username: xxx
   password: xxx
   driverClassName: com.mysql.cj.jdbc.Driver
  s1:
   jdbcUrl: jdbc:mysql://192.168.xxx.xxx:xxxx/db2?........
   username: xxx
   password: xxx
   driverClassName: com.mysql.cj.jdbc.Driver

@Configuration
@MapperScan(basePackages = "com.hosjoy.xxx.xxx.xxx.xxx.mapper.master", sqlSessionFactoryRef = "masterSqlSessionFactory")
public class MasterDataSourceConfig {

  //默認(rèn)數(shù)據(jù)源
  @Bean(name = "masterDataSource")
  @Primary
  @ConfigurationProperties(prefix = "spring.datasource.master")
  public HikariDataSource masterDataSource() {
    return DataSourceBuilder.create().type(HikariDataSource.class).build();
  }

  @Bean(name = "masterSqlSessionFactory")
  @Primary
  public SqlSessionFactory masterSqlSessionFactory(@Qualifier("masterDataSource") DataSource datasource, PaginationInterceptor paginationInterceptor)
      throws Exception {
    MybatisSqlSessionFactoryBean bean = new MybatisSqlSessionFactoryBean();
    bean.setDataSource(datasource);
    bean.setMapperLocations(
        // 設(shè)置mybatis的xml所在位置
        new PathMatchingResourcePatternResolver().getResources("classpath*:mapper/master/**/**.xml"));
    bean.setPlugins(new Interceptor[]{paginationInterceptor});
    return bean.getObject();
  }

  @Bean(name = "masterSqlSessionTemplate")
  @Primary
  public SqlSessionTemplate masterSqlSessionTemplate(
      @Qualifier("masterSqlSessionFactory") SqlSessionFactory sessionfactory) {
    return new SqlSessionTemplate(sessionfactory);
  }
}

@Configuration
@MapperScan(basePackages = "com.hosjoy.xxx.xxx.xxx.xxx.mapper.s1", sqlSessionFactoryRef = "s1SqlSessionFactory")
public class S1DataSourceConfig {

  @Bean(name = "s1DataSource")
  @ConfigurationProperties(prefix = "spring.datasource.s1")
  public HikariDataSource s1DataSource() {
    return DataSourceBuilder.create().type(HikariDataSource.class).build();
  }

  @Bean(name = "s1SqlSessionFactory")
  public SqlSessionFactory s1SqlSessionFactory(@Qualifier("s1DataSource") DataSource datasource
      , PaginationInterceptor paginationInterceptor)
      throws Exception {
    MybatisSqlSessionFactoryBean bean = new MybatisSqlSessionFactoryBean();
    bean.setDataSource(datasource);
    bean.setMapperLocations(
        // 設(shè)置mybatis的xml所在位置
        new PathMatchingResourcePatternResolver().getResources("classpath*:mapper/s1/**/**.xml"));
    bean.setPlugins(new Interceptor[]{paginationInterceptor});
    return bean.getObject();
  }

  @Bean(name = "s1SqlSessionTemplate")
  public SqlSessionTemplate s1SqlSessionTemplate(
      @Qualifier("s1SqlSessionFactory") SqlSessionFactory sessionfactory) {
    return new SqlSessionTemplate(sessionfactory);
  }
}

/**偽代碼**/
int userType = reqUser.getType();
if (userType == 1){
  //切換到數(shù)據(jù)源slave1
  //數(shù)據(jù)庫(kù)操作
} else if(userType == 2){
  //切換到數(shù)據(jù)源slave2
  //數(shù)據(jù)庫(kù)操作
}

spring:
 datasource:
  master:
   jdbcUrl: jdbc:mysql://192.168.xx.xxx:xxx/db1?........
   username: xxx
   password: xxx
   driverClassName: com.mysql.cj.jdbc.Driver
  slave1:
   jdbcUrl: jdbc:sqlserver://192.168.xx.xxx:xxx;DatabaseName=db2
   username: xxx
   password: xxx
   driverClassName: com.microsoft.sqlserver.jdbc.SQLServerDriver
  slave2:
   jdbcUrl: jdbc:sqlserver://192.168.xx.xxx:xxx;DatabaseName=db3
   username: xxx
   password: xxx
   driverClassName: com.microsoft.sqlserver.jdbc.SQLServerDriver

public class DynamicDataSource extends AbstractRoutingDataSource {
  /*存儲(chǔ)所有數(shù)據(jù)源*/
  private Map<Object, Object> backupTargetDataSources;

  public Map<Object, Object> getBackupTargetDataSources() {
    return backupTargetDataSources;
  }
  /*defaultDataSource為默認(rèn)數(shù)據(jù)源*/
  public DynamicDataSource(DataSource defaultDataSource, Map<Object, Object> targetDataSource) {
    backupTargetDataSources = targetDataSource;
    super.setDefaultTargetDataSource(defaultDataSource);
    super.setTargetDataSources(backupTargetDataSources);
    super.afterPropertiesSet();
  }
  public void addDataSource(String key, DataSource dataSource) {
    this.backupTargetDataSources.put(key, dataSource);
    super.setTargetDataSources(this.backupTargetDataSources);
    super.afterPropertiesSet();
  }
  /*返回當(dāng)前線(xiàn)程的數(shù)據(jù)源的key*/
  @Override
  protected Object determineCurrentLookupKey() {
    return DynamicDataSourceContextHolder.getContextKey();
  }
}

public class DynamicDataSourceContextHolder {
  /**
   * 存儲(chǔ)線(xiàn)程和數(shù)據(jù)源key的映射關(guān)系
   */
  private static final ThreadLocal<String> DATASOURCE_CONTEXT_KEY_HOLDER = new ThreadLocal<>();

  /***
   * 設(shè)置當(dāng)前線(xiàn)程數(shù)據(jù)源key
   */
  public static void setContextKey(String key) {
    DATASOURCE_CONTEXT_KEY_HOLDER.set(key);
  }
  /***
   * 獲取當(dāng)前線(xiàn)程數(shù)據(jù)源key
   */
  public static String getContextKey() {
    String key = DATASOURCE_CONTEXT_KEY_HOLDER.get();
    return key == null ? DataSourceConstants.DS_KEY_MASTER : key;
  }
  /***
   * 刪除當(dāng)前線(xiàn)程數(shù)據(jù)源key
   */
  public static void removeContextKey() {
    DynamicDataSource dynamicDataSource = RequestHandleMethodRegistry.getContext().getBean(DynamicDataSource.class);
    String currentKey = DATASOURCE_CONTEXT_KEY_HOLDER.get();
    if (StringUtils.isNotBlank(currentKey) && !"master".equals(currentKey)) {
      dynamicDataSource.getBackupTargetDataSources().remove(currentKey);
    }
    DATASOURCE_CONTEXT_KEY_HOLDER.remove();
  }
}

@Configuration
@MapperScan(basePackages = "com.hosjoy.xxx.xxx.modules.xxx.mapper")
public class DynamicDataSourceConfig {
  @Autowired
  private BeanFactory beanFactory;
  @Autowired
  private DynamicDataSourceProperty dynamicDataSourceProperty;
  /**
   * 功能描述: <br>
   * 〈動(dòng)態(tài)數(shù)據(jù)源bean 自動(dòng)配置注冊(cè)所有數(shù)據(jù)源〉
   *
   * @param
   * @return javax.sql.DataSource
   * @Author li.he
   * @Date 2020/6/4 16:47
   * @Modifier
   */
  @Bean
  @Primary
  public DataSource dynamicDataSource() {
    DefaultListableBeanFactory listableBeanFactory = (DefaultListableBeanFactory) beanFactory;
    /*獲取yml所有數(shù)據(jù)源配置*/
    Map<String, Object> datasource = dynamicDataSourceProperty.getDatasource();
    Map<Object, Object> dataSourceMap = new HashMap<>(5);
    Optional.ofNullable(datasource).ifPresent(map -> {
      for (Map.Entry<String, Object> entry : map.entrySet()) {
        //創(chuàng)建數(shù)據(jù)源對(duì)象
        HikariDataSource dataSource = (HikariDataSource) DataSourceBuilder.create().build();
        String dataSourceId = entry.getKey();
        configeDataSource(entry, dataSource);
        /*bean工廠注冊(cè)每個(gè)數(shù)據(jù)源bean*/
        listableBeanFactory.registerSingleton(dataSourceId, dataSource);
        dataSourceMap.put(dataSourceId, dataSource);
      }
    });
    //AbstractRoutingDataSource設(shè)置主從數(shù)據(jù)源
    return new DynamicDataSource(beanFactory.getBean("master", DataSource.class),     dataSourceMap);
  }

  private void configeDataSource(Map.Entry<String, Object> entry, HikariDataSource dataSource) {
    Map<String, Object> dataSourceConfig = (Map<String, Object>) entry.getValue();
    dataSource.setJdbcUrl(MapUtils.getString(dataSourceConfig, "jdbcUrl"));
    dataSource.setDriverClassName(MapUtils.getString(dataSourceConfig, "driverClassName"));
    dataSource.setUsername(MapUtils.getString(dataSourceConfig, "username"));
    dataSource.setPassword(MapUtils.getString(dataSourceConfig, "password"));
  }

}

@Slf4j
@Component
public class DataSourceUtil {
  @Autowired
  private DataSourceConfiger dataSourceConfiger;
  
  /*根據(jù)業(yè)務(wù)條件切換數(shù)據(jù)源*/
  public void switchDataSource(String key, Predicate<? super Map<String, Object>> predicate) {
    try {
      //生成當(dāng)前線(xiàn)程數(shù)據(jù)源key
      String newDsKey = System.currentTimeMillis() + "";
      List<Map<String, Object>> configValues = dataSourceConfiger.getConfigValues(key);
      Map<String, Object> db = configValues.stream().filter(predicate)
          .findFirst().get();
      String id = MapUtils.getString(db, "id");
      //根據(jù)ID從數(shù)據(jù)源緩存池中獲取數(shù)據(jù)源對(duì)象，并再次添加到backupTargetDataSources
      addDataSource(newDsKey, id);
      //設(shè)置當(dāng)前線(xiàn)程對(duì)應(yīng)的數(shù)據(jù)源key
      DynamicDataSourceContextHolder.setContextKey(newDsKey);
      log.info("當(dāng)前線(xiàn)程數(shù)據(jù)源切換成功，當(dāng)前數(shù)據(jù)源ID:{}", id);

    }
    catch (Exception e) {
      log.error("切換數(shù)據(jù)源失敗，請(qǐng)檢查數(shù)據(jù)源配置文件。key:{}, 條件:{}", key, predicate.toString());
      throw new ClientException("切換數(shù)據(jù)源失敗，請(qǐng)檢查數(shù)據(jù)源配置", e);
    }
  }
  
  /*將數(shù)據(jù)源添加至多數(shù)據(jù)源緩存池中*/
  public static void addDataSource(String key, String dataSourceId) {
    DynamicDataSource dynamicDataSource = RequestHandleMethodRegistry.getContext().getBean(DynamicDataSource.class);
    DataSource dataSource = (DataSource) dynamicDataSource.getBackupTargetDataSources().get(dataSourceId);
    dynamicDataSource.addDataSource(key, dataSource);
  }
}

public void doExecute(ReqTestParams reqTestParams){
  //構(gòu)造條件
  Predicate<? super Map<String, Object>> predicate =.........;
  //切換數(shù)據(jù)源
  dataSourceUtil.switchDataSource("testKey", predicate);
  //數(shù)據(jù)庫(kù)操作
  mapper.testQuery();
  //清理緩存，避免內(nèi)存泄漏
  DynamicDataSourceContextHolder.removeContextKey();
}

@Aspect
@Component
@Slf4j
public class RequestHandleMethodAspect {
  @After("xxxxxxxxxxxxxxExecution表達(dá)式xxxxxxxxxxxxxxxxxx")
  public void afterRunning(JoinPoint joinPoint){
    String name = joinPoint.getSignature().toString();
    long id = Thread.currentThread().getId();
    log.info("方法執(zhí)行完畢，開(kāi)始清空當(dāng)前線(xiàn)程數(shù)據(jù)源，線(xiàn)程id:{},代理方法:{}",id,name);
    DynamicDataSourceContextHolder.removeContextKey();
    log.info("當(dāng)前線(xiàn)程數(shù)據(jù)源清空完畢,已返回至默認(rèn)數(shù)據(jù)源:{}",id);
  }
}

public static void switchDataSource(String dataSourceId) {
  if (StringUtils.isBlank(dataSourceId)) {
    throw new ClientException("切換數(shù)據(jù)源失敗，數(shù)據(jù)源ID不能為空");
  }
  try {
    String threadDataSourceKey = UUID.randomUUID().toString();
    DataSourceUtil.addDataSource(threadDataSourceKey, dataSourceId);
    DynamicDataSourceContextHolder.setContextKey(threadDataSourceKey);
  }
  catch (Exception e) {
    log.error("切換數(shù)據(jù)源失敗，未找到指定的數(shù)據(jù)源，請(qǐng)確保所指定的數(shù)據(jù)源ID已在配置文件中配置。dataSourceId:{}", dataSourceId);
    throw new ClientException("切換數(shù)據(jù)源失敗，未找到指定的數(shù)據(jù)源，請(qǐng)確保所指定的數(shù)據(jù)源ID已在配置文件中配置。dataSourceId:" + dataSourceId, e);
  }
}

@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
public @interface UsingDataSource {

  String dataSourceId() default "master";
}

@Aspect
@Component
@Slf4j
@Order(value = 1)
public class DynamicDataSourceAspect {

  //攔截UsingDataSource注解標(biāo)記的方法，方法執(zhí)行前切換數(shù)據(jù)源
  @Before(value = "@annotation(usingDataSource)")
  public void before(JoinPoint joinPoint, UsingDataSource usingDataSource) {
    String dataSourceId = usingDataSource.dataSourceId();
    log.info("執(zhí)行目標(biāo)方法前開(kāi)始切換數(shù)據(jù)源，目標(biāo)方法:{}, dataSourceId:{}", joinPoint.getSignature().toString(), dataSourceId);
    try {
      DataSourceUtil.switchDataSource(dataSourceId);
    }
    catch (Exception e) {
      log.error("切換數(shù)據(jù)源失??！數(shù)據(jù)源可能未配置或不可用,數(shù)據(jù)源ID:{}", dataSourceId, e);
      throw new ClientException("切換數(shù)據(jù)源失?。?shù)據(jù)源可能未配置或不可用,數(shù)據(jù)源ID:" + dataSourceId, e);
    }
    log.info("目標(biāo)方法:{} , 已切換至數(shù)據(jù)源:{}", joinPoint.getSignature().toString(), dataSourceId);
  }

  //攔截UsingDataSource注解標(biāo)記的方法，方法執(zhí)行后清理數(shù)據(jù)源，防止內(nèi)存泄漏
  @After(value = "@annotation(com.hosjoy.hbp.dts.common.annotation.UsingDataSource)")
  public void after(JoinPoint joinPoint) {
    log.info("目標(biāo)方法執(zhí)行完畢，執(zhí)行清理，切換至默認(rèn)數(shù)據(jù)源，目標(biāo)方法:{}", joinPoint.getSignature().toString());
    try {
      DynamicDataSourceContextHolder.removeContextKey();
    }
    catch (Exception e) {
      log.error("清理數(shù)據(jù)源失敗", e);
      throw new ClientException("清理數(shù)據(jù)源失敗", e);
    }
    log.info("目標(biāo)方法:{} , 數(shù)據(jù)源清理完畢，已返回默認(rèn)數(shù)據(jù)源", joinPoint.getSignature().toString());
  }
}

@UsingDataSource(dataSourceId = "slave1")
@Transactional
public void test(){
  AddressPo po = new AddressPo();
  po.setMemberCode("asldgjlk");
  po.setName("lihe");
  po.setPhone("13544986666");
  po.setProvince("asdgjwlkgj");
  addressMapper.insert(po);
  int i = 1 / 0;
}

public String testUsingNewDataSource(){
    DynamicDataSource dynamicDataSource = RequestHandleMethodRegistry.getContext().getBean("dynamicDataSource", DynamicDataSource.class);
    try {
      //模擬從服務(wù)器讀取數(shù)據(jù)源信息
      //..........................
      //....................
      
      //創(chuàng)建新數(shù)據(jù)源
      HikariDataSource dataSource = (HikariDataSource)          DataSourceBuilder.create().build();
      dataSource.setJdbcUrl("jdbc:mysql://192.168.xxx.xxx:xxxx/xxxxx?......");
      dataSource.setDriverClassName("com.mysql.cj.jdbc.Driver");
      dataSource.setUsername("xxx");
      dataSource.setPassword("xxx");
      
      //DynamicDataSource注冊(cè)新數(shù)據(jù)源
      dynamicDataSource.addDataSource("test_ds_id", dataSource);

      //設(shè)置當(dāng)前線(xiàn)程數(shù)據(jù)源key、添加臨時(shí)數(shù)據(jù)源
      DataSourceUtil.switchDataSource("test_ds_id");

      //數(shù)據(jù)庫(kù)操作（必須在另一個(gè)service實(shí)現(xiàn)，否則無(wú)法控制事務(wù)）
      serviceB.testInsert();
    }
    finally {
      //清理當(dāng)前線(xiàn)程數(shù)據(jù)源key
      DynamicDataSourceContextHolder.removeContextKey();

      //清理剛剛注冊(cè)的數(shù)據(jù)源
      dynamicDataSource.removeDataSource("test_ds_id");

    }
    return "aa";
  }

@Transactional(rollbackFor = Exception.class)
  public void testInsert() {
    AddressPo po = new AddressPo();
    po.setMemberCode("555555555");
    po.setName("李郃");
    po.setPhone("16651694996");
    po.setProvince("江蘇省");
    po.setCity("南京市");
    po.setArea("浦口區(qū)");
    po.setAddress("南京市浦口區(qū)寧六路219號(hào)");
    po.setDef(false);
    po.setCreateBy("23958");
    addressMapper.insert(po);
    //測(cè)試事務(wù)回滾
    int i = 1 / 0;
  }

public class DynamicDataSource extends AbstractRoutingDataSource {
  
      .................
      .................
      .................
  public void removeDataSource(String key){
    this.backupTargetDataSources.remove(key);
    super.setTargetDataSources(this.backupTargetDataSources);
    super.afterPropertiesSet();
  }
  
      .................
      .................
      .................
}

public void updateMuilty(){
   serviceB.updateDb1();
   serviceB.updateDb2();
}

@UsingDataSource(dataSourceId = "slave1")
@Transactional
public void updateDb1(){
  //業(yè)務(wù)邏輯......
}

@UsingDataSource(dataSourceId = "slave2")
@Transactional
public void updateDb2(){
  //業(yè)務(wù)邏輯......
}