java list转tree实现树状菜单

蔚落 2022-03-18 13:54 240阅读 0赞

有时候我们从数据库中读取出了一个表的数据，比如存储的是中国的省市县的ID、名称与父节点ID，读出来的数据并不是前台想要的（当然现在好多tree库可以使用，不要拆台，最起码不好看是吧==），这个时候我们要想法处理一下都出来的list，将它变为一个树。

比如直接查出来是左边的数据，我们要的效果是右边的数据  
![SouthEast][]![SouthEast 1][]

新建实体类：

public class SysDept {
    
        private Long id;
    
        private String name;
        
        private Long parentId;
    
        private Integer orderNum;
    
        private Byte delFlag;
        
        private List<SysDept> children;
        
        // 非数据库字段
        private String parentName;
        // 非数据库字段
        private Integer level;
        
    	.....Getter/Setter
    
    }

实现层：

public List<SysDept> findTree() {
    		List<SysDept> sysDepts = new ArrayList<>();
    		List<SysDept> depts = sysDeptMapper.findAll();
    		for (SysDept dept : depts) {
    			if (dept.getParentId() == null || dept.getParentId() == 0) {
    				dept.setLevel(0);
    				sysDepts.add(dept);
    			}
    		}
    		findChildren(sysDepts, depts);
    		return sysDepts;
    	}
    
    	private void findChildren(List<SysDept> sysDepts, List<SysDept> depts) {
    		for (SysDept sysDept : sysDepts) {
    			List<SysDept> children = new ArrayList<>();
    			for (SysDept dept : depts) {
    				if (sysDept.getId() != null && sysDept.getId().equals(dept.getParentId())) {
    					dept.setParentName(dept.getName());
    					dept.setLevel(sysDept.getLevel() + 1);
    					children.add(dept);
    				}
    			}
    			sysDept.setChildren(children);
    			findChildren(children, depts);
    		}
    	}

这边是用递归实现的，其他方法请参考：[点击这里][Link 1]

[SouthEast]: https://img-blog.csdn.net/20180204205818514?watermark/2/text/aHR0cDovL2Jsb2cuY3Nkbi5uZXQveGN5bW9ybmluZ3N1bg==/font/5a6L5L2T/fontsize/400/fill/I0JBQkFCMA==/dissolve/70/gravity/SouthEast
[SouthEast 1]: /images/20220318/b3ab90efd202410cb0a88ba4811894ee.png
[Link 1]: https://blog.csdn.net/xcymorningsun/article/details/79254819