欧美bbbwbbbw肥妇,免费乱码人妻系列日韩,一级黄片

SpringBoot實現動態(tài)數據源切換的方法總結

 更新時間：2023年12月25日 15:04:09   作者：程序員小富  
項目開發(fā)中經常會遇到多數據源同時使用的場景,比如冷熱數據的查詢等情況,所以接下來本文就來介紹一下如何使用實現自定義注解的形式來實現動態(tài)數據源切換吧

簡介

項目開發(fā)中經常會遇到多數據源同時使用的場景，比如冷熱數據的查詢等情況，我們可以使用類似現成的工具包來解決問題，但在多數據源的使用中通常伴隨著定制化的業(yè)務，所以一般的公司還是會自行實現多數據源切換的功能，接下來一起使用實現自定義注解的形式來實現一下。

基礎配置

yml配置

pom.xml文件引入必要的Jar

<?xml version="1.0" encoding="UTF-8"?>
<project xmlns="http://maven.apache.org/POM/4.0.0"
         xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
         xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
    <modelVersion>4.0.0</modelVersion>
    <parent>
        <groupId>org.springframework.boot</groupId>
        <artifactId>spring-boot-starter-parent</artifactId>
        <version>2.7.6</version>
    </parent>
    <groupId>com.dynamic</groupId>
    <artifactId>springboot-dynamic-datasource</artifactId>
    <version>0.0.1-SNAPSHOT</version>
    <properties>
        <maven.compiler.source>8</maven.compiler.source>
        <maven.compiler.target>8</maven.compiler.target>
        <mybatis.plus.version>3.5.3.1</mybatis.plus.version>
        <mysql.connector.version>8.0.32</mysql.connector.version>
        <druid.version>1.2.6</druid.version>
    </properties>

    <dependencies>
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter</artifactId>
        </dependency>
        <!-- springboot核心包 -->
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-web</artifactId>
        </dependency>
        <!-- mysql驅動包 -->
        <dependency>
            <groupId>com.mysql</groupId>
            <artifactId>mysql-connector-j</artifactId>
            <version>${mysql.connector.version}</version>
        </dependency>
        <!-- lombok工具包 -->
        <dependency>
            <groupId>org.projectlombok</groupId>
            <artifactId>lombok</artifactId>
            <optional>true</optional>
        </dependency>
        <!-- MyBatis Plus -->
        <dependency>
            <groupId>com.baomidou</groupId>
            <artifactId>mybatis-plus-boot-starter</artifactId>
            <version>${mybatis.plus.version}</version>
        </dependency>
        <!-- druid -->
        <dependency>
            <groupId>com.alibaba</groupId>
            <artifactId>druid-spring-boot-starter</artifactId>
            <version>${druid.version}</version>
        </dependency>
        <dependency>
            <groupId>org.apache.commons</groupId>
            <artifactId>commons-lang3</artifactId>
            <version>3.7</version>
        </dependency>
        <dependency>
            <groupId>org.aspectj</groupId>
            <artifactId>aspectjweaver</artifactId>
        </dependency>
    </dependencies>

    <build>
        <plugins>
            <plugin>
                <groupId>org.springframework.boot</groupId>
                <artifactId>spring-boot-maven-plugin</artifactId>
            </plugin>
        </plugins>
    </build>
</project>

管理數據源

我們應用ThreadLocal來管理數據源信息，通過其中內容的get，set，remove方法來獲取、設置、刪除當前線程對應的數據源。

/**
 * ThreadLocal存放數據源變量
 *
 * @author 公眾號：程序員小富
 * @date 2023/11/27 11:02
 */
public class DataSourceContextHolder {

    private static final ThreadLocal<String> DATASOURCE_HOLDER = new ThreadLocal<>();

    /**
     * 獲取當前線程的數據源
     *
     * @return 數據源名稱
     */
    public static String getDataSource() {
        return DATASOURCE_HOLDER.get();
    }

    /**
     * 設置數據源
     *
     * @param dataSourceName 數據源名稱
     */
    public static void setDataSource(String dataSourceName) {
        DATASOURCE_HOLDER.set(dataSourceName);
    }

