Java树状图展示的实现

Java树状图展示的实现

在项目中经常会遇到将数据以树状图的形式展现出来,如图:

有时候不仅要构建树结构,还要计算每个节点拥有的成员数量.

由于我们不能确定树的大小和层级就不能用常规的方法进行展示,我的解决放入如下,直接上代码:

首先准备实体类:
我已近将构造实体类的方法写在实体类里面了,以后需要构造树形结构的类只要写上自己独有的字段,然后继承这个类,就能直接调用写好的构造树形结构的方法,相当于一个工具类.

/**
 * ClassName: BaseTreeVo <br/>
 * Description: <br/>
 * date: 2022/8/20 12:33<br/>
 * 构造树形结构的类都应该继承该类
 *
 * @author 无情的花裤衩<br />
 */
@Data
public class BaseTreeVo {

    @ApiModelProperty(value = "节点id")
    private Long id;
    @ApiModelProperty(value = "上级节点的id 上级节点为0时为根节点")
    private Long superId;
    @ApiModelProperty(value = "子节点")
    private List<BaseTreeVo> children;
    @ApiModelProperty(value = "拥有成员总数量")
    private Integer memberNum;
    @ApiModelProperty(value = "直属子成员的id集合")
    private Set<Long> directlyMemberIds;
	//这里使用set集合是因为可能会出现一个员工属于多个部门的情况,
	//防止一个员工被重复计算为多个员工导致数量不准确

    /**
     * 计算当前节点拥有成员的总数
     *
     * @param element 需要计算拥有成员总数量的节点
     */
    public static void countMemberNum(BaseTreeVo element) {
        Set<Long> memberIds = Sets.newHashSet();
        if (!CollectionUtils.isEmpty(element.getDirectlyMemberIds())) {
            memberIds.addAll(element.getDirectlyMemberIds());
        }
        memberIds.addAll(getChildrenMemberIds(element.getChildren()));
        //设置拥有的成员数量
        element.setMemberNum(memberIds.size());
    }

    /**
     * 获取子节点拥有的全部成员的id集合
     *
     * @param children 某个节点的全部的子节点
     * @return 全部子节点拥有成员的id集合
     */
    public static Set<Long> getChildrenMemberIds(List<? extends BaseTreeVo> children) {
        //全部子节点拥有成员的id集合
        Set<Long> result = Sets.newHashSet();
        for (BaseTreeVo child : children) {
            //没有子节点  这个节点是叶子节点
            if (CollectionUtils.isEmpty(child.getChildren())) {
                //设置拥有的成员数量
                if (CollectionUtils.isEmpty(child.getDirectlyMemberIds())) {
                    child.setMemberNum(0);
                } else {
                    child.setMemberNum(child.getDirectlyMemberIds().size());
                    result.addAll(child.getDirectlyMemberIds());
                }
            } else {
                //这个节点还有子节点 不是叶子节点
                Set<Long> childrenMemberIds = getChildrenMemberIds(child.getChildren());
                if (!CollectionUtils.isEmpty(child.getDirectlyMemberIds())) {
                    childrenMemberIds.addAll(child.getDirectlyMemberIds());
                }
                child.setMemberNum(childrenMemberIds.size());
                result.addAll(childrenMemberIds);
            }
        }
        return result;
    }

    /**
     * 获取根节点
     * 多个根节点时用这个方法获取根节点
     *
     * @param elements 全部的需要构建树形结构的元素
     */
    public static List<BaseTreeVo> getRootNodes(List<? extends BaseTreeVo> elements) {
        List<BaseTreeVo> roots = Lists.newLinkedList();
        Iterator<? extends BaseTreeVo> iterator = elements.iterator();
        while (iterator.hasNext()) {
            BaseTreeVo element = iterator.next();
            if (element.getSuperId() == 0) {
                roots.add(element);
                //将根节点移除 降低以后的循环元素
                iterator.remove();
            }
        }
        return roots;
    }

