树

  接口

 /**
     * 获取公共机构树
     * @return
     */
    public List<ThreeWyfwData> getThreeWyList(List<WyfwBuildingManage> blist,List<WyfwOwnerManage> wlist) {

        List<WyfwBuildingManage> returnList = new ArrayList<>();
        HashSet<String> tempSet = new HashSet<>();
        for (WyfwBuildingManage bu : blist) {
            tempSet.add(bu.getId());
        }
        for (WyfwBuildingManage bu : blist) {
            // 如果是顶级节点, 遍历该父节点的所有子节点
            if (!tempSet.contains(bu.getParentId())) {
                recursionFn(blist, bu);
                returnList.add(bu);
            }
        }
        Map<String, List<WyfwOwnerManage>> devMapList = wlist.stream()
                .collect(Collectors.groupingBy(WyfwOwnerManage::getBuildingId));
        for (WyfwBuildingManage bu : blist) {
            String tempId = String.valueOf(bu.getId());
            if (devMapList.containsKey(tempId)) {
                List<WyfwOwnerManage> tempList = devMapList.get(tempId);
                if (!Objects.isNull(tempList) && tempList.size() > 0) {
                    List<WyfwBuildingManage> addList = new ArrayList<>();
                    tempList.forEach(item -> {
                        WyfwBuildingManage buy = new WyfwBuildingManage();
                        buy.setId(String.valueOf(item.getId()));
                        buy.setParentId(String.valueOf(item.getBuildingId()));
                        buy.setBuildingName(item.getCompanyName());
                        buy.setLat(item.getLatitude());
                        buy.setLon(item.getLongitude());
                        addList.add(buy);
                    });
                    if (Objects.isNull(bu.getChildren())) {
                        bu.setChildren(addList);
                    } else {
                        bu.getChildren().addAll(addList);
                    }
                }
            }
        }

        return returnList.stream().map(ThreeWyfwData::new).collect(Collectors.toList());
    }


    /**
     * 递归列表
     */
    private void recursionFn(List<WyfwBuildingManage> list, WyfwBuildingManage t) {
        // 得到子节点列表
        List<WyfwBuildingManage> childList = getChildList(list, t);
        t.setChildren(childList);
        for (WyfwBuildingManage tChild : childList) {
            if (hasChild(list, tChild)) {
                recursionFn(list, tChild);
            }
        }
    }

    /**
     * 得到子节点列表
     */
    private List<WyfwBuildingManage> getChildList(List<WyfwBuildingManage> list, WyfwBuildingManage t) {
        List<WyfwBuildingManage> tlist = new ArrayList<WyfwBuildingManage>();
        for (WyfwBuildingManage n : list) {
            if (StringUtils.isNotNull(n.getParentId()) && n.getParentId().equals(t.getId())) {
                tlist.add(n);
            }
        }
        return tlist;
    }
    /**
     * 判断是否有子节点
     */
    private boolean hasChild(List<WyfwBuildingManage> list, WyfwBuildingManage t) {
        return getChildList(list, t).size() > 0;
    }

　　

  @GetMapping("/getThreeWyList")
    public AjaxResult getThreeWyList(@RequestParam("name") String name) {
        WyfwOwnerManage wy =new WyfwOwnerManage();
        wy.setBuildingName(name);
        List<WyfwOwnerManage> list = wyfwOwnerManageService.selectWyfwOwnerManageList(wy);
        List<WyfwBuildingManage> wyList = WyfwBuildingManageMapper.selectWyfwBuildingManageList(new WyfwBuildingManage());
        return AjaxResult.success(wyfwOwnerManageService.getThreeWyList(wyList,list));
    }

　　树结构中间对象：

public class ThreeWyfwData implements Serializable {




    /** 节点ID */
    private String id;
    private String pid;

    /** 节点名称 */
    private String label;

    private String latitude;

    /** 纬度 */
    private String longitude;
    /** 子节点 */
    @JsonInclude(JsonInclude.Include.NON_EMPTY)
    private List<ThreeWyfwData> children;

    public String getId() {
        return id;
    }

    public void setId(String id) {
        this.id = id;
    }

    public String getPid() {
        return pid;
    }

    public void setPid(String pid) {
        this.pid = pid;
    }

    public String getLabel() {
        return label;
    }

    public void setLabel(String label) {
        this.label = label;
    }

    public String getLatitude() {
        return latitude;
    }

    public void setLatitude(String latitude) {
        this.latitude = latitude;
    }

    public String getLongitude() {
        return longitude;
    }

    public void setLongitude(String longitude) {
        this.longitude = longitude;
    }

    public List<ThreeWyfwData> getChildren() {
        return children;
    }

    public void setChildren(List<ThreeWyfwData> children) {
        this.children = children;
    }
    public ThreeWyfwData() {}

    public ThreeWyfwData(WyfwBuildingManage wyfwBuildingManage)
    {
        this.id = wyfwBuildingManage.getId();
        this.pid = wyfwBuildingManage.getParentId();
        this.label = wyfwBuildingManage.getBuildingName();
        this.latitude = wyfwBuildingManage.getLat();
        this.longitude = wyfwBuildingManage.getLon();
       this.children = wyfwBuildingManage.getChildren().stream().map(ThreeWyfwData::new).collect(Collectors.toList());
    }
}