Springbootマルチデータソースの構成

Springbootマルチデータソースの構成
プロジェクトでは、ライブラリを適用したり、読み書きを分離したりすることができます.そのため、マルチデータソース構成が必要です.
プロファイル

spring:
  datasource:
    oracle:
      username: root
      password: 123456
      driver-class-name: com.mysql.jdbc.Driver
      jdbc-url: jdbc:mysql://localhost:3306/demo?serverTimezone=UTC&useUnicode=true@characterEncoding=utf-8
    mysql:
      username: root
      password: 123456
      driver-class-name: com.mysql.jdbc.Driver
      jdbc-url: jdbc:mysql://192.168.85.111:3306/demo?serverTimezone=UTC&useUnicode=true@characterEncoding=utf-8

プロファイルに複数のデータソースを書き、名前を勝手に付けます.
マルチデータソース列挙クラスの作成

package com.jishengchong.datasource;

public enum DataSourceType {
    ORACLE,
    MYSQL
}

データソース切替処理
ThreadLocalに格納

package com.jishengchong.datasource;

public class DynamicDataSourceContextHolder {

    /**
     *   ThreadLocal    ，ThreadLocal                    ，
     *                      ，               。
     */
    private static final ThreadLocal<String> CONTEXT_HOLDER = new ThreadLocal<>();

    /**
     *        
     * @param dataSourceType
     */
    public static void setDataSourceType(String dataSourceType){
        System.out.printf("   {%s}   ", dataSourceType);
        CONTEXT_HOLDER.set(dataSourceType);
    }

    /**
     *        
     * @return
     */
    public static String getDataSourceType(){
        return CONTEXT_HOLDER.get();
    }

    /**
     *        
     */
    public static void clearDataSourceType(){
        CONTEXT_HOLDER.remove();
    }
}

AbstractRoutingDataSourceの継承

	           AbstractRoutingDataSource。    AbstractRoutingDataSource   ，  determineCurrentLookupKey  ，            。      AbstractRoutingDataSource     defaultTargetDataSource targetDataSources。defaultTargetDataSource       ，targetDataSources（map  ）          。

package com.jishengchong.datasource;

import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource;

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

public class DynamicDataSource extends AbstractRoutingDataSource {

    public DynamicDataSource(DataSource defaultTargetDataSource, Map<Object, Object> targetDataSources) {
        super.setDefaultTargetDataSource(defaultTargetDataSource);
        super.setTargetDataSources(targetDataSources);
        // afterPropertiesSet()        targetDataSources     resolvedDataSources  
        super.afterPropertiesSet();
    }

    /**
     *   Key        
     *
     * @return
     */
    @Override
    protected Object determineCurrentLookupKey() {
        return DynamicDataSourceContextHolder.getDataSourceType();
    }
}

データソースの注入

package com.jishengchong.datasource;

import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty;
import org.springframework.boot.context.properties.ConfigurationProperties;
import org.springframework.boot.jdbc.DataSourceBuilder;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.Primary;

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

@Configuration
public class DataSourceConfig {
    @Bean
    @ConfigurationProperties("spring.datasource.oracle")
    public DataSource oracleDataSource() {
        return DataSourceBuilder.create().build();
    }

    @Bean
    @ConfigurationProperties("spring.datasource.mysql")
    public DataSource mysqlDataSource() {
        return DataSourceBuilder.create().build();
    }
    
    @Bean(name = "dynamicDataSource")
    @Primary
    public DynamicDataSource dataSource(DataSource oracleDataSource, DataSource mysqlDataSource) {
        Map<Object, Object> targetDataSources = new HashMap<>();
        targetDataSources.put(DataSourceType.ORACLE.name(), oracleDataSource);
        targetDataSources.put(DataSourceType.MYSQL.name(), mysqlDataSource);
        return new DynamicDataSource(oracleDataSource, targetDataSources);
    }
}

カスタムマルチデータソース切り替え注記
データ・ソースをブロックする注釈を設定し、特定のクラスに設定したり、特定のメソッドに設定したりできます.

package com.jishengchong.mult;

import java.lang.annotation.*;

@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
@Documented
public @interface DataSource {
    /**
     *        
     */
    DataSourceType value() default DataSourceType.ORACLE;
}

AOPブロッキングを使用

         ，              SQL       ，CONTEXT_HOLDER.set(dataSourceType)                      ，                  。

package com.jishengchong.datasource;

import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.annotation.Around;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Pointcut;
import org.aspectj.lang.reflect.MethodSignature;
import org.springframework.core.annotation.Order;
import org.springframework.stereotype.Component;

import java.lang.reflect.Method;

@Aspect
@Order(1)
@Component
public class DataSourceAspect {

    @Pointcut("@annotation(com.jishengchong.datasource.DataSource)")
    public void dsPointCut() {

    }

    @Around("dsPointCut()")
    public Object around(ProceedingJoinPoint point) throws Throwable {
        MethodSignature signature = (MethodSignature) point.getSignature();
        Method method = signature.getMethod();
        DataSource dataSource = method.getAnnotation(DataSource.class);
        if (dataSource != null) {
            DynamicDataSourceContextHolder.setDataSourceType(dataSource.value().name());
        }
        try {
            return point.proceed();
        } finally {
            //              
            DynamicDataSourceContextHolder.clearDataSourceType();
        }
    }
}