    /**
     * 刪除當前數據源
     */
    public static void removeDataSource() {
        DATASOURCE_HOLDER.remove();
    }
}

重置數據源

創(chuàng)建 DynamicDataSource 類并繼承 AbstractRoutingDataSource，這樣我們就可以重置當前的數據庫路由，實現切換成想要執(zhí)行的目標數據庫。

import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource;
import javax.sql.DataSource;
import java.util.Map;

/**
 * 重置當前的數據庫路由，實現切換成想要執(zhí)行的目標數據庫
 *
 * @author 公眾號：程序員小富
 * @date 2023/11/27 11:02
 */
public class DynamicDataSource extends AbstractRoutingDataSource {

    public DynamicDataSource(DataSource defaultDataSource, Map<Object, Object> targetDataSources) {
        super.setDefaultTargetDataSource(defaultDataSource);
        super.setTargetDataSources(targetDataSources);
    }

    /**
     * 這一步是關鍵，獲取注冊的數據源信息
     * @return
     */
    @Override
    protected Object determineCurrentLookupKey() {
        return DataSourceContextHolder.getDataSource();
    }
}

配置數據庫

在 application.yml 中配置數據庫信息，使用dynamic_datasource_1、dynamic_datasource_2兩個數據庫

spring:
  datasource:
    type: com.alibaba.druid.pool.DruidDataSource
    druid:
      master:
        url: jdbc:mysql://127.0.0.1:3306/dynamic_datasource_1?useUnicode=true&characterEncoding=utf-8&useSSL=false&serverTimezone=Asia/Shanghai&allowPublicKeyRetrieval=true
        username: root
        password: 12345
        driver-class-name: com.mysql.cj.jdbc.Driver
      slave:
        url: jdbc:mysql://127.0.0.1:3306/dynamic_datasource_2?useUnicode=true&characterEncoding=utf-8&useSSL=false&serverTimezone=Asia/Shanghai&allowPublicKeyRetrieval=true
        username: root
        password: 12345
        driver-class-name: com.mysql.cj.jdbc.Driver

再將多個數據源注冊到DataSource.

import com.alibaba.druid.spring.boot.autoconfigure.DruidDataSourceBuilder;
import org.springframework.boot.context.properties.ConfigurationProperties;
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;

/**
 * 注冊多個數據源
 *
 * @author 公眾號：程序員小富
 * @date 2023/11/27 11:02
 */
@Configuration
public class DateSourceConfig {

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

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

    @Bean(name = "dynamicDataSource")
    @Primary
    public DynamicDataSource createDynamicDataSource() {
        Map<Object, Object> dataSourceMap = new HashMap<>();
        // 設置默認的數據源為Master
        DataSource defaultDataSource = dynamicDatasourceMaster();
        dataSourceMap.put("master", defaultDataSource);
        dataSourceMap.put("slave", dynamicDatasourceSlave());
        return new DynamicDataSource(defaultDataSource, dataSourceMap);
    }
}

啟動類配置

在啟動類的@SpringBootApplication注解中排除DataSourceAutoConfiguration，否則會報錯。

@SpringBootApplication(exclude = DataSourceAutoConfiguration.class)

到這多數據源的基礎配置就結束了，接下來測試一下

測試切換

準備SQL

創(chuàng)建兩個庫dynamic_datasource_1、dynamic_datasource_2，庫中均創(chuàng)建同一張表 t_dynamic_datasource_data。

CREATE TABLE `t_dynamic_datasource_data` (
  `id` bigint(20) NOT NULL AUTO_INCREMENT,
  `source_name` varchar(255) DEFAULT NULL,
  PRIMARY KEY (`id`)
);

dynamic_datasource_1.t_dynamic_datasource_data表中插入

insert into t_dynamic_datasource_data (source_name) value ('dynamic_datasource_master');

dynamic_datasource_2.t_dynamic_datasource_data表中插入

insert into t_dynamic_datasource_data (source_name) value ('dynamic_datasource_slave');

手動切換數據源

這里我準備了一個接口來驗證，傳入的 datasourceName 參數值就是剛剛注冊的數據源的key。

/**
 * 動態(tài)數據源切換
 *
 * @author 公眾號：程序員小富
 * @date 2023/11/27 11:02
 */
