JAVA查詢樹結構數據(省市區)使用hutool工具實現
代碼:
@PostMapping("/getTree")
public Object getTree() {
// 查詢數據
List<LxhpArea> areaList = areaService.list();
// 構建的整個樹數據
List<TreeNode<String>> treeNodeList = areaList.stream().map(area -> {
// 擴展字段賦值,下面要取這里的值
Map<String, Object> extraMap = new HashMap<>();
extraMap.put("domain", area.getDomain());
extraMap.put("isOpen", area.getIsOpen());
extraMap.put("bigArea", area.getBigArea());
// 單個樹數據構建
TreeNode<String> treeNode = new TreeNode<String>()
.setId(String.valueOf(area.getId())) // 主鍵
.setParentId(String.valueOf(area.getReid())) // 父節點ID
.setName(area.getName()) // 省份名稱
.setWeight(area.getSortOrder()) //權重,排序
.setExtra(extraMap); // 擴展字段
return treeNode;
}).collect(Collectors.toList());
// 配置
TreeNodeConfig treeNodeConfig = new TreeNodeConfig();
// 自定義屬性名(修改默認名稱)
treeNodeConfig.setWeightKey("sortOrder");
treeNodeConfig.setChildrenKey("childrenNode");
// 最大遞歸深度
treeNodeConfig.setDeep(1);
//轉換器
List<Tree<String>> treeNodes = TreeUtil.build(treeNodeList, "1", treeNodeConfig,
(treeNode, tree) -> {
// 給樹節點賦值(還能set 父 或子節點樹)
tree.setId(treeNode.getId());
tree.setParentId(treeNode.getParentId());
tree.setWeight(treeNode.getWeight());
tree.setName(treeNode.getName());
// 擴展屬性值賦值
// treeNode.getExtra().getOrDefault("domain", null) 是獲取上面Map放進去的值,沒有就是 null
tree.putExtra("domain", treeNode.getExtra().getOrDefault("domain", null));
tree.putExtra("isOpen", treeNode.getExtra().getOrDefault("isOpen", null));
tree.putExtra("bigArea", treeNode.getExtra().getOrDefault("bigArea", null));
});
return JSONUtil.parse(treeNodes);
}
表結構:
CREATE TABLE `lxhp_area` ( `id` smallint(5) unsigned NOT NULL AUTO_INCREMENT, `reid` smallint(5) unsigned NOT NULL DEFAULT '0', `name` varchar(120) NOT NULL DEFAULT '', `sort_order` smallint(5) unsigned NOT NULL DEFAULT '0', `is_open` tinyint(3) unsigned NOT NULL DEFAULT '0', `domain` varchar(10) NOT NULL, `big_area` int(4) DEFAULT NULL, PRIMARY KEY (`id`) ) ENGINE=MyISAM AUTO_INCREMENT=3433 DEFAULT CHARSET=utf8;
數據結構:

官方文檔:https://hutool.cn/docs/#/core/%E8%AF%AD%E8%A8%80%E7%89%B9%E6%80%A7/%E6%A0%91%E7%BB%93%E6%9E%84/%E6%A0%91%E7%BB%93%E6%9E%84%E5%B7%A5%E5%85%B7-TreeUtil
