SpringBoot+Druid配置多数据源（注解切换）

禁止掉DataSourceAutoConfiguration：

@SpringBootApplication(exclude = DataSourceAutoConfiguration.class)
public class DatahubApplication implements CommandLineRunner {

    public static void main(String[] args) {
        SpringApplication.run(DatahubApplication.class, args);
    }

 

yml配置文件:

(注：每个数据源都需完整配置好参数)

spring:
  datasource:
    druid:
      mysql:
        type: com.alibaba.druid.pool.DruidDataSource
        url: jdbc:mysql://10.0.0.10:3306/xxxx?useUnicode=true&useSSL=false&zeroDateTimeBehavior=convertToNull&characterEncoding=utf8&serverTimezone=GMT%2B8
        username: xx
        password: xx
        driver-class-name: com.mysql.cj.jdbc.Driver
        initial-size: 10
        max-active: 100
        min-idle: 10
        max-wait: 60000
        pool-prepared-statements: true
        max-pool-prepared-statement-per-connection-size: 20
        time-between-eviction-runs-millis: 60000
        min-evictable-idle-time-millis: 300000
        #Oracle需要打开注释
        #validation-query: SELECT 1 FROM DUAL
        test-while-idle: true
        test-on-borrow: false
        test-on-return: false
        stat-view-servlet:
          enabled: true
          url-pattern: /druid/*
          #login-username: admin
          #login-password: admin
        filter:
          stat:
            log-slow-sql: true
            slow-sql-millis: 1000
            merge-sql: false
          wall:
            config:
              multi-statement-allow: true
      db2:
        type: com.alibaba.druid.pool.DruidDataSource
        url: jdbc:db2://10.3.2.5:50000/micszhc
        username: xx
        password: xx
        driver-class-name: com.ibm.db2.jcc.DB2Driver
        initial-size: 10
        max-active: 100
        min-idle: 10
        max-wait: 60000
        pool-prepared-statements: true
        max-pool-prepared-statement-per-connection-size: 20
        time-between-eviction-runs-millis: 60000
        min-evictable-idle-time-millis: 300000
        #Oracle需要打开注释
        #validation-query: SELECT 1 FROM DUAL
        test-while-idle: true
        test-on-borrow: false
        test-on-return: false
        stat-view-servlet:
          enabled: true
          url-pattern: /druid/*
          #login-username: admin
          #login-password: admin
        filter:
          stat:
            log-slow-sql: true
            slow-sql-millis: 1000
            merge-sql: false
          wall:
            config:
              multi-statement-allow: true

 

生成@DataSource注解,在Mapper层中使用(防止在service出现调用其他service层的方法，防止同时用到两种数据源存在冲突)

 

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

 

 自定义数据源常量：

public class DataSourceConstants {

    public static final String MYSQL = "mysql";

    public static final String DB2 = "db2";

}

自定义DynamicDataSourceConfig：注入数据源

import javax.sql.DataSource;
@Configuration
public class DynamicDataSourceConfig {

    @Bean
    @ConfigurationProperties("spring.datasource.druid.mysql")
    public DataSource mysqlDataSource() {
        return DruidDataSourceBuilder.create().build();
    }

    @Bean
    @ConfigurationProperties("spring.datasource.druid.db2")
    public DataSource db2lDataSource() {
        return DruidDataSourceBuilder.create().build();
    }


    @Bean
    @Primary
    public DynamicDataSource dataSource(DataSource mysqlDataSource, DataSource db2lDataSource) {
        Map<Object, Object> targetDataSource = new HashMap<>();
        targetDataSource.put(DataSourceConstants.MYSQL, mysqlDataSource);
        targetDataSource.put(DataSourceConstants.DB2, db2lDataSource);
        return new DynamicDataSource(mysqlDataSource, targetDataSource);
    }
}

 自定义DynamicDataSource：

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();
    }
}

 自定义DataSourceAspect：

@Aspect
@Component
public class DataSourceAspect implements Ordered {

    protected Logger logger = LoggerFactory.getLogger(getClass());

    @Pointcut("@annotation(com.pci.imc.datahub.common.constants.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(DataSourceConstants.MYSQL);
            logger.info("set mysql");
        }else{
            DynamicDataSource.setDataSource(ds.name());
            logger.info("set datasource");
        }

        try{
           return point.proceed();
        }finally {
            DynamicDataSource.clearDataSource();
            logger.info("clear datasource");
        }
    }
    @Override
    public int getOrder() {
        return 1;
    }
}

 Mapper层只需在方法上添加注解@DataSource(name = DB2) 就行了，不添加默认使用第一个数据源mysql

   @DataSource(name = DB2)
    List<Map<String, Object>> getLevel5Devices(@Param("name") String name, @Param("locationId") Integer locationId);