Java处理子父级菜单的方式二
处理存在子父级关系的数据是写代码的过程中常见的操作，前面讲解过使用递归的方法来做，
可以参考这篇博客 https://www.cnblogs.com/yilangcode/p/16831867.html
今天来聊聊一种新的处理方式。使用List集合多轮遍历，添加子父级菜单信息。
建表SQL
DROP TABLE IF EXISTS `sa_menu`;
CREATE TABLE `sa_menu` (
`id` int(11) NOT NULL AUTO_INCREMENT COMMENT '主键',
`menu_code` varchar(16) DEFAULT NULL COMMENT '菜单编号',
`menu_parent_code` varchar(32) DEFAULT NULL COMMENT '菜单上级编号',
`menu_name` varchar(100) DEFAULT NULL COMMENT '菜单名称',
`menu_marks` varchar(100) DEFAULT NULL COMMENT '备注',
`menu_url` varchar(100) DEFAULT NULL COMMENT '菜单路径',
PRIMARY KEY (`id`)
) ENGINE=InnoDB AUTO_INCREMENT=8 DEFAULT CHARSET=utf8;
数据
INSERT INTO `sa_menu` VALUES ('1', '1', '0', '一级菜单一', '一级菜单一', 'one_level_menu');
INSERT INTO `sa_menu` VALUES ('2', '2', '1', '子菜-1-1', '子菜-1-1', 'child_menu_one');
INSERT INTO `sa_menu` VALUES ('3', '3', '1', '子菜-1-2', '子菜-1-2', 'child_menu_two');
INSERT INTO `sa_menu` VALUES ('4', '4', '0', '一级菜单二', '一级菜单二', 'two_level_menu');
INSERT INTO `sa_menu` VALUES ('5', '5', '4', '子菜单-2-1', '子菜单-2-1', 'child_menu_three');
INSERT INTO `sa_menu` VALUES ('6', '6', '4', '子菜单-2-2', '子菜单-2-2', 'child_menu_four');
INSERT INTO `sa_menu` VALUES ('7', '7', '4', '子菜单-2-3', '子菜单-2-3', 'child_menu_five');
数据如下
它们的层级关系如下所示
问题是如何构建这样带层级关系的子父级结构的数据呢？经过在实际项目中的
检验，测试，可以使用一种比较简单的方式来处理，下面逐一说明。
构建的菜单类如下
@Data
public class SaMenuPo {
/*
* 主键ID
* */
private int id;
/*
* 菜单 code
* */
private String menuCode;
/*
* 菜单父 code
* */
private String menuParentCode;
/*
* 菜单名称
* */
private String menuName;
/*
* 备注
* */
private String menuRemake;
/*
* 菜单 url
* */
private String menuUrl;
/*
* 子菜单
* */
private List<SaMenuPo> childMenu;
}
大致的实现思路如下:
查询数据时，将菜单数据全部取出，按照升序排序，便于处理的菜单数据是按照顺序排列。
遍历集合的时候，每次添加一条菜单数据，则删除该条菜单数据，可以提高效率，到最后的时候
传入的菜单都会被删除。删除后的菜单信息保存到一个新的菜单列表当中，在这个新的菜单列表中
来构建具有子父级关系的菜单信息。需要使用到集合遍历时的迭代器，来进行删除最简单。或者查询时
降序排序，添加时倒着遍历集合，然后添加元素后直接删除即可。
查询方式如下
<select id="menuTest" resultType="mybatis.po.SaMenuPo">
SELECT
t.id,
t.menu_code menuCode,
t.menu_parent_code menuParentCode,
t.menu_name menuName,
t.menu_marks menuMarks,
t.menu_url menuUrl
FROM sa_menu t
order by t.menu_code asc
</select>
主要的处理代码如下
private List<SaMenuPo> handlerMenu(List<SaMenuPo> menuPoList){
List<SaMenuPo> resultList = new ArrayList<>();
// 首先获取一级菜单，添加一个菜单之后，就删除一个菜单
Iterator<SaMenuPo> iterator = menuPoList.iterator();
SaMenuPo tempMenu;
while (iterator.hasNext()){
tempMenu = iterator.next();
// 可以确定的一点是，一级菜单的 menuParentCode 为0
if("0".equals(tempMenu.getMenuParentCode())){
// 添加一级菜单
resultList.add(tempMenu);
// 删除菜单
iterator.remove();
}
}
// 添加二级菜单
for(int i = 0; i < resultList.size(); i++){
List<SaMenuPo> childMenu = new ArrayList<>();
// 需要重新获取迭代器，因为添加完一次菜单后，原菜单列表menuPoList中总数有变化
iterator = menuPoList.iterator();
while (iterator.hasNext()){
tempMenu = iterator.next();
// 一级菜单的menuCode为二级菜单的 menuParentCode
if(resultList.get(i).getMenuCode().equals(tempMenu.getMenuParentCode())){
// 添加二级菜单
childMenu.add(tempMenu);
// 删除二级菜单
iterator.remove();
}
}
// 添加完一个一级菜单的所有子菜单，则设置子菜单
resultList.get(i).setChildMenu(childMenu);
}
return resultList;
}
返回的JSON数据格式如下
{"responseDate":"Sat Jul 15 11:18:48 CST 2023","status":"SUCCESS","errorCode":"","errorMessage":"","responseBody":[{"id":1,"menuCode":"1","menuParentCode":"0","menuName":"一级菜单一","menuMarks":"一级菜单一","menuUrl":"one_level_menu","childMenu":[{"id":2,"menuCode":"2","menuParentCode":"1","menuName":"子菜单-1-1","menuMarks":"子菜单-1-1","menuUrl":"child_menu_one","childMenu":null},{"id":3,"menuCode":"3","menuParentCode":"1","menuName":"子菜单-1-2","menuMarks":"子菜单-1-2","menuUrl":"child_menu_two","childMenu":null}]},{"id":4,"menuCode":"4","menuParentCode":"0","menuName":"一级菜单二","menuMarks":"一级菜单二","menuUrl":"two_level_menu","childMenu":[{"id":5,"menuCode":"5","menuParentCode":"4","menuName":"子菜单-2-1","menuMarks":"子菜单-2-1","menuUrl":"child_menu_three","childMenu":null},{"id":6,"menuCode":"6","menuParentCode":"4","menuName":"子菜单-2-2","menuMarks":"子菜单-2-2","menuUrl":"child_menu_four","childMenu":null},{"id":7,"menuCode":"7","menuParentCode":"4","menuName":"子菜单-2-3","menuMarks":"子菜单-2-3","menuUrl":"child_menu_five","childMenu":null}]}]}
然后去这个网站 https://jsoncrack.com/editor 将返回的数据放在左侧部分，整个JSON数据的结构就会
很清晰的显示出来。页面中拿到数据后再去渲染菜单就会方便很多。
这里只有两层菜单，处理起来比较方便，只需要处理两次即可。如果有三层呢？那就需要在处理一次，先遍历一级菜单，
然后遍历二级菜单，添加二级菜单的子菜单数据。这时候就可以考虑使用递归来做，代码会更加地简洁，使用更加地方便。
上面的菜单使用递归方式来处理如下，需要拆分为如下的两个方法来处理，这样就可以处理三级，四级菜单；或者是处理子父级
关系的数据比如省-市-区-县数据；或者是其他有子父级关系的数据；或是具有层级关系的数据，比如文件夹、文件数据。
private List<SaMenuPo> handlerOneLevelMenu(List<SaMenuPo> menuPoList){
List<SaMenuPo> resultMenu = new ArrayList<>();
for(SaMenuPo saMenuPo : menuPoList){
if("0".equals(saMenuPo.getMenuParentCode())){
// 添加一级菜单
resultMenu.add(saMenuPo);
}
}
for(SaMenuPo saMenuPo : resultMenu){
recursionHandlerMenu(menuPoList, saMenuPo);
}
return resultMenu;
}
private SaMenuPo recursionHandlerMenu(List<SaMenuPo> menuPoList, SaMenuPo parentSaMenuPo){
List<SaMenuPo> childMenu = new ArrayList<>();
for(SaMenuPo saMenuPo : menuPoList){
if(parentSaMenuPo.getMenuCode().equals(saMenuPo.getMenuParentCode())){
// 有子菜单数据，则递归调用
recursionHandlerMenu(menuPoList, saMenuPo);
// 添加子菜单数据
childMenu.add(saMenuPo);
}
}
if(childMenu.size() < 1){
// 递归调用时，没有子菜单数据，则直接返回
return parentSaMenuPo;
}
// 有子菜单数据，则添加子菜单数据
parentSaMenuPo.setChildMenu(childMenu);
return parentSaMenuPo;
}
为了演示方便，这时候添加一条三级菜单数据，如下
INSERT INTO `sa_menu` (`id`, `menu_code`, `menu_parent_code`, `menu_name`, `menu_marks`, `menu_url`) VALUES ('8', '8', '2', '子菜单-3-1', '子菜单-3-1', 'child_menu_six');
然后进行测试，返回数据如下
{"responseDate":"Sat Jul 15 11:55:49 CST 2023","status":"SUCCESS","errorCode":"","errorMessage":"","responseBody":[{"id":1,"menuCode":"1","menuParentCode":"0","menuName":"一级菜单一","menuMarks":"一级菜单一","menuUrl":"one_level_menu","childMenu":[{"id":2,"menuCode":"2","menuParentCode":"1","menuName":"子菜单-1-1","menuMarks":"子菜单-1-1","menuUrl":"child_menu_one","childMenu":[{"id":8,"menuCode":"8","menuParentCode":"2","menuName":"子菜单-3-1","menuMarks":"子菜单-3-1","menuUrl":"child_menu_six","childMenu":null}]},{"id":3,"menuCode":"3","menuParentCode":"1","menuName":"子菜单-1-2","menuMarks":"子菜单-1-2","menuUrl":"child_menu_two","childMenu":null}]},{"id":4,"menuCode":"4","menuParentCode":"0","menuName":"一级菜单二","menuMarks":"一级菜单二","menuUrl":"two_level_menu","childMenu":[{"id":5,"menuCode":"5","menuParentCode":"4","menuName":"子菜单-2-1","menuMarks":"子菜单-2-1","menuUrl":"child_menu_three","childMenu":null},{"id":6,"menuCode":"6","menuParentCode":"4","menuName":"子菜单-2-2","menuMarks":"子菜单-2-2","menuUrl":"child_menu_four","childMenu":null},{"id":7,"menuCode":"7","menuParentCode":"4","menuName":"子菜单-2-3","menuMarks":"子菜单-2-3","menuUrl":"child_menu_five","childMenu":null}]}]}
最终展示的JSON数据结构如下
这篇博客就写到这里，如果有其他更好建议的小伙伴，欢迎留言讨论。