    /**
     * 获取根节点
     * 只有一个根节点时用这个方法
     *
     * @param elements 全部的需要构建树形结构的元素
     */
    public static BaseTreeVo getRootNode(List<? extends BaseTreeVo> elements) {
        BaseTreeVo result = null;
        for (BaseTreeVo element : elements) {
            if (element.getSuperId() == 0) {
                //将集合中的根节点移除,减少后续的循环元素的个数
                elements.remove(element);
                result = element;
                return result;
            }
        }
        return null;
    }

    /**
     * 递归，建立子树形结构
     *
     * @param rootNode 根节点
     * @param elements 全部的子节点
     */
    public static BaseTreeVo buildTree(BaseTreeVo rootNode, List<? extends BaseTreeVo> elements) {
        List<BaseTreeVo> children = Lists.newLinkedList();
        for (BaseTreeVo element : elements) {
            if (element.getSuperId().equals(rootNode.getId())) {
                children.add(buildTree(element, elements));
            }
        }
        rootNode.setChildren(children);
        return rootNode;
    }
}

使用代码的示例:
表结构:
父子关系是通过superId进行绑定的

需要构造树形结构的类:

/**
 * 树形展示租户部门列表vo类
 *
 * @author yanglixin2
 */
@Data
@EqualsAndHashCode(callSuper = true)
@ApiModel(description = "部门树形列表")
public class TenantDepartmentListVo extends BaseTreeVo {



    @ApiModelProperty(value = "部门名称")
    private String departmentName;
    @ApiModelProperty(value = "本级排序")
    private Integer sort;
}

/**
     * 树形展示部门列表
     *
     * @return 部门树形列表
     */
    @Override
    public TenantDepartmentListVo queryDepartmentList() {
        //1.查询出全部部门
        List<TenantDepartment> departmentList = queryDepartmentAllList();

        List<TenantDepartmentListVo> departmentVoList = Lists.newLinkedList();
        //将实体类对象转化为视图对象
        for (TenantDepartment tenantDepartment : departmentList) {
            TenantDepartmentListVo departmentListVo = BeanHelper.copyProperties(tenantDepartment, TenantDepartmentListVo.class);
            departmentVoList.add(departmentListVo);
        }

        //查询每个部门的直属人员的数量
        //首先部门的id集合
        List<Long> departmentIds = departmentVoList.stream().map(BaseTreeVo::getId).collect(Collectors.toList());
        //通过部门id查询出每个部门拥有的员工id
        List<TenantDepartmentAccountIdsVo> departmentAccountIds = this.baseMapper.queryDepartmentAccountIds(departmentIds);
        List<Long> departmentHasAccountIds = departmentAccountIds.stream().map(TenantDepartmentAccountIdsVo::getDepartmentId).collect(Collectors.toList());
        //给每个部门设置直属账号的id集合
        for (TenantDepartmentListVo tenantDepartmentListVo : departmentVoList) {
            if (!departmentHasAccountIds.contains(tenantDepartmentListVo.getId())) {
                //当前部门没有直属账号
                tenantDepartmentListVo.setDirectlyMemberIds(null);
            } else {
                for (TenantDepartmentAccountIdsVo departmentAccountId : departmentAccountIds) {
                    if (departmentAccountId.getDepartmentId().equals(tenantDepartmentListVo.getId())) {
                        tenantDepartmentListVo.setDirectlyMemberIds(departmentAccountId.getAccountIds());
                    }
                }
            }
        }

        //2.获取根部门
        BaseTreeVo rootNode = BaseTreeVo.getRootNode(departmentVoList);
        if (Objects.isNull(rootNode)) {
            throw new BusinessException(ErrorDesc.SERVER_ERROR.getDesc());
        }
        //3.构建树结构  并将类型转换回来
        TenantDepartmentListVo resultTree = (TenantDepartmentListVo) BaseTreeVo.buildTree(rootNode, departmentVoList);
        //4.设置账号总数
        BaseTreeVo.countMemberNum(resultTree);
        return resultTree;
    }