17370845950

新闻动态

Java实现多数据源动态切换的详细配置教程

配置多个数据源：在spring配置文件中定义多个数据库连接信息。2. 创建数据源配置类：使用@configurationproperties初始化各个数据源bean。3. 实现abstractroutingdatasource：继承该类并重写determinecurrentlookupkey()方法，结合threadlocal实现动态选择。4. 配置dynamicdatasource：将多个数据源注入到dynamicdatasource中，并设置默认数据源。5. 使用aop切换数据源：通过自定义注解和切面实现自动切换。6. 手动切换数据源：调用setdatasource()和cleardatasource()进行手动控制。7. 配置mybatis：确保sqlsessionfactory使用dynamicdatasource。8. 确保事务一致性：通过platformtransactionmanager与@transactional注解管理事务。9. 处理连接失败：添加健康检查、重试机制及日志记录。10. 应用于微服务架构：支持灵活访问不同数据库，适用于读写分离等场景。

Java实现多数据源动态切换，核心在于配置和管理多个数据源，并在运行时根据需要选择合适的数据源进行操作。这通常涉及Spring框架、AbstractRoutingDataSource以及ThreadLocal等技术。

解决方案：

配置多个数据源： 首先，在Spring配置文件（例如application.properties或application.yml）中定义多个数据源。每个数据源对应一个数据库连接配置，包括URL、用户名、密码、驱动类等。

spring.datasource.primary.url=jdbc:mysql://localhost:3306/db_primary?useSSL=false&serverTimezone=UTC
spring.datasource.primary.username=root
spring.datasource.primary.password=password
spring.datasource.primary.driver-class-name=com.mysql.cj.jdbc.Driver

spring.datasource.secondary.url=jdbc:mysql://localhost:3306/db_secondary?useSSL=false&serverTimezone=UTC
spring.datasource.secondary.username=root
spring.datasource.secondary.password=password
spring.datasource.secondary.driver-class-name=com.mysql.cj.jdbc.Driver

创建数据源配置类： 创建一个Java配置类，用于初始化这些数据源Bean。

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 javax.sql.DataSource;

@Configuration
public class DataSourceConfig {

    @Primary
    @Bean(name = "primaryDataSource")
    @ConfigurationProperties(prefix = "spring.datasource.primary")
    public DataSource primaryDataSource() {
        return DataSourceBuilder.create().build();
    }

    @Bean(name = "secondaryDataSource")
    @ConfigurationProperties(prefix = "spring.datasource.secondary")
    public DataSource secondaryDataSource() {
        return DataSourceBuilder.create().build();
    }
}

实现AbstractRoutingDataSource： 创建一个类继承AbstractRoutingDataSource，并重写determineCurrentLookupKey()方法。这个方法负责决定使用哪个数据源。

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

public class DynamicDataSource extends AbstractRoutingDataSource {

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

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

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

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

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

配置DynamicDataSource： 在配置类中配置DynamicDataSource，并将之前定义的数据源注入。

import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.jdbc.datasource.DataSourceTransactionManager;
import org.springframework.transaction.PlatformTransactionManager;

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

@Configuration
public class DynamicDataSourceConfig {

    @Bean
    public DynamicDataSource dynamicDataSource(@Qualifier("primaryDataSource") DataSource primaryDataSource,
                                              @Qualifier("secondaryDataSource") DataSource secondaryDataSource) {
        Map targetDataSources = new HashMap<>();
        targetDataSources.put("primary", primaryDataSource);
        targetDataSources.put("secondary", secondaryDataSource);

        DynamicDataSource dynamicDataSource = new DynamicDataSource();
        dynamicDataSource.setTargetDataSources(targetDataSources);
        dynamicDataSource.setDefaultTargetDataSource(primaryDataSource); // 默认数据源
        return dynamicDataSource;
    }

    @Bean
    public PlatformTransactionManager transactionManager(DynamicDataSource dynamicDataSource) {
        return new DataSourceTransactionManager(dynamicDataSource);
    }
}

使用AOP切换数据源： (可选) 可以使用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.springframework.stereotype.Component;

@Aspect
@Component
public class DataSourceAspect {

    @Before("@annotation(UsePrimaryDataSource)") // 自定义注解
    public void setPrimaryDataSource(JoinPoint joinPoint) {
        DynamicDataSource.setDataSource("primary");
    }

    @Before("@annotation(UseSecondaryDataSource)") // 自定义注解
    public void setSecondaryDataSource(JoinPoint joinPoint) {
        DynamicDataSource.setDataSource("secondary");
    }

    @After("@annotation(UsePrimaryDataSource) || @annotation(UseSecondaryDataSource)")
    public void restoreDataSource(JoinPoint joinPoint) {
        DynamicDataSource.clearDataSource();
    }
}

需要定义UsePrimaryDataSource和UseSecondaryDataSource注解。

手动切换数据源： 在需要切换数据源的地方，调用DynamicDataSource.setDataSource("dataSourceName")方法。完成后，调用DynamicDataSource.clearDataSource()清除ThreadLocal中的数据源信息。

配置Mybatis： 配置Mybatis使用DynamicDataSource。

import org.apache.ibatis.session.SqlSessionFactory;
import org.mybatis.spring.SqlSessionFactoryBean;
import org.mybatis.spring.SqlSessionTemplate;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.core.io.support.PathMatchingResourcePatternResolver;
import org.springframework.core.io.support.ResourcePatternResolver;

import javax.sql.DataSource;

@Configuration
public class MybatisConfig {

    @Autowired
    private DynamicDataSource dynamicDataSource;

    @Bean
    public SqlSessionFactory sqlSessionFactory() throws Exception {
        SqlSessionFactoryBean factoryBean = new SqlSessionFactoryBean();
        factoryBean.setDataSource(dynamicDataSource);
        ResourcePatternResolver resolver = new PathMatchingResourcePatternResolver();
        factoryBean.setMapperLocations(resolver.getResources("classpath:mapper/*.xml")); // mapper文件路径
        return factoryBean.getObject();
    }

    @Bean
    public SqlSessionTemplate sqlSessionTemplate(SqlSessionFactory sqlSessionFactory) throws Exception {
        return new SqlSessionTemplate(sqlSessionFactory);
    }
}

如何确保数据源切换的事务一致性？

使用@Transactional注解结合PlatformTransactionManager可以管理事务。确保PlatformTransactionManager使用的是DynamicDataSource。在切换数据源前开启事务，切换数据源后，事务会自动应用到当前数据源。

如何优雅地处理数据源连接失败的情况？

在DynamicDataSource中，可以添加数据源健康检查机制。定期检查数据源的连接状态，如果连接失败，可以尝试重新连接或者切换到备用数据源。可以使用Spring Retry模块实现自动重试。此外，可以记录数据源连接失败的日志，方便排查问题。

多数据源动态切换在微服务架构中的应用场景？

在微服务架构中，每个服务可能需要访问不同的数据库。多数据源动态切换可以帮助服务灵活地访问不同的数据源，而无需修改代码。例如，一个服务可能需要访问订单数据库、用户数据库和商品数据库。通过多数据源动态切换，可以根据业务需求选择合适的数据源。此外，多数据源动态切换还可以用于读写分离，将读操作路由到只读数据库，将写操作路由到主数据库。

17370845950

关于我们

服务项目

广告推广

案例欣赏