SpringBoot实现动态数据源配置

发布网友发布时间：2024-10-02 14:39

共1个回答

热心网友时间：2024-11-24 02:01

场景描述：

在一个新项目中，我需要使用两个数据源，分别是行云数据库和OceanBase数据库。根据需求，我需要在不同情况下查询不同的数据源。

二、实现思路

为了实现多数据源并自动切换，我们需要了解SpringBoot自动配置的原理。在此之前，我们通常使用单数据源，而现在需要搞懂动态数据源，就必须先搞懂自动配置。

现在，我们要实现多数据源，并能够自动切换。例如，查询A时连接行云数据库，查询B时连接OceanBase数据库。

由于不能使用DataSourceAutoConfiguration，我们需要使用exclude排除它。然后，Spring Boot提供了AbstractRoutingDataSource抽象类，它可以根据用户定义的规则选择当前的数据源。

AbstractRoutingDataSource是一个抽象类，继承了AbstractDataSource抽象类，而AbstractDataSource实现了DataSource接口。这意味着AbstractRoutingDataSource实际上就是一个DataSource。

AbstractRoutingDataSource中有两个关键方法：设置目标数据源的方法和决定当前使用哪个数据源的方法。我们需要准备两个数据源，并将它们配置到Map targetDataSources中备用。

我们继承AbstractRoutingDataSource并实现抽象方法determineCurrentLookupKey()，这样我们自身也是一个数据源。对于数据源，必然有连接数据库的动作。AbstractRoutingDataSource的getConnection()方法实际上调用determineTargetDataSource()返回的数据源的getConnection()方法，这样我们可以在执行查询之前设置使用的数据源。

三、动态数据源注解@DS

在实现动态数据源之前，我们需要了解一些前置知识点，如Java注解和AOP。动态数据源实际上有现成的依赖包可以使用，可以参考使用。