java递归返回树状图

最新推荐文章于 2024-06-06 11:29:47 发布

阳光下的阿牛哥

最新推荐文章于 2024-06-06 11:29:47 发布

阅读量380

点赞数 3

文章标签： java

本文链接：https://blog.csdn.net/weixin_43840872/article/details/106591605

版权

java返回树状图

package com.cmft.service.impl;

import java.util.ArrayList;
import java.util.List;

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import com.cmft.dto.CompanyInfoDTO;
import com.cmft.mapper.RiskExposureInfoMapper;
import com.cmft.service.RiskExposureInfoService;
import com.cmft.utils.BeanMapper;

/**
 * 递归查询树结构
 * 
 * @author 
 * @date 2020年6月5日
 */
@Service
public class RiskExposureInfoServiceImpl implements RiskExposureInfoService {
    @Autowired
    private RiskExposureInfoMapper companyMapper;
    @Autowired
    private BeanMapper beanMapper;
    @Override
    public List<CompanyInfoDTO> getSubsidiary() {
        // 获取第一层级（目录级别）
        List<CompanyInfoDTO> dateList = beanMapper.mapList(companyMapper.getSubsidiary(), CompanyInfoDTO.class);
        List<CompanyInfoDTO> list = new ArrayList<>();
        for (CompanyInfoDTO cp : dateList) {
            CompanyInfoDTO tree = getTree(cp);
            if (tree != null) {
                list.add(tree);
            }
        }
        return list;
    }
    // 获取目录的叶子和子叶子
    public CompanyInfoDTO getTree(CompanyInfoDTO c) {
        List<CompanyInfoDTO> list =
            beanMapper.mapList(companyMapper.getData(c.getCompanyCode()), CompanyInfoDTO.class); 获取叶子的数据
        if (list != null) {// 判断是否还有子叶子
            c.setChildlist(list);
            for (CompanyInfoDTO cd : list) {
                getTree(cd);// 递归找子叶子。。。。
            }
        }
        return c;
    }

}