数据源连起来：无缝漫游数据库

后端

2023-05-22 20:14:18

多数据源：利用 MyBatis Plus 插件简化多数据库操作

什么是多数据源？

在软件开发中，多数据源是指在一个项目中同时使用多个数据库连接池。这种设计模式通常用于以下场景：

不同的业务模块使用不同的数据库
需要对数据进行分库分表
需要进行数据迁移

MyBatis Plus 插件：动态数据源实现原理

MyBatis Plus 插件提供了一种便捷的方法来实现动态数据源。其工作原理主要包括以下步骤：

在 MyBatis Plus XML 配置文件中配置多个数据源。
在 MyBatis Plus 插件类中，根据不同的请求，动态选择使用哪个数据源。

MyBatis Plus 插件：动态数据源使用教程

1. 配置数据源

在 MyBatis Plus XML 配置文件中，配置多个数据源：

<!-- 配置主数据源 -->
<bean id="dataSource" class="com.zaxxer.hikari.HikariDataSource">
  <property name="jdbcUrl" value="jdbc:mysql://127.0.0.1:3306/test" />
  <property name="username" value="root" />
  <property name="password" value="root" />
</bean>

<!-- 配置第二个数据源 -->
<bean id="dataSource2" class="com.zaxxer.hikari.HikariDataSource">
  <property name="jdbcUrl" value="jdbc:mysql://127.0.0.1:3307/test1" />
  <property name="username" value="root" />
  <property name="password" value="root" />
</bean>

2. 配置 MyBatis Plus 插件

编写一个 MyBatis Plus 插件类来动态选择数据源：

@Component
public class DynamicDataSourcePlugin implements Interceptor {

  @Override
  public Object intercept(Invocation invocation) throws Throwable {
    // 根据不同的请求，动态地选择使用哪个数据源
    DynamicDataSourceContextHolder.setDataSourceKey("dataSource");
    return invocation.proceed();
  }

  @Override
  public Object plugin(Object target) {
    return Plugin.wrap(target, this);
  }

  @Override
  public void setProperties(Properties properties) {

  }
}

3. 使用动态数据源

在代码中，通过 DynamicDataSourceContextHolder 设置使用的数据源：

@Service
public class UserService {

  @Autowired
  private UserMapper userMapper;

  public User getUserById(Long id) {
    return userMapper.selectById(id);
  }

  public User getUserByName(String name) {
    // 使用第二个数据源
    DynamicDataSourceContextHolder.setDataSourceKey("dataSource2");
    return userMapper.selectOne(new QueryWrapper<User>().eq("name", name));
  }
}