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

阿超

发布于 2022-08-17 20:43:48

2.6K0

发布于 2022-08-17 20:43:48

文章被收录于专栏：快乐阿超快乐阿超

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

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

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

/**
 * 过滤不需要查询的字段
 *
 * @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);
}

使用方式：

LambdaQueryWrapper<Product> wrapper = new LambdaQueryWrapper<>(new Product());
MybatisPlusUtils.filterProperties(wrapper, Product::getDetail, Product::getParams)

或者直接

LambdaQueryWrapper<Product> lambdaQueryWrapper = MybatisPlusUtils.filterProperties(new LambdaQueryWrapper<>(new Product()), Product::getDetail, Product::getParams);

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

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

/**
 * 过滤不需要查询的字段
 *
 * @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;
}

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

本文参与腾讯云自媒体分享计划，分享自作者个人站点/博客。

原始发表：2021-10-12，如有侵权请联系 cloudcommunity@tencent.com 删除

mybatis