MyBatis-Plus中LambdaQueryWrapper执行count()方法时获取实体类的问题解析

问题背景

在使用MyBatis-Plus进行开发时，开发者经常需要执行count()查询来获取记录数。当使用LambdaQueryWrapper构建查询条件时，如果需要对查询字段进行加密处理，就需要获取实体类的信息。然而，在某些情况下，开发者发现无法从LambdaQueryWrapper中获取到实体类对象。

问题现象

具体表现为：当使用count(new LambdaQueryWrapper<HedgeExchange>().eq(xxx))>0这种写法时，在拦截器中尝试通过queryWrapper.getEntityClass()获取实体类时返回null，导致无法进行后续的加密处理。

原因分析

MyBatis-Plus的LambdaQueryWrapper提供了两种构造方式：

1. 无参构造：new LambdaQueryWrapper<HedgeExchange>()
2. 带实体类参数的构造：new LambdaQueryWrapper<>(HedgeExchange.class)

当使用第一种无参构造方式时，LambdaQueryWrapper内部的entityClass字段确实会被初始化为null。这是因为泛型类型信息在运行时会被擦除，MyBatis-Plus无法直接从泛型参数中获取实际的实体类类型。

解决方案

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

方案一：使用带实体类参数的构造方法

count(new LambdaQueryWrapper<>(HedgeExchange.class).eq(xxx))>0

这种方式明确指定了实体类，可以确保在任何情况下都能正确获取到实体类信息。

方案二：设置实体类实例

LambdaQueryWrapper<HedgeExchange> wrapper = new LambdaQueryWrapper<>();
wrapper.setEntity(new HedgeExchange());
count(wrapper.eq(xxx))>0

通过设置实体实例，也可以让Wrapper正确识别实体类类型。

方案三：自定义获取实体类的方法

如果必须使用无参构造方式，可以尝试通过其他途径获取实体类信息：

private Class<?> getEntityClass(AbstractWrapper queryWrapper) {
    Class<?> entityClass = queryWrapper.getEntityClass();
    if (entityClass == null) {
        // 尝试从其他途径获取实体类信息
        // 例如从SQL语句中解析表名再映射回实体类
    }
    return entityClass;
}

最佳实践建议

1. 推荐使用带实体类参数的构造方法：这是最直接和可靠的方式，代码意图明确，不易出错。

2. 保持一致性：在整个项目中统一使用一种构造方式，避免混用导致理解困难。

3. 考虑封装工具类：如果需要频繁处理加密逻辑，可以封装一个工具类来统一处理Wrapper的创建和实体类获取。

4. 注意泛型类型擦除：理解Java泛型在运行时的类型擦除特性，避免依赖运行时不可用的类型信息。

总结

MyBatis-Plus作为优秀的ORM框架，提供了灵活的查询构建方式。理解其内部机制有助于我们更好地使用它。在需要获取实体类信息的场景下，明确指定实体类是最可靠的做法。通过本文的分析，希望开发者能够避免类似的问题，写出更健壮的代码。