Springboot环境中多个DataSource基于自定义注解进行切换使用过程

本文介绍如何在Spring Boot项目中配置主从数据库,并通过注解动态切换数据源,实现读写分离,提高系统可用性。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

前面配置了mysql数据库的主从复制模式,在数据库上实现了master-slave配置,通过这种方式可以实现一主一从,或者一主多从,从而提升系统的高可用。
这是数据库层面的实现。在数据库实现了主从模式之后,我们需要考率的问题就是,在我们的应用代码中,如何将不同的数据库操作按需要分配到不同的数据库去执行。

1.需要的依赖

    implementation 'org.springframework.boot:spring-boot-starter'
    implementation 'org.springframework.boot:spring-boot-starter-web'
    implementation 'org.springframework.boot:spring-boot-starter-aop'
    implementation 'mysql:mysql-connector-java:8.0.25'
    implementation 'com.zaxxer:HikariCP:4.0.3'
    

2.yml配置

在application.yml文件中,数据源相关配置如下:

# 自定义的动态数据源配置
custom:
  datasource:
    - key: master
      type: com.zaxxer.hikari.HikariDataSource
      driver-class-name: com.mysql.cj.jdbc.Driver
      url: jdbc:mysql://192.168.161.114:3306/gts?useSSL=false&autoReconnect=true&characterEncoding=UTF-8&serverTimezone=UTC
      username: gts
      password: mysql
      default: true
    - key: slave1
      type: com.zaxxer.hikari.HikariDataSource
      driver-class-name: com.mysql.cj.jdbc.Driver
      url: jdbc:mysql://192.168.161.115:3306/gts?useSSL=false&autoReconnect=true&characterEncoding=UTF-8&serverTimezone=UTC
      username: gts
      password: mysql
    - key: slave2
      type: com.zaxxer.hikari.HikariDataSource
      driver-class-name: com.mysql.cj.jdbc.Driver
      url: jdbc:mysql://192.168.161.114:3306/gts?useSSL=false&autoReconnect=true&characterEncoding=UTF-8&serverTimezone=UTC
      username: gts
      password: mysql 

在custom.datasource,定义了一组数据源配置。master配置到主库,slave1喝slave2分贝表示多组从库。本文用主库来表示从库。

3.动态数据源配置

数据源定义:

package com.dhb.gts.javacourse.week7.dynamic;

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

public class DynamicDataSource extends AbstractRoutingDataSource {

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

动态数据源切面配置:

package com.dhb.gts.javacourse.week7.dynamic;

import lombok.extern.slf4j.Slf4j;
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.springframework.core.annotation.Order;
import org.springframework.stereotype.Component;

@Aspect
@Order(-1)// 保证该AOP在@Transactional之前执行
@Component
@Slf4j
public class DynamicDataSourceAspect {
	@Before("@annotation(ds)")
	public void changeDataSource(JoinPoint point, TargetDataSource ds) throws Throwable {
		String dsId = ds.name();
		if (!DynamicDataSourceContextHolder.containsDataSource(dsId)) {
			log.error("数据源[{}]不存在,使用默认数据源 > {}", ds.name(), poi
Spring Boot 中,为了在运行时通过自定义注解动态切换数据源,可以利用 Expression Language (EL) 表达式结合 Spring 的 `@Profile` 注解。首先,你需要创建两个或更多的数据源配置,比如 `dataSourceA` 和 `dataSourceB`。然后,在应用上下文中为每个数据源配置一个 profile,例如: ```java @Configuration @Profile("dev") public class DevDataSourceConfig { @Bean(name = "primaryDataSource") public DataSource devDataSource() { // 配置开发环境的数据源... } } @Configuration @Profile("prod") public class ProdDataSourceConfig { @Bean(name = "primaryDataSource") public DataSource prodDataSource() { // 配置生产环境的数据源... } } ``` 接下来,你可以创建一个自定义注解,如 `@DataSourceSwitch`,并在需要切换数据源的地方使用 EL 表达式: ```java import org.springframework.context.annotation.Profile; import org.springframework.stereotype.Component; @Component @Profile("#{@environment.getProperty('spring.datasource.active') == 'dev'} ? 'devDataSource' : 'prodDataSource'") public class MyService { @Autowired private DataSource dataSource; // 这里的数据源会根据 EL 表达式的值自动注入对应的数据源 // ... } ``` 在这个例子中,`#{...}` 是 EL 表达式语法,`@environment.getProperty('spring.datasource.active')` 获取的是应用环境中 `spring.datasource.active` 属性的值,它通常由外部配置管理工具设置。如果属性值为 `'dev'`,则使用 `devDataSource`;否则,使用 `prodDataSource`。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值