springboot动态切换数据源+jpa

1、application-dev.properties配置

spring.datasource.primary.type=com.alibaba.druid.pool.DruidDataSource
spring.datasource.primary.driver-class-name=com.mysql.cj.jdbc.Driver
spring.datasource.primary.jdbc-url=jdbc:mysql:///project_datahub?useSSL=false&useUnicode=true&characterEncoding=utf-8&serverTimezone=Asia/Shanghai&nullCatalogMeansCurrent=true
spring.datasource.primary.username=root
spring.datasource.primary.password=root

spring.datasource.secondary.type=com.alibaba.druid.pool.DruidDataSource
spring.datasource.secondary.driver-class-name=com.mysql.cj.jdbc.Driver
spring.datasource.secondary.jdbc-url=jdbc:mysql:///project_datahub_second?useSSL=false&useUnicode=true&characterEncoding=utf-8&serverTimezone=Asia/Shanghai&nullCatalogMeansCurrent=true
spring.datasource.secondary.username=root
spring.datasource.secondary.password=root

2、不同请求,不同线程,数据源的隔离类

/**
 * @Description 切换数据源
 * @Author lxk
 * @version V1.0.0
 */
public class DBContextHolder{

    /**
     * ThreadLocal数据隔离,但是并非数据的复制,而是在每一个线程中创建一个新的数据对象,然后每一个线程使用的是不一样的
     */
    private static final ThreadLocal<String> CONTEXT_HOLDER = new ThreadLocal<>();

    public static void setDataSource(String dataSource) {
        CONTEXT_HOLDER.set(dataSource);
    }

    public static String getDataSource() {
        return CONTEXT_HOLDER.get();
    }

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

}

3、spring官方AbstractRoutingDataSource核心类实现

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

public class MyRoutingDataSource extends AbstractRoutingDataSource {

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

}

4、常量配置

/**
 * @Description
 * @Author lxk
 * @version V1.0.0
 */
public interface DataSourceNames {

    String FIRST = "first";

    String SECOND = "second";

}

5、数据源配置

import org.springframework.beans.factory.annotation.Qualifier;
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 org.springframework.jdbc.datasource.DataSourceTransactionManager;
import org.springframework.transaction.PlatformTransactionManager;

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

/**
 * @Description
 * @Author lxk
 * @version V1.0.0
 */
@Configuration
public class DataSourceConfig {

    @Bean(name = "firstDataSource")
    @Qualifier("firstDataSource")
    @ConfigurationProperties(prefix="spring.datasource.primary")
    public DataSource firstDataSource() {
        System.out.println("primary db built");
        return DataSourceBuilder.create().build();
    }

    @Bean(name = "secondDataSource")
    @Qualifier("secondDataSource")
    @ConfigurationProperties(prefix="spring.datasource.secondary")
    public DataSource secondDataSource() {
        System.out.println("secondary db built");
        return DataSourceBuilder.create().build();
    }


    @Primary
    @Bean
    public DataSource dataSource(DataSource firstDataSource, DataSource secondDataSource) {
        Map<Object, Object> targetDataSources = new HashMap<>(5);
        targetDataSources.put(DataSourceNames.FIRST, firstDataSource);
        targetDataSources.put(DataSourceNames.SECOND, secondDataSource);

        MyRoutingDataSource myRoutingDataSource = new MyRoutingDataSource();
        // 如果没有指定数据源自动切换主数据源
        myRoutingDataSource.setDefaultTargetDataSource(firstDataSource);
        myRoutingDataSource.setTargetDataSources(targetDataSources);
        return myRoutingDataSource;
    }
   
}

6、aop通过注解,动态切换数据源

import org.aspectj.lang.JoinPoint;
import org.aspectj.lang.annotation.After;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Before;
import org.aspectj.lang.annotation.Pointcut;
import org.aspectj.lang.reflect.MethodSignature;
import org.springframework.core.Ordered;
import org.springframework.stereotype.Component;

import java.lang.reflect.Method;

/**
 * 多数据源,切面处理类
 * @Author lxk
 * @version V1.0.0
 */
@Aspect
@Component
public class DataSourceAspect implements Ordered {

    @Pointcut("@annotation(com.magus.datahub.master.orghandler.config.CurDataSource)")
    public void dataSourcePointCut() {

    }

    @Before("dataSourcePointCut()")
    public void doBefore(JoinPoint  point){
        MethodSignature signature = (MethodSignature) point.getSignature();
        Method method = signature.getMethod();
        CurDataSource ds = method.getAnnotation(CurDataSource.class);
        if (ds == null) {
            DBContextHolder.setDataSource(DataSourceNames.FIRST);
        } else {
            DBContextHolder.setDataSource(ds.name());
        }
    }

    @After("dataSourcePointCut()")
    public void after(JoinPoint point) {
        //清理掉当前设置的数据源,让默认的数据源不受影响
        DBContextHolder.clearDataSource();
    }

    @Override
    public int getOrder() {
        return -1;
    }
}

7、动态切换数据源,示例

controller示例

@RequestMapping("/ehr")
public class EhrController {

    @Autowired
    private EhrService ehrService;

    @GetMapping("/test/switch")
    public ResponseEntity<List<MasterCompany>> ehrServiceSwitch(){
        return ResponseEntity.ok(ehrService.findAllSwitch());
    }

    @GetMapping("/test")
    public ResponseEntity<List<MasterCompany>> ehrService(){
        return ResponseEntity.ok(ehrService.findAll());
    }
}

service示例

@Service
public class EhrService {

    @Autowired
    private MasterCompanyRepository companyRepository;

    public List<MasterCompany> findAll(){
       return companyRepository.findAll();
    }
    
    @CurDataSource(name = DataSourceNames.SECOND)
    public List<MasterCompany> findAllSwitch(){
        return companyRepository.findAll();
    }
}

事务控制,可参考 https://www.cnblogs.com/jpfss/p/8295692.html

 

 

  • 0
    点赞
  • 11
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
Spring Boot 3(假设这是一个未来的版本,目前Spring Boot的最新稳定版是2.x)中的动态数据源功能允许你在运行时切换或配置不同的数据库连接,这对于需要处理多个数据源或者根据业务需求动态调整数据访问策略的应用非常有用。以下是一些关键点: 1. **DataSource bean**:在Spring Boot中,你可以为每个数据源创建一个`DataSource` bean,并通过`spring.datasource`的配置属性来配置它们,如URL、用户名和密码等。 2. **@ConfigurationProperties**:Spring Boot提供了`@ConfigurationProperties`注解来自动绑定外部配置文件中的属性,这样你就可以轻松地从配置中读取不同数据源的配置信息。 3. **Profile-aware**:使用Spring Profile,你可以根据应用的运行环境(如开发、测试、生产)切换数据源。例如,你可以有一个`dev-datasource.properties`和`prod-datasource.properties`,然后在启动应用时指定哪个配置文件生效。 4. **JdbcTemplate或JPA**:Spring Boot提供了`JdbcTemplate`和`JPA`库来与数据源交互。在支持动态数据源的情况下,你需要确保每次切换数据源时,相关的`JdbcTemplate`或`EntityManagerFactory`也会被正确设置。 5. **AOP(面向切面编程)**:有时,你可能想要在切换数据源的同时执行一些额外的操作,比如日志记录或者性能统计。这时可以使用AOP来编写通知,在数据源切换时自动执行。 相关问题-- 1. 如何在Spring Boot 3中启用动态数据源? 2. 动态数据源在高并发场景下有哪些优势? 3. Spring Boot 3如何管理多个数据源的事务协调?

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值