近期项目有个需求,需要将组织机构数据拼成树型结构返回至前端。我的做法如下: 1.由于前端展示使用的组件需要特定的属性key,所以我定义了一个VO对象,代码如下: (本人使用了Lombok插件)
package top.yuxuange.vo;
import lombok.Data;
import java.util.List;
@Data
public class TreeVO {
/**
* id
*/
private String id;
/**
* title
*/
private String title;
/**
* 子级
*/
private List<TreeVO> children;
}
2.编写树型结构工具类:
这是我数据库中组织机构的根节点CODE的截图
package top.yuxuange.util;
import org.springframework.stereotype.Component;
import top.yuxuange.model.SysDept;
import top.yuxuange.vo.TreeVO;
import java.util.LinkedList;
import java.util.List;
import java.util.stream.Collectors;
/**
* 获取组织机构树
*
* @author yuxuange
* @Date 2021-02-03
*/
@Component
public class TreeUtil {
/**
* 组织机构表,根节点CODE
*/
private static final String TREE_ROOT_CODE = "0";
/**
* 标准版
*/
public static List<TreeVO> getTree(List<SysDept> record) {
List<TreeVO> treeList = new LinkedList();
for (SysDept sysDept : record) {
if (TREE_ROOT_CODE.equals(sysDept.getParentId())) {
TreeVO treeVO = new TreeVO();
treeVO.setId(sysDept.getCode());
treeVO.setTitle(sysDept.getName());
treeVO.setChildren(getChild(sysDept.getCode(), record));
treeList.add(treeVO);
}
}
return treeList;
}
private static List<TreeVO> getChild(String code, List<SysDept> record) {
List<TreeVO> childrenList = new LinkedList();
for (SysDept sysDept : record) {
if (code.equals(sysDept.getParentId())) {
TreeVO treeVO = new TreeVO();
treeVO.setId(sysDept.getCode());
treeVO.setTitle(sysDept.getName());
treeVO.setChildren(getChild(sysDept.getCode(), record));
childrenList.add(treeVO);
}
}
return childrenList;
}
/**
* lamda表达式版
*/
public static List<TreeVO> getTreeEX(List<SysDept> record){
return record.stream().filter(sysDept -> {return TREE_ROOT_CODE.equals(sysDept.getParentId());}).map(sysDept -> {
TreeVO treeVO = new TreeVO();
treeVO.setId(sysDept.getCode());
treeVO.setTitle(sysDept.getName());
treeVO.setChildren(getChildEX(sysDept.getCode(),record));
return treeVO;
}).collect(Collectors.toList());
}
private static List<TreeVO> getChildEX(String code,List<SysDept> record){
return record.stream().filter(sysDept -> {return code.equals(sysDept.getParentId());}).map(sysDept -> {
TreeVO treeVO = new TreeVO();
treeVO.setId(sysDept.getCode());
treeVO.setTitle(sysDept.getName());
treeVO.setChildren(getChildEX(sysDept.getCode(),record));
return treeVO;
}).collect(Collectors.toList());
}
}
3.查询数据,调用工具类,返回结果如下:
版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 举报,一经查实,本站将立刻删除。