关于树形结构查询的接口设计-Toy模板网

这篇具有很好参考价值的文章主要介绍了关于树形结构查询的接口设计。希望对大家有所帮助。如果存在错误或未考虑完全的地方，请大家不吝赐教，您也可以点击"举报违法"按钮提交疑问。

在实际开发中，经常会遇到树形结构的查询，如：菜单树、公司组织机构树、地区区划树等等业务，这里写下两个接口设计方案以供日后使用。

1.表结构设计

//一般单表树形结构基本上都为以下结构：
id,name,parentId,其他业务指标
//多表树形同理,设计思路都是通过parentId关联在一起

2.树形返回类设计

@Data
public class TreeNodeNew implements Serializable {

    private String id;
	//名称
    private String name;
	//上级id
    private String parentId;
	//子节点
    private List<TreeNodeNew> childrenTreeNode;
    //其他业务字段...
}

3.树形方案一：递归获取全部父子节点(若数据量太大则不推荐用此方法)

	//传入需查询的根节点
    public List<TreeNodeNew> findTree(String node) {
    	if(StrUtil.isEmpty(node)){
            node = 当前登录人所绑定的node节点
        }
        //LambdaQueryWrapper<TreeEntity>wrapper = new LambdaQueryWrapper<>();
        //wrapper.likeRight(TreeEntity::getId,node);
        //查出全部数据,这里可以优化:想办法只查询出需要的所有数据,比如51-5101-510101这种格式的结构like51就行
        List<TreeEntity> treeList = list(wrapper);
        //转为返回树形集合对象
        List<TreeNodeNew> trees = treeList.stream().map(op -> {
            TreeNodeNew tree = new TreeNodeNew();
            tree.setId(op.getCode());
            tree.setParentId(op.getParentId());
            tree.setName(op.getName());
            return tree;
        }).collect(Collectors.toList());
        //用传入的根节点做第一级,调用getChildren设置所有下属节点
        List<TreeNodeNew> collect = trees .stream()
                .filter(item -> node.equals(item.getId()))
                .map(item -> {
                    item.setChildrenTreeNode(getChildren(item, trees));
                    return item;
                }).sorted(Comparator.comparing(TreeNodeNew::getId)).
                collect(Collectors.toList());

         return collect;
	}
	//传入父节点和所有数据节点信息集合,先查出所有父节点id为传入父节点id的下属数据,再递归set下属数据的childrens
    private static List<TreeNodeNew> getChildren(TreeNodeNew treeEntity, List<TreeNodeNew> treeEntityList) {
        List<TreeNodeNew> collect = treeEntityList.stream()
                .filter(item -> treeEntity.getId().equals(item.getParentId()))
                .map(item -> {
                    item.setChildrenTreeNode(getChildren(item, treeEntityList));
                    return item;
                }).sorted(Comparator.comparing(TreeNodeNew::getId))
                .collect(Collectors.toList());
        return collect;
    }

4.树形方案二：懒加载

//懒加载实现：每次传入一个根节点,只需要返回所有parentId为传入根节点的数据集合即可,前端每次拿到数据
//显示在根节点下面,点一次不同根节点请求一次,渲染一次
    public List<TreeEntity> findTree(String node) {
        LambdaQueryWrapper<TreeEntity> wrapper = new LambdaQueryWrapper<>();
        //当传入根节点为空时,为第一次进入,默认返回该根节点信息
        if(StrUtil.isEmpty(node)){
        	node = "";//根节点id
            wrapper.eq(TreeEntity::getId,node);
        }else{
        //否则返回所有父节点id为传入根节点id的下级信息
            wrapper.eq(TreeEntity::getParentId,node);
        }
        List<TreeEntity> list = list(wrapper);
        return list;
   	}