@RestController
public class DynamicSwitchController {

    @Resource
    private DynamicDatasourceDataMapper dynamicDatasourceDataMapper;

    @GetMapping("/switchDataSource/{datasourceName}")
    public String switchDataSource(@PathVariable("datasourceName") String datasourceName) {
        DataSourceContextHolder.setDataSource(datasourceName);
        DynamicDatasourceData dynamicDatasourceData = dynamicDatasourceDataMapper.selectOne(null);
        DataSourceContextHolder.removeDataSource();
        return dynamicDatasourceData.getSourceName();
    }
}

傳入參數master時：127.0.0.1:9004/switchDataSource/master

傳入參數slave時：127.0.0.1:9004/switchDataSource/slave

通過執(zhí)行結果，我們看到傳遞不同的數據源名稱，已經實現了查詢對應的數據庫數據。

注解切換數據源

上邊已經成功實現了手動切換數據源，但這種方式頂多算是半自動，下邊我們來使用注解方式實現動態(tài)切換。

定義注解

我們先定一個名為DS的注解，作用域為METHOD方法上，由于@DS中設置的默認值是：master，因此在調用主數據源時，可以不用進行傳值。

/**
 * 定于數據源切換注解
 *
 * @author 公眾號：程序員小富
 * @date 2023/11/27 11:02
 */
@Target({ElementType.METHOD, ElementType.TYPE})
@Retention(RetentionPolicy.RUNTIME)
@Documented
@Inherited
public @interface DS {
    // 默認數據源master
    String value() default "master";
}

實現AOP

定義了@DS注解后，緊接著實現注解的AOP邏輯，拿到注解傳遞值，然后設置當前線程的數據源

import com.dynamic.config.DataSourceContextHolder;
import lombok.extern.slf4j.Slf4j;
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.stereotype.Component;
import java.lang.reflect.Method;
import java.util.Objects;

/**
 * 實現@DS注解的AOP切面
 *
 * @author 公眾號：程序員小富
 * @date 2023/11/27 11:02
 */
@Aspect
@Component
@Slf4j
public class DSAspect {

    @Pointcut("@annotation(com.dynamic.aspect.DS)")
    public void dynamicDataSource() {
    }

    @Around("dynamicDataSource()")
    public Object datasourceAround(ProceedingJoinPoint point) throws Throwable {
        MethodSignature signature = (MethodSignature) point.getSignature();
        Method method = signature.getMethod();
        DS ds = method.getAnnotation(DS.class);
        if (Objects.nonNull(ds)) {
            DataSourceContextHolder.setDataSource(ds.value());
        }
        try {
            return point.proceed();
        } finally {
            DataSourceContextHolder.removeDataSource();
        }
    }
}

測試注解

再添加兩個接口測試，使用@DS注解標注，使用不同的數據源名稱，內部執(zhí)行相同的查詢條件，看看結果如何？

@DS(value = "master")
@GetMapping("/dbMaster")
public String dbMaster() {
    DynamicDatasourceData dynamicDatasourceData = dynamicDatasourceDataMapper.selectOne(null);
    return dynamicDatasourceData.getSourceName();
}


@DS(value = "slave")
@GetMapping("/dbSlave")
public String dbSlave() {
    DynamicDatasourceData dynamicDatasourceData = dynamicDatasourceDataMapper.selectOne(null);
    return dynamicDatasourceData.getSourceName();
}

通過執(zhí)行結果，看到通過應用@DS注解也成功的進行了數據源的切換。

事務管理

在動態(tài)切換數據源的時候有一個問題是要考慮的，那就是事務管理是否還會生效呢？

我們做個測試，新增一個接口分別插入兩條記錄，其中在插入第二條數據時將值設置超過了字段長度限制，會產生Data too long for column異常。

    /**
     * 驗證一下事物控制
     */
//    @Transactional(rollbackFor = Exception.class)
    @DS(value = "slave")
    @GetMapping("/dbTestTransactional")
    public void dbTestTransactional() {

        DynamicDatasourceData datasourceData = new DynamicDatasourceData();
        datasourceData.setSourceName("test");
        dynamicDatasourceDataMapper.insert(datasourceData);

        DynamicDatasourceData datasourceData1 = new DynamicDatasourceData();
        datasourceData1.setSourceName("testtesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttest");
        dynamicDatasourceDataMapper.insert(datasourceData1);
    }

