[后端]通用的递归向下查询方法

NullPointerZZZ

于 2024-04-19 22:34:42 发布

阅读量79

点赞数 1

分类专栏：其他文章标签：其他

本文链接：https://blog.csdn.net/nullPoninter/article/details/137979584

版权

其他专栏收录该内容

6 篇文章 0 订阅

订阅专栏

适用场景

源码:


    /**
     * 从给定的pos中向下加载所有次级数据
     *
     * 这是一个通用的递归查询方法 ,适用业务范围 : 评论,分类,推荐关联
     * 这些业务的表设计都存在嵌套形式 因此可以使用该方法完成通用的递归查询
     *
     * @param poService po的service对象 用于操作mybatis-plus执行数据库查询
     * @param pos 给定的 pos 数据
     * @param vCls vo对象类型
     * @param linkField 次级数据和主数据 的关联关系字段
     * @param customWrapper 用户自定义的wrapper查询条件
     * @return 所有次级数据(包含这次提供的主数据)
     *
     * @param <P>  主级类类对象泛型 (主级类必须实现BasePo接口)
     * @param <V>  次级类对象泛型 (次级类必须实现 RecursionBean接口 , 必须继承自po )
     * @param <M1> 关联关系字段泛型
     */
    @SuppressWarnings("unchecked")
    public static<P ,V, M1> List< V > uploadSubsForPos( IService<P> poService,
                                                        List<P> pos,
                                                        Class<V> vCls,
                                                        SFunction<P, M1> linkField,
                                                        Consumer<LambdaQueryWrapper<P>> customWrapper) {

        //---------------------------------------------构建查询次级数据的条件---------------------------------------------

        //查询次级数据时 条件必定需要主数据所有的主键id 那么把所有主数据的id提取出来
        List<M1> ids =( List<M1> ) ListUtils.collectIds( pos );

        //开始构建查询条件 使用提供的 linkField 和ids 构建初始查询条件
        LambdaQueryWrapper<P> wrapper = MybatisPlusUtils.in( ids , linkField );

        //持续补充用户自定义的条件...
        if(null!=wrapper && null!= customWrapper){
            wrapper = wrapper.and( customWrapper );
        }

        //-------------------------------------------------查询次级数据--------------------------------------------------

        //查询 次级数据
        List<P> allSubs = MybatisPlusUtils.safeList(poService,wrapper);

        //根据提供的linkField 给次级数据分组 方便主数据获取属于自己的次级数据
        Map<M1, List<P>> subMap = ListUtils.groupingBy( allSubs, linkField );

        //------------------------------------------------开始封装主数据-------------------------------------------------

        //pos转vos (主数据类型是没有subs字段的 因此转为vo 才可以封装subs字段)
        List<V> vos = BeanUtils.copyList( pos, vCls );

        //todo 此处的泛型转换相当于 PO --> VO --> VO实现的RecursionBean 为什么可以这么转换?

        //开始遍历封装
        for (V v : vos) {
            if(v instanceof RecursionBean){

                //将这个主数据转为 RecursionBean类 才可以设置subs
                RecursionBean<V> vo = (RecursionBean<V>) v;
                M1 id = (M1) vo.getId();

                //获取这个主数据的次级数据
                List<P> subs = subMap.getOrDefault( id ,new ArrayList<>() );

                //给这个主数据 设置它的次级数据 (此处开始递归... 会封装次级数据的次级数据的次级数据的....)
                vo.setSubs(  uploadSubsForPos( poService , subs ,vCls, linkField, customWrapper ) );

            }else {
                //todo 未正确继承异常 后续统一异常信息提示
                throw new RuntimeException("");
            }
        }
        return vos;
    }