Spring AbstractRoutingDatasource 动态数据源的实例讲解

Spring AbstractRoutingDatasource 动态数据源的实例讲解

在实际的应用中，我们可能需要操作多个数据库，例如主数据库和从数据库。如果使用传统的方式，需要在每次操作数据库时都手动指定使用哪个数据源，这样非常繁琐。

Spring提供了AbstractRoutingDataSource类来实现动态数据源的管理，可以在运行时根据需要动态切换数据源，从而实现对多个数据源的自由管理。

具体实现方法：

1. 自定义数据源

首先，在Spring配置文件中定义多个数据源，例如：

<bean id="dataSource1" class="org.apache.commons.dbcp.BasicDataSource" destroy-method="close">
    <property name="driverClassName"><value>com.mysql.jdbc.Driver</value></property>
    <property name="url"><value>jdbc:mysql://localhost:3306/test</value></property>
    <property name="username"><value>root</value></property>
    <property name="password"><value>root</value></property>
</bean>

<bean id="dataSource2" class="org.apache.commons.dbcp.BasicDataSource" destroy-method="close">
    <property name="driverClassName"><value>com.mysql.jdbc.Driver</value></property>
    <property name="url"><value>jdbc:mysql://localhost:3306/test2</value></property>
    <property name="username"><value>root</value></property>
    <property name="password"><value>root</value></property>
</bean>

1. 实现Datasource接口

然后，自定义一个继承AbstractRoutingDataSource的类，重写其determineCurrentLookupKey()方法，该方法返回当前使用的数据源的标识。例如：

public class DynamicDataSource extends AbstractRoutingDataSource {

    @Override
    protected Object determineCurrentLookupKey() {
        return DataSourceContextHolder.getDbType();
    }

}

在上面的代码中，DataSourceContextHolder.getDbType()方法返回当前使用的数据源标识，标识可以是字符串、枚举等类型。

1. 配置

最后，在Spring配置文件中配置DynamicDataSource类，例如：

<bean id="dataSource" class="com.example.DynamicDataSource">
    <property name="defaultTargetDataSource" ref="dataSource1"/>
    <property name="targetDataSources">
        <map key-type="java.lang.Object">
            <entry key="dataSource1" value-ref="dataSource1"/>
            <entry key="dataSource2" value-ref="dataSource2"/>
        </map>
    </property>
</bean>

在上面的代码中，defaultTargetDataSource属性指定默认数据源，targetDataSources属性指定所有数据源。

1. 使用

使用时，可以通过DataSourceContextHolder类设置当前使用的数据源标识，例如：

DataSourceContextHolder.setDbType("dataSource1");

在需要切换数据源时，只需要调用DataSourceContextHolder.SetDbType()方法即可。

示例1

假设我们需要在数据源dataSource1中查询用户，并在数据源dataSource2中查询订单。我们可以按照如下方式实现：

// 查询用户
DataSourceContextHolder.setDbType("dataSource1");
JdbcTemplate jdbcTemplate1 = new JdbcTemplate(dataSource);
List<User> userList = jdbcTemplate1.query("SELECT * FROM user", new BeanPropertyRowMapper<>(User.class));

// 查询订单
DataSourceContextHolder.setDbType("dataSource2");
JdbcTemplate jdbcTemplate2 = new JdbcTemplate(dataSource);
List<Order> orderList = jdbcTemplate2.query("SELECT * FROM order", new BeanPropertyRowMapper<>(Order.class));

在上面的代码中，我们首先通过DataSourceContextHolder.setDbType()方法设置当前使用的数据源，然后使用JdbcTemplate类从数据源中操作数据。

示例2

假设我们需要在不同的方法中使用不同的数据源，我们可以按照如下方式实现：

// Service类
public class UserService {

    private JdbcTemplate jdbcTemplate;

    public UserService(DataSource dataSource) {
        jdbcTemplate = new JdbcTemplate(dataSource);
    }

    public void insertUser(User user) {
        DataSourceContextHolder.setDbType("dataSource1");
        jdbcTemplate.update("INSERT INTO user (id, name) VALUES (?, ?)", user.getId(), user.getName());
    }

    public List<User> getUserList() {
        DataSourceContextHolder.setDbType("dataSource2");
        return jdbcTemplate.query("SELECT * FROM user", new BeanPropertyRowMapper<>(User.class));
    }
}

// Controller类
public class UserController {

    @Autowired
    private UserService userService;

    @PostMapping("/user")
    public void insertUser(User user) {
        userService.insertUser(user);
    }

    @GetMapping("/user")
    public List<User> getUserList() {
        return userService.getUserList();
    }
}

在上面的代码中，我们首先在Service类的构造函数中初始化JdbcTemplate对象，然后在不同的方法中使用不同的数据源。在Controller类中，我们使用@Autowired注入UserService对象，并实现了插入用户和查询用户列表的方法。