若依框架多數(shù)據(jù)源切換方式

更新時間：2024年01月23日 09:06:54 作者：Code quotient

這篇文章主要介紹了若依框架多數(shù)據(jù)源切換方式,具有很好的參考價值,希望對大家有所幫助,如有錯誤或未考慮完全的地方,望不吝賜教

前言

在開發(fā)過程中時常遇到需要使用多個數(shù)據(jù)庫的情況，讀寫分離也好還是不同數(shù)據(jù)庫也好都需要配置多數(shù)據(jù)源在需要的時候能夠切換到對應(yīng)的數(shù)據(jù)源。

若依框架自帶的多數(shù)據(jù)源配置切換即可實現(xiàn)。

使用方法如下：

使用

1.首先yml里要配置正確的主從庫數(shù)據(jù)源

其中的enabled:true 代表的是是否開啟從庫，默認(rèn)是關(guān)閉的需要改成true。

# 數(shù)據(jù)源配置
spring:
    datasource:
        type: com.alibaba.druid.pool.DruidDataSource
        driverClassName: com.mysql.cj.jdbc.Driver
        druid:
            # 主庫數(shù)據(jù)源
            master:
                url: jdbc:mysql://****:3306/tablename?useUnicode=true&characterEncoding=utf8&zeroDateTimeBehavior=convertToNull&useSSL=true&serverTimezone=GMT%2B8
                username: *****
                password: *****
            # 從庫數(shù)據(jù)源
            slave:
                # 從數(shù)據(jù)源開關(guān)/默認(rèn)關(guān)閉
                enabled: true
                url: jdbc:mysql://******:3306/tablename?useUnicode=true&characterEncoding=utf8&zeroDateTimeBehavior=convertToNull&useSSL=true&serverTimezone=GMT%2B8
                username: *****
                password: ******
            # 初始連接數(shù)
            initialSize: 5
            # 最小連接池數(shù)量
            minIdle: 10
            # 最大連接池數(shù)量
            maxActive: 20
            # 配置獲取連接等待超時的時間
            maxWait: 60000
            # 配置間隔多久才進(jìn)行一次檢測，檢測需要關(guān)閉的空閑連接，單位是毫秒
            timeBetweenEvictionRunsMillis: 60000
            # 配置一個連接在池中最小生存的時間，單位是毫秒
            minEvictableIdleTimeMillis: 300000
            # 配置一個連接在池中最大生存的時間，單位是毫秒
            maxEvictableIdleTimeMillis: 900000
            # 配置檢測連接是否有效
            validationQuery: SELECT 1 FROM DUAL
            testWhileIdle: true
            testOnBorrow: false
            testOnReturn: false
            webStatFilter:
                enabled: true
            statViewServlet:
                enabled: true
                # 設(shè)置白名單，不填則允許所有訪問
                allow:
                url-pattern: /druid/*
                # 控制臺管理用戶名和密碼
                login-username: ruoyi
                login-password: 123456
            filter:
                stat:
                    enabled: true
                    # 慢SQL記錄
                    log-slow-sql: true
                    slow-sql-millis: 1000
                    merge-sql: true
                wall:
                    config:
                        multi-statement-allow: true

2.在DataSourceType中配置從庫

若有多個從庫就添加一個 eg: //從庫2 SLAVE2

package com.ruoyi.common.enums;
 
/**
 * 數(shù)據(jù)源
 * 
 * @author ruoyi
 */
public enum DataSourceType
{
    /**
     * 主庫
     */
    MASTER,
 
    /**
     * 從庫
     */
    SLAVE
}

3.配置多數(shù)據(jù)源信息再DruidConfig里

多一個數(shù)據(jù)源就要多一個配置

@Configuration
public class DruidConfig
{
    @Bean
    @ConfigurationProperties("spring.datasource.druid.master")
    public DataSource masterDataSource(DruidProperties druidProperties)
    {
        DruidDataSource dataSource = DruidDataSourceBuilder.create().build();
        return druidProperties.dataSource(dataSource);
    }
 
    @Bean
    @ConfigurationProperties("spring.datasource.druid.slave")
    @ConditionalOnProperty(prefix = "spring.datasource.druid.slave", name = "enabled", havingValue = "true")
    public DataSource slaveDataSource(DruidProperties druidProperties)
    {
        DruidDataSource dataSource = DruidDataSourceBuilder.create().build();
        return druidProperties.dataSource(dataSource);
    }
 
    @Bean(name = "dynamicDataSource")
    @Primary
    public DynamicDataSource dataSource(DataSource masterDataSource)
    {
        Map<Object, Object> targetDataSources = new HashMap<>();
        targetDataSources.put(DataSourceType.MASTER.name(), masterDataSource);
        setDataSource(targetDataSources, DataSourceType.SLAVE.name(), "slaveDataSource");
        return new DynamicDataSource(masterDataSource, targetDataSources);
    }
 
    /**
     * 設(shè)置數(shù)據(jù)源
     *
     * @param targetDataSources 備選數(shù)據(jù)源集合
     * @param sourceName 數(shù)據(jù)源名稱
     * @param beanName bean名稱
     */
    public void setDataSource(Map<Object, Object> targetDataSources, String sourceName, String beanName)
    {
        try
        {
            DataSource dataSource = SpringUtils.getBean(beanName);
            targetDataSources.put(sourceName, dataSource);
        }
        catch (Exception e)
        {
        }
    }

以上信息配置完成后即可在需要切換數(shù)據(jù)源的時候在對應(yīng)的mapper或service實現(xiàn)類上添加

@DataSource(value = DataSourceType.MASTER) 注解

eg：

    @DataSource(value = DataSourceType.SLAVE)//切換數(shù)據(jù)源
    @Override
    public AjaxResult selectNowUserList() {
        List<SysUser> sysUsers = userMapper.selectUserList(new SysUser());
        return AjaxResult.success(sysUsers);
    }

以上就是若依框架自帶的多數(shù)據(jù)源切換配置操作，其原理就是AbstractRoutingDataSource的源碼里通過determineCurrentLookupKey 這個方法返回的key去resolvedDataSources里面找對應(yīng)的DataSource，之后更新一期手動配置一套多數(shù)據(jù)切換。