經過測試發(fā)現執(zhí)行結果如下，即便實現動態(tài)切換數據源，本地事務依然可以生效。

  • 不加上@Transactional注解第一條記錄可以插入，第二條插入失敗
  • 加上@Transactional注解兩條記錄都不會插入成功

本文案例地址：

以上就是SpringBoot實現動態(tài)數據源切換的方法總結的詳細內容，更多關于SpringBoot動態(tài)數據源切換的資料請關注腳本之家其它相關文章！

相關文章

  • Mybatis框架之代理模式(Proxy Pattern)的實現

    Mybatis框架之代理模式(Proxy Pattern)的實現

    本文主要介紹了MyBatis框架中使用代理模式ProxyPattern的原理和實現,文中通過示例代碼介紹的非常詳細,對大家的學習或者工作具有一定的參考學習價值,需要的朋友們下面隨著小編來一起學習學習吧
    2024-11-11
  • SpringBoot整合atomikos實現跨庫事務的詳細方案

    SpringBoot整合atomikos實現跨庫事務的詳細方案

    這篇文章主要介紹了SpringBoot整合atomikos實現跨庫事務,業(yè)務主要涉及政府及企業(yè)且并發(fā)量不大，所以采用XA事務，雖然性能有所損失，但是可以保證數據的強一致性，需要的朋友可以參考下
    2022-06-06
  • Maven Scope 取值范圍小結

    Maven Scope 取值范圍小結

    在Maven項目的pom.xml文件中,通常會給dependency設置scope屬性,本文主要介紹了Maven Scope 取值范圍,具有一定的參考價值,感興趣的可以了解一下
    2024-05-05
  • Java基礎之throw和throws的示例詳解

    Java基礎之throw和throws的示例詳解

    throw是用來拋出一個具體的異常實例,而throws是用來聲明方法可能會拋出哪些類型的異常,是對調用者的一種通知和要求,這篇文章主要介紹了Java基礎：throw和throws的詳解,需要的朋友可以參考下
    2024-06-06
  • java實現順時針打印矩陣

    java實現順時針打印矩陣

    這篇文章主要為大家詳細介紹了java實現順時針打印矩陣的相關資料，具有一定的參考價值，感興趣的小伙伴們可以參考一下
    2019-03-03
  • 淺析Java編程中類和對象的定義

    淺析Java編程中類和對象的定義

    下面小編就為大家?guī)硪黄獪\析Java編程中類和對象的定義。小編覺得挺不錯的，現在分享給大家，也給大家做個參考。一起跟隨小編過來看看吧，祝大家游戲愉快哦
    2016-05-05
  • Sa-Token中的SaSession對象使用學習示例詳解

    Sa-Token中的SaSession對象使用學習示例詳解

    這篇文章主要為大家介紹了Sa-Token中的SaSession對象使用學習示例詳解，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進步，早日升職加薪
    2023-07-07
  • java實現微信支付結果通知

    java實現微信支付結果通知

    這篇文章主要為大家詳細介紹了java實現微信支付結果通知，具有一定的參考價值，感興趣的小伙伴們可以參考一下
    2019-01-01
  • Spring boot如何基于攔截器實現訪問權限限制

    Spring boot如何基于攔截器實現訪問權限限制

    這篇文章主要介紹了Spring boot如何基于攔截器實現訪問權限限制,文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值,需要的朋友可以參考下
    2020-10-10
  • Spring AI 框架中集成 MCP的方法

    Spring AI 框架中集成 MCP的方法

    SpringAIMCP集成Java/Spring與模型上下文協議,支持stdio和SSE兩種通信模式,通過注解@Tool暴露AI工具,提供標準化接口與數據源交互,兼顧本地部署與遠程調用場景,包含配置、工具注冊及性能安全等關鍵實現要點,本文介紹Spring AI框架中如何集成 MCP,感興趣的朋友一起看看吧
    2025-07-07

最新評論