Spring Boot中利用递归算法查询到所有下级用户，并手动进行分页

Spring Boot中利用递归算法查询到所有下级用户，并手动进行分页

前提：语言用的是kotlin（和Java一样，但更简洁），写下这篇文章用来记录编程过程中遇到的一些难点

1、功能需求

前端用户A点击“我的推荐”后，调用后台的接口，查询到所有的下家（不仅包括直接下家）

如图所示，当前用户A要查询所有下家用户，就是以自己为树的“根”节点，遍历所有下家用户，并返回所有下家：B、C、D、E···。

2、代码实现

//我的推荐
    fun queryLower(request: QueryLowerRequestModel,userToken: UserToken): PageData2<QueryLowerResponseModel> {
        //创建一个list，用于存储递归后的返回
        var mutableList = mutableListOf<String>()
        GetMyRecommend(userToken.userId!!,mutableList)

        var responseList = mutableListOf<QueryLowerResponseModel>()

        var userAsset = assetRepository.findAll()
        for(i in mutableList) {
            userAsset.forEach { e ->
                if (e.username == i)
                {
                    responseList.add(QueryLowerResponseModel(
                            userName = i,
                            baseMoney = e.base_money,
                            create_time = e.base_money_time,
                            isDirect = 0
                    ))
                }
            }
        }
        var toIndex = (request.pageNo+1)*request.size-1
        if( (request.pageNo+1)*request.size > mutableList.size)
        {
            toIndex = mutableList.size
        }
        var totalPages :Int = (mutableList.size / request.size)+1

        return PageData2(
                list = responseList.subList(request.pageNo*request.size,toIndex),
                totalNum = mutableList.size.toLong(),
                pageNo = request.pageNo,
                totalPages = totalPages,
                hasMore = false
        )
    }

其中的GetMyRecommend函数是我自定义的一个递归函数，函数体如下：

    //递归获取我的推荐
    fun GetMyRecommend(id:Long ,nameList:MutableList<String> )
    {

        var userList = usersRepository.selectLowerUsers(id)
        if(userList.isEmpty())
        {
            return
        }
        else
        {
            for( i in userList)
            {
                nameList.add(i)
                var userId = usersRepository.findUserEntityByUsername(i)
                GetMyRecommend(userId.id,nameList)
            }
        }
    }

可以看到，在递归过程中，每次我都进行了数据库的操作，虽然这样的效率很低，但在敏捷开发过程中，首要的任务就是将功能实现，其次再考虑性能的问题。这个递归我利用了list传地址的特性，每次对list操作都会改变list中的值，这样就记录下来了所有下家。其次进行分页（分页是基础操作），并返还给前端。