mybatis-plus过滤不需要查询的字段

2022-08-17 20:43:48 浏览数 (1)

仁义忠信,乐善不倦,此天爵也 。一一孟子

之前写过过滤出需要查询的字段,也简单介绍了下Mybatis-Plusselect函数

今天写了个小函数,可以直接传入不需要查询出来的字段

代码语言:javascript复制
/**
 * 过滤不需要查询的字段
 *
 * @param wrapper   条件构造器
 * @param functions 字段
 * @return com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper<T>
 * @author <achao1441470436@gmail.com>
 * @since 2021/10/12 15:51
 */
@SafeVarargs
public static <T> LambdaQueryWrapper<T> filterProperties(LambdaQueryWrapper<T> wrapper, SFunction<T, Serializable>... functions) {
    return Optional.ofNullable(functions).filter(ArrayUtils::isNotEmpty).map(array -> Arrays.stream(array).map(LambdaUtils::resolve).map(SerializedLambda::getImplMethodName).map(PropertyNamer::methodToProperty).collect(Collectors.toList())).map(properties -> wrapper.select(i -> !properties.contains(i.getProperty()))).orElse(wrapper);
}

使用方式:

代码语言:javascript复制
LambdaQueryWrapper<Product> wrapper = new LambdaQueryWrapper<>(new Product());
MybatisPlusUtils.filterProperties(wrapper, Product::getDetail, Product::getParams)

或者直接

代码语言:javascript复制
LambdaQueryWrapper<Product> lambdaQueryWrapper = MybatisPlusUtils.filterProperties(new LambdaQueryWrapper<>(new Product()), Product::getDetail, Product::getParams);

注意,LambdaQueryWrapper需要使用带实体的有参构造

当然也可以不用,我们只需要稍作修改:

代码语言:javascript复制
/**
 * 过滤不需要查询的字段
 *
 * @param wrapper   条件构造器
 * @param functions 字段
 * @return com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper<T>
 * @author <achao1441470436@gmail.com>
 * @since 2021/10/12 15:51
 */
@SafeVarargs
@SuppressWarnings("unchecked")
public static <T> LambdaQueryWrapper<T> filterProperties(LambdaQueryWrapper<T> wrapper, SFunction<T, Serializable>... functions) {
    Set<SerializedLambda> lambdas = Optional.ofNullable(functions).filter(ArrayUtils::isNotEmpty).map(array -> Arrays.stream(array).map(LambdaUtils::resolve).collect(Collectors.toSet())).orElseGet(Collections::emptySet);
    Set<String> properties = lambdas.stream().map(SerializedLambda::getImplMethodName).map(PropertyNamer::methodToProperty).collect(Collectors.toSet());
    lambdas.stream().findAny().ifPresent(lam -> wrapper.select((Class<T>) lam.getImplClass(), i -> !properties.contains(i.getProperty())));
    return wrapper;
}

这样就可以不用使用带实体的有参构造啦

0 人点赞