MyBatis-Plus多数据源环境下批量插入的表名动态替换问题解析

问题背景

在使用MyBatis-Plus进行数据库操作时，开发者经常会遇到需要动态切换数据源和表名的场景。特别是在多租户系统中，不同租户的数据可能存储在不同的表或不同的数据库中。MyBatis-Plus提供了DynamicTableNameInnerInterceptor拦截器来实现表名的动态替换功能。

问题现象

当开发者同时使用MyBatis-Plus的多数据源功能和批量插入(saveBatch)方法时，可能会遇到表名动态替换失效的问题。具体表现为：

1. 在执行批量插入操作时，只有第一条数据的表名被正确替换
2. 后续数据的表名恢复为原始表名
3. 日志显示每次插入都会触发拦截器的hook方法

问题根源分析

通过分析源码和问题现象，我们发现问题的根本原因在于：

1. 批量操作的处理机制：MyBatis-Plus的saveBatch方法实际上是将批量操作分解为多次单条插入操作
2. 拦截器的执行时机：DynamicTableNameInnerInterceptor的hook方法会在每次SQL执行前被调用
3. 表名上下文管理：问题代码中在hook方法执行后移除了表名下标，导致后续操作无法获取正确的表名

解决方案

针对这个问题，我们有以下几种解决方案：

方案一：修改表名上下文管理逻辑

在hook方法中不要立即移除表名下标，而是确保在整个批量操作完成后才清理上下文：

public class DynamicTableNameHelper {
    private static final ThreadLocal<String> TABLE_NAME = new ThreadLocal<>();
    
    public static void setTableName(String tableName) {
        TABLE_NAME.set(tableName);
    }
    
    public static String getTableName() {
        return TABLE_NAME.get();
    }
    
    // 不提供remove方法，或者只在业务逻辑完成后调用
}

方案二：使用MyBatis原生批量操作

改用MyBatis原生的批量操作方式，避免多次触发拦截器：

@Autowired
private SqlSessionTemplate sqlSessionTemplate;

public void batchInsert(List<Entity> list) {
    SqlSession sqlSession = sqlSessionTemplate.getSqlSessionFactory().openSession(ExecutorType.BATCH, false);
    try {
        Mapper mapper = sqlSession.getMapper(Mapper.class);
        for (Entity entity : list) {
            mapper.insert(entity);
        }
        sqlSession.commit();
    } finally {
        sqlSession.close();
    }
}

方案三：自定义批量插入实现

继承ServiceImpl并重写saveBatch方法，实现真正的批量插入：

public class CustomServiceImpl<M extends BaseMapper<T>, T> extends ServiceImpl<M, T> {
    @Override
    public boolean saveBatch(Collection<T> entityList, int batchSize) {
        String sqlStatement = getSqlStatement(SqlMethod.INSERT_ONE);
        return executeBatch(entityList, batchSize, (sqlSession, entity) -> {
            sqlSession.insert(sqlStatement, entity);
        });
    }
}

最佳实践建议

1. 上下文管理：在多数据源和动态表名场景下，确保上下文信息在整个业务操作期间保持一致
2. 批量操作选择：根据数据量大小选择合适的批量操作方式
3. 拦截器设计：拦截器实现应考虑批量操作的特殊性，避免状态过早清理
4. 日志监控：在开发阶段增加详细的SQL日志，便于排查类似问题

总结

MyBatis-Plus在多数据源环境下批量插入的表名动态替换问题，本质上是一个上下文管理和批量操作分解的综合问题。通过理解MyBatis-Plus的内部工作机制，我们可以采取多种方式解决这个问题。在实际开发中，建议根据具体业务场景选择最适合的解决方案，同时注意保持代码的可维护性和可扩展性。