DynamicDataSource是一個(gè)數(shù)據(jù)源路由器，可以根據(jù)上下文動態(tài)選擇數(shù)據(jù)源?？梢栽诿總€(gè)請求或線程中將數(shù)據(jù)源設(shè)置為當(dāng)前需要使用的數(shù)據(jù).

1. 創(chuàng)建一個(gè)DynamicDataSource類來實(shí)現(xiàn)數(shù)據(jù)源路由邏輯

創(chuàng)建一個(gè)DynamicDataSource類，它繼承自AbstractRoutingDataSource。在該類中重寫**determineCurrentLookupKey()**方法，該方法返回一個(gè)字符串，用于指示當(dāng)前要使用哪個(gè)數(shù)據(jù)源

public class DynamicDataSource extends AbstractRoutingDataSource {
    @Override
    protected Object determineCurrentLookupKey() {
        return DynamicDataSourceContextHolder.getDataSourceName();
    }

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

2. 創(chuàng)建DynamicDataSourceContextHolder線程安全類

在多線程環(huán)境下，如果多個(gè)線程同時(shí)訪問同一個(gè)方法，并且每個(gè)線程要使用不同的數(shù)據(jù)源，那么就需要對數(shù)據(jù)源進(jìn)行動態(tài)切換。如果在方法中使用一個(gè)共享的變量來存儲當(dāng)前要使用的數(shù)據(jù)源名稱，那么就會存在線程安全問題，可能會導(dǎo)致不同線程之間的數(shù)據(jù)源切換混亂，或者數(shù)據(jù)源切換不成功的情況發(fā)生。
為了解決這個(gè)問題，可以使用ThreadLocal來存儲當(dāng)前要使用的數(shù)據(jù)源名稱。ThreadLocal是一種線程本地存儲機(jī)制，它可以為每個(gè)線程提供一個(gè)獨(dú)立的變量副本，使得每個(gè)線程都可以獨(dú)立地操作自己的變量副本，而不會影響其他線程的變量副本。
在使用DynamicDataSource進(jìn)行數(shù)據(jù)源切換時(shí)，每個(gè)線程都可以通過ThreadLocal來獨(dú)立地設(shè)置和獲取當(dāng)前要使用的數(shù)據(jù)源名稱，避免了多個(gè)線程之間數(shù)據(jù)源切換的混亂和不成功的情況。同時(shí)，在方法執(zhí)行完畢后，使用ThreadLocal也可以避免內(nèi)存泄漏問題。
使用ThreadLocal來存儲當(dāng)前要使用的數(shù)據(jù)源名稱。**setDataSource()**方法用于設(shè)置當(dāng)前要使用的數(shù)據(jù)源名稱，**getDataSource()**方法用于獲取當(dāng)前要使用的數(shù)據(jù)源名稱，**clearDataSource()**方法用于清除當(dāng)前要使用的數(shù)據(jù)源名稱

public class DynamicDataSourceContextHolder {

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

    /**
     * 設(shè)置當(dāng)前數(shù)據(jù)源的名稱
     */
    public static void setDataSourceName(String dataSourceName) {
        contextHolder.set(dataSourceName);
    }

    /**
     * 獲取當(dāng)前數(shù)據(jù)源的名稱
     */
    public static String getDataSourceName() {
        return contextHolder.get();
    }

    /**
     * 清除當(dāng)前數(shù)據(jù)源的名稱
     */
    public static void clearDataSourceName() {
        contextHolder.remove();
    }

}

3.創(chuàng)建多數(shù)據(jù)源配置

通過yml配置,將多數(shù)據(jù)源獲取到MutilDataSourceProperties中

spring:
  datasource:
    driver-class-name: com.mysql.cj.jdbc.Driver
    url: jdbc:mysql://****:3306/emp_ts?nullNamePatternMatchesAll=true&tinyInt1isBit=false&useSSL=false&nullCatalogMeansCurrent=true&useUnicode=true&characterEncoding=UTF-8&serverTimezone=GMT%2B8
    username: root
    password: root
mutil-datasource:
  connection:
    - dbName: dataSource1
      dbDriver: com.mysql.cj.jdbc.Driver
      dbUrl: jdbc:mysql://****:3306/emp_ts_1?nullNamePatternMatchesAll=true&tinyInt1isBit=false&useSSL=false&nullCatalogMeansCurrent=true&useUnicode=true&characterEncoding=UTF-8&serverTimezone=GMT%2B8
      dbUsername: root
      dbPassword: root

public class DbConnection {
    private String dbName;
    private String dbDialect;
    private String dbDriver;
    private String dbUrl;
    private String dbUsername;
    private String dbPassword;

    public DbConnection() {
    }

    public String getDbName() {
        return this.dbName;
    }

    public String getDbDialect() {
        return this.dbDialect;
    }

    public String getDbDriver() {
        return this.dbDriver;
    }

    public String getDbUrl() {
        return this.dbUrl;
    }

    public String getDbUsername() {
        return this.dbUsername;
    }

    public String getDbPassword() {
        return this.dbPassword;
    }

    public void setDbName(String dbName) {
        this.dbName = dbName;
    }

    public void setDbDialect(String dbDialect) {
        this.dbDialect = dbDialect;
    }

    public void setDbDriver(String dbDriver) {
        this.dbDriver = dbDriver;
    }

    public void setDbUrl(String dbUrl) {
        this.dbUrl = dbUrl;
    }

    public void setDbUsername(String dbUsername) {
        this.dbUsername = dbUsername;
    }

    public void setDbPassword(String dbPassword) {
        this.dbPassword = dbPassword;
    }
}

@Configuration
@ConfigurationProperties(
    prefix = "mutil-datasource"
)
public class MutilDataSourceProperties {
    private List<DbConnection> connection = new ArrayList();

    public MutilDataSourceProperties() {
    }

    public List<DbConnection> getConnection() {
        return this.connection;
    }

    public void setConnection(List<DbConnection> connection) {
        this.connection = connection;
    }
}

4.創(chuàng)建DataSourceConfig

在配置類中創(chuàng)建默認(rèn)數(shù)據(jù)源及獲取其他多數(shù)據(jù)源進(jìn)行創(chuàng)建,默認(rèn)數(shù)據(jù)源為spring.datasource下配置的數(shù)據(jù)源

@Configuration
public class DataSourceConfig {
    @Resource
    private MutilDataSourceProperties mutilDataSourceProperties;

    public DataSourceConfig() {
    }

    @Bean
    @ConfigurationProperties("spring.datasource")
    public DataSource hikariDataSource(DataSourceProperties properties) {
        return DataSourceBuilder.create(properties.getClassLoader()).driverClassName(properties.determineDriverClassName()).url(properties.determineUrl()).username(properties.determineUsername()).password(properties.determinePassword()).build();
    }

    @Bean
    @Primary
    public DynamicDataSource dynamicDataSource(DataSource hikariDataSource) {
        Map<Object, Object> targetDataSources = this.createTargetDataSource();
        return new DynamicDataSource(hikariDataSource, targetDataSources);
    }

     private Map<Object, Object> createTargetDataSource() {
        Map<Object, Object> targetDataSources = new HashMap();
        List<DbConnection> connections = this.mutilDataSourceProperties.getConnection();
        connections.forEach(e -> {
            DataSource dataSource = this.createDataSource(e);
            targetDataSources.put(e.getDbName(), dataSource);
        });
        return targetDataSources;
     }

     public DataSource createDataSource(DbConnection connection) {
        return DataSourceBuilder.create().driverClassName(connection.getDbDriver()).url(connection.getDbUrl())
                .username(connection.getDbName()).password(connection.getDbPassword()).build();
     }
}

5. 使用

@Service
public class UserServiceImpl implements UserService {

    @Autowired
    private UserMapper userMapper;
    
    @Override
    public void addUser(User user) {
        DynamicDataSource.setDataSource("dataSource1");
        userMapper.insert(user);
    }

    @Override
    public void updateUser(User user) {
        DynamicDataSource.setDataSource("dataSource1");
        userMapper.updateById(user);
    }
}

6.優(yōu)缺點(diǎn)

優(yōu)點(diǎn)：

• 簡單易用：使用 DynamicDataSource 和 ThreadLocal 進(jìn)行動態(tài)數(shù)據(jù)源切換，配置相對簡單，易于上手。
• 可擴(kuò)展性強(qiáng)：通過繼承 AbstractRoutingDataSource 類，可以實(shí)現(xiàn)自定義的數(shù)據(jù)源路由策略。并且，由于采用了抽象類，擴(kuò)展性也比較好。
• 線程安全：使用 ThreadLocal 來存儲當(dāng)前數(shù)據(jù)源的名稱，可以避免多線程之間數(shù)據(jù)源切換的混亂和不成功的情況。

缺點(diǎn)：

• 不能同時(shí)訪問多個(gè)數(shù)據(jù)源：在使用 DynamicDataSource 進(jìn)行動態(tài)數(shù)據(jù)源切換時(shí)，同一時(shí)間只能訪問一個(gè)數(shù)據(jù)源，不能同時(shí)訪問多個(gè)數(shù)據(jù)源。

• 執(zhí)行效率稍低：在使用 DynamicDataSource 進(jìn)行動態(tài)數(shù)據(jù)源切換時(shí)，每次數(shù)據(jù)源切換都需要進(jìn)行一次路由選擇，會稍微影響執(zhí)行效率。

• 不支持事務(wù)嵌套：在使用 DynamicDataSource 進(jìn)行動態(tài)數(shù)據(jù)源切換時(shí)，如果在一個(gè)事務(wù)中需要訪問多個(gè)數(shù)據(jù)源，那么就需要進(jìn)行事務(wù)管理，而 DynamicDataSource 并不支持事務(wù)嵌套。

  綜上所述，使用 DynamicDataSource 和 ThreadLocal 進(jìn)行動態(tài)數(shù)據(jù)源切換，優(yōu)點(diǎn)是簡單易用、可擴(kuò)展性強(qiáng)、線程安全，缺點(diǎn)是不能同時(shí)訪問多個(gè)數(shù)據(jù)源、執(zhí)行效率稍低、不支持事務(wù)嵌套。在具體使用時(shí)需要根據(jù)業(yè)務(wù)場景進(jìn)行選擇。文章來源地址http://www.zghlxwxcb.cn/news/detail-446553.html

到了這里，關(guān)于springboot使用DynamicDataSource來動態(tài)切換數(shù)據(jù)源的文章就介紹完了。如果您還想了解更多內(nèi)容，請?jiān)谟疑辖撬阉鱐OY模板網(wǎng)以前的文章或繼續(xù)瀏覽下面的相關(guān)文章，希望大家以后多多支持TOY模板網(wǎng)！