springBoot设置多数据源

1.数据库配置

spring:
  datasource:
    # 默认数据源
    default:
      url: jdbc:mysql://localhost:3306/maindb
      username: default
      password: depassword
      driver-class-name: com.mysql.cj.jdbc.Driver

    # 第二个数据源
    second:
      url: jdbc:mysql://localhost:3306/seconddb
      username: second
      password: secondpassword
      driver-class-name: com.mysql.cj.jdbc.Driver

2.创建数据源Bean

@Configuration
@EnableTransactionManagement
public class MultipleDataSourceConfig {
   @Bean(name ="default")
   @ConfigurationProperties(prefix = "spring.datasource.default")
   public DataSource default(){
      return DataSourceBuilder.create.build():
   }

   @Bean(name ="second")
   @ConfigurationProperties(prefix = "spring.datasource.second")
   public DataSource second(){
      return DataSourceBuilder.create.build():
   }

}

//上下文数据源设置
public class DynamicDataSourceContext(){
  
  private static ThreadLocal<String> contextHolder = null;
  synchronized public static ThreadLocal<String> getInstance() {
     if (contextHolder == null) {
        contextHolder = new ThreadLocal<>();
     }
     return contextHolder;
  }
 public static void setDataSource(String dataSource) {
      getInstance().set(dataSource);
 }
 
 public static String getDataSource(){
     return Objects.isNull(getInstance().get()) ? "default": getDataSource().get();
 }

 public static void clearDataSource(){
    getInstance().remove();
 }
}

//AbstractRoutingDataSource 是 Spring 提供的一个抽象类，用于支持在运行时动态切换数据源。它允许根据特定的标识符来选择要使用的数据源，这样您可以根据需要在不同的情况下选择不同的数据源。

//AbstractRoutingDataSource 中的 determineCurrentLookupKey() 方法是一个抽象方法，您需要在自定义的数据源路由类中覆盖此方法。它的作用是确定当前要使用的数据源的标识符，通常用来从一些上下文信息中获取数据源的标识。

public class DynamicDataSource extends AbstractRoutingDataSource {

   @Override
   protected Object determineCurrentLookupKey(){
        return DynamicDataSourceContext.getDataSource();
   }
}

@Configuration
public class DynamicDataSourceConfig {
   
   @Bean
   @Primary
   public DataSource dataSOurce(@Qualifier("default") DataSource defeult,@Qualifier("second") DataSource second){
      Map<Object,Object> datasource = new HashMap();
      datasource.put("default",defeult);
      datasource.put("second",second);
      DynamicDataSource dynamicDataSource = new DynamicDataSource();
      dynamicDataSource.setTargetDataSources(datasource);
      dynamicDataSource.setDefaultTargetDataSource(default);
      return dynamicDataSource;
   } 
}

3.定义注解，设置具体数据源

import java.lang.annotation.*;

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

//切面，略微有些问题，根据自己需要修改

@Aspect
@Component
public class DataSourceAspect {

    @Pointcut("@annotation(xx.xx.xxx.DynamicDataSourceSwitch)")
    public void myCustomAnnotationPointcut() {
    }

    @Before("myCustomAnnotationPointcut()")
    public void beforeMyCustomAnnotation( JoinPoint point) {
       Signature signature = point.getSignature();
       DynamicDataSourceSwitch switercher;
       if(signature instanceof MethodSignature ms){// java 17写法
           //方法上的注解
           switercher = ms.getMethod().getAnnotation(DynamicDataSourceSwitch.class);
           if(Objects.isNull(switercher)){
             //class上的注解
             switercher = (DynamicDataSourceSwitch)signature.getDeclaringType().getAnnotation(DynamicDataSourceSwitch.class);
           }
           if(Objects.nonNull(switercher)) {
           //设置数据源
             DynamicDataSourceContext.setDataSource(switercher.value);
           }
       }
    }

    @After("myCustomAnnotationPointcut()")
    public void afterMyCustomAnnotation(JoinPoint point) {
        DynamicDataSourceContext.clearDataSource();
    }

    @AfterThrowing(pointcut = "myCustomAnnotationPointcut()")
    public void afterThrowingMyCustomAnnotation(JoinPoint point) {
        DynamicDataSourceContext.clearDataSource();
    }
}

4.使用

@Service
@DynamicDataSourceSwitch("second")
public class TestService {
  public void test(){
  //你的逻辑
  }
}

• 备注：若不想显示指定，还可以在拦截器中拦截，根据路径或者header里的信息来指定数据源

代码007普通

打赏 收藏 海报 链接