使用SpringBoot實現動態數據源配置

一、什麼是動態數據源配置

傳統的數據源配置方式是在應用程序啟動時，通過配置文件定義數據源。此時，所有的數據源都已經被指定，數據源的使用是固定的。而動態數據源配置則可以在應用程序運行時動態地創建新的數據源或切換已有的數據源，從而提高了應用程序的靈活性和可擴展性。

二、使用SpringBoot實現動態數據源配置的原理

SpringBoot中實現動態數據源配置的關鍵是藉助於AOP和Spring的事務管理。在應用啟動時，通過動態代理使用AOP增強了DataSource的實現，使得在每次數據源的獲取時，可以根據當前的事務類型選擇不同的數據源。這些數據源可以在運行時動態創建，並配置到數據源路由中，供後續的數據訪問使用。

三、實現動態數據源配置的步驟

1、引入相關依賴

    <dependencies>
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-jdbc</artifactId>
        </dependency>
        <dependency>
            <groupId>org.apache.commons</groupId>
            <artifactId>commons-dbcp2</artifactId>
        </dependency>
        <dependency>
            <groupId>org.aspectj</groupId>
            <artifactId>aspectjweaver</artifactId>
        </dependency>
    </dependencies>

2、定義數據源路由

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

3、配置數據源信息

@Configuration
public class DataSourceConfig {
    @Bean
    public DataSource dataSource() {
        Map<Object, Object> targetDataSources = new HashMap<>();
        DataSource dataSource1 = DataSourceBuilder.create().url("jdbc:mysql://localhost:3306/db1")
                .driverClassName("com.mysql.jdbc.Driver")
                .username("root")
                .password("123456")
                .build();
        DataSource dataSource2 = DataSourceBuilder.create().url("jdbc:mysql://localhost:3306/db2")
                .driverClassName("com.mysql.jdbc.Driver")
                .username("root")
                .password("123456")
                .build();
        targetDataSources.put("dataSource1", dataSource1);
        targetDataSources.put("dataSource2", dataSource2);
        DynamicDataSource dynamicDataSource = new DynamicDataSource();
        dynamicDataSource.setTargetDataSources(targetDataSources);
        dynamicDataSource.setDefaultTargetDataSource(dataSource1);
        return dynamicDataSource;
    }
}

4、配置切面和事務管理器

@Configuration
@EnableTransactionManagement
public class AOPConfig {
    @Bean(name = "txManager")
    public DataSourceTransactionManager txManager(DataSource dataSource) {
        return new DataSourceTransactionManager(dataSource);
    }

    @Bean
    @ConditionalOnMissingBean
    public DynamicDataSourceInterceptor dynamicDataSourceInterceptor() {
        return new DynamicDataSourceInterceptor();
    }

    @Bean
    public Advisor dynamicAdvisor() {
        AspectJExpressionPointcut pointcut = new AspectJExpressionPointcut();
        pointcut.setExpression("execution(* org.example..service.*.*(..))");
        return new DefaultPointcutAdvisor(pointcut, dynamicDataSourceInterceptor());
    }
}

四、如何使用動態數據源

在具體的業務邏輯中，只需要在需要使用不同數據源的方法上增加@TargetDataSource註解即可：

@Service
public class UserService {

    @Autowired
    private UserMapper userMapper;

    @TargetDataSource(value = "dataSource1")
    public void addUser(User user) {
        userMapper.insert(user);
    }

    @TargetDataSource(value = "dataSource2")
    public void updateUser(User user) {
        userMapper.update(user);
    }
}

總結

通過以上的步驟，我們已經可以實現動態數據源配置了。使用動態數據源配置可以讓我們的應用程序更加靈活和可擴展，能夠更好地適應業務變化的需求。