java mybatis 分流查询 实例

假设我们有一个主数据库和两个从数据库，需要在使用 MyBatis 进行数据库查询时进行分流查询，以下是一个 Java MyBatis 分流查询的示例代码：

首先，在 MyBatis 的配置文件中，我们需要定义两个数据源，一个是主数据库，另一个是从数据库。这里以 XML 配置文件为例：

<environments default="production">
  <environment id="production">
    <transactionManager type="JDBC" />
    <dataSource type="POOLED">
      <property name="driver" value="com.mysql.jdbc.Driver" />
      <property name="url" value="jdbc:mysql://localhost:3306/main_db" />
      <property name="username" value="root" />
      <property name="password" value="root" />
    </dataSource>
  </environment>
  <environment id="from1">
    <transactionManager type="JDBC" />
    <dataSource type="POOLED">
      <property name="driver" value="com.mysql.jdbc.Driver" />
      <property name="url" value="jdbc:mysql://localhost:3306/from_db1" />
      <property name="username" value="root" />
      <property name="password" value="root" />
    </dataSource>
  </environment>
  <environment id="from2">
    <transactionManager type="JDBC" />
    <dataSource type="POOLED">
      <property name="driver" value="com.mysql.jdbc.Driver" />
      <property name="url" value="jdbc:mysql://localhost:3306/from_db2" />
      <property name="username" value="root" />
      <property name="password" value="root" />
    </dataSource>
  </environment>
</environments>

然后，我们需要定义一个动态数据源，用于根据条件选择数据源。可以使用 Spring AOP 或者自定义 MyBatis 插件实现动态数据源。

下面是一个自定义 MyBatis 插件的示例代码：

@Intercepts({
    @Signature(type = Executor.class, method = "query", args = {MappedStatement.class, Object.class, RowBounds.class, ResultHandler.class}),
    @Signature(type = Executor.class, method = "query", args = {MappedStatement.class, Object.class, RowBounds.class, ResultHandler.class, CacheKey.class, BoundSql.class}),
    @Signature(type = Executor.class, method = "update", args = {MappedStatement.class, Object.class})
})
public class DynamicDataSourceInterceptor implements Interceptor {

    private static final ThreadLocal<String> dataSourceHolder = new ThreadLocal<>();

    @Override
    public Object intercept(Invocation invocation) throws Throwable {
        Object[] args = invocation.getArgs();
        MappedStatement ms = (MappedStatement) args[0];
        String dataSource = determineDataSource(ms);
        dataSourceHolder.set(dataSource);
        try {
            return invocation.proceed();
        } finally {
            dataSourceHolder.remove();
        }
    }

    private String determineDataSource(MappedStatement ms) {
        String dataSource = "production";
        if (ms.getId().startsWith("from1")) {
            dataSource = "from1";
        } else if (ms.getId().startsWith("from2")) {
            dataSource = "from2";
        }
        return dataSource;
    }

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

    @Override
    public void setProperties(Properties properties) {
        // do nothing
    }

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

最后，在 MyBatis 的 Mapper 接口中，我们需要加上 `@Mapper` 注解，并在 SQL 语句中使用 `${@packageName.DynamicDataSourceInterceptor@getDataSource()}` 获取当前数据源的名称。例如：

@Mapper
public interface UserMapper {

    @Select("SELECT * FROM user WHERE id = #{id}")
    @ResultMap("userResultMap")
    User getUserById(@Param("id") Long id);

    @Select("SELECT * FROM from_db1.user WHERE id = #{id}")
    @ResultMap("userResultMap")
    User getUserFromDb1ById(@Param("id") Long id);

    @Select("SELECT * FROM from_db2.user WHERE id = #{id}")
    @ResultMap("userResultMap")
    User getUserFromDb2ById(@Param("id") Long id);

}

这样，我们就实现了 Java MyBatis 分流查询的功能。