java從數據庫讀取菜單，遞歸生成菜單樹

首先看一下菜單的樣子

根據這個樣子我們定義菜單類

public class Menu { // 菜單id private String id; // 菜單名稱 private String name; // 父菜單id private String parentId; // 菜單url private String url; // 菜單圖標 private String icon; // 菜單順序 private int order; // 子菜單 private List<Menu> childMenus; // ... 省去getter和setter方法以及toString方法 }

 

我們根據這個類定義數據庫，並插入菜單數據

DROP TABLE IF EXISTS `jrbac_menu`; CREATE TABLE `jrbac_menu` ( `id` varchar(32) NOT NULL COMMENT '主鍵id,uuid32位', `name` varchar(64) NOT NULL COMMENT '登錄用戶名', `parent_id` varchar(32) DEFAULT NULL COMMENT '父菜單id', `url` varchar(64) DEFAULT NULL COMMENT '訪問地址', `icon` varchar(32) DEFAULT NULL COMMENT '菜單圖標', `order` tinyint(4) DEFAULT '0' COMMENT '菜單順序', PRIMARY KEY (`id`) ) ENGINE=InnoDB DEFAULT CHARSET=utf8 COMMENT='菜單表'; -- ---------------------------- -- Records of jrbac_menu -- ---------------------------- INSERT INTO `jrbac_menu` VALUES ('1', 'Forms', null, 'forms.html', 'fa fa-edit', '0'); INSERT INTO `jrbac_menu` VALUES ('2', 'UI Elements', null, '', 'fa fa-wrench', '1'); INSERT INTO `jrbac_menu` VALUES ('3', 'Buttons', '2', 'buttons.html', '', '0'); INSERT INTO `jrbac_menu` VALUES ('4', 'Icons', '2', 'icons.html', null, '1'); INSERT INTO `jrbac_menu` VALUES ('5', 'Multi-Level Dropdown', '', '', 'fa fa-sitemap', '2'); INSERT INTO `jrbac_menu` VALUES ('6', 'Second Level Item', '5', 'second.html', null, '0'); INSERT INTO `jrbac_menu` VALUES ('7', 'Third Level', '5', null, '', '1'); INSERT INTO `jrbac_menu` VALUES ('8', 'Third Level Item', '7', 'third.html', null, '0');

 

 

為了演示，我們把可展開的沒有做完，僅僅插入幾條數據能出效果就可以了。

測試方法與遞歸方法

private final Gson gson = new GsonBuilder().disableHtmlEscaping().create(); @Test public void testQueryMenuList() { // 原始的數據 List<Menu> rootMenu = menuDao.queryMenuList(null); // 查看結果 for (Menu menu : rootMenu) { System.out.println(menu); } // 最后的結果 List<Menu> menuList = new ArrayList<Menu>(); // 先找到所有的一級菜單 for (int i = 0; i < rootMenu.size(); i++) { // 一級菜單沒有parentId if (StringUtils.isBlank(rootMenu.get(i).getParentId())) { menuList.add(rootMenu.get(i)); } } // 為一級菜單設置子菜單，getChild是遞歸調用的 for (Menu menu : menuList) { menu.setChildMenus(getChild(menu.getId(), rootMenu)); } Map<String,Object> jsonMap = new HashMap<>(); jsonMap.put("menu", menuList); System.out.println(gson.toJson(jsonMap)); } /** * 遞歸查找子菜單 * * @param id * 當前菜單id * @param rootMenu * 要查找的列表 * @return */ private List<Menu> getChild(String id, List<Menu> rootMenu) { // 子菜單 List<Menu> childList = new ArrayList<>(); for (Menu menu : rootMenu) { // 遍歷所有節點，將父菜單id與傳過來的id比較 if (StringUtils.isNotBlank(menu.getParentId())) { if (menu.getParentId().equals(id)) { childList.add(menu); } } } // 把子菜單的子菜單再循環一遍 for (Menu menu : childList) {// 沒有url子菜單還有子菜單 if (StringUtils.isBlank(menu.getUrl())) { // 遞歸 menu.setChildMenus(getChild(menu.getId(), rootMenu)); } } // 遞歸退出條件 if (childList.size() == 0) { return null; } return childList; }

 

 

menuDao.queryMenuList(null);查找的結果是一條一條的數據

meuDao

package com.jrbac.dao; import java.util.List; import com.jrbac.entity.LoginUser; import com.jrbac.entity.Menu; public interface MenuDao { /** * 查找用戶的菜單 * @param loginUser * @return */ public List<Menu> queryMenuList(LoginUser loginUser); } 

 

mybatis

<?xml version="1.0" encoding="UTF-8"?> <!DOCTYPE mapper PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN" "http://mybatis.org/dtd/mybatis-3-mapper.dtd"> <mapper namespace="com.jrbac.dao.MenuDao"> <select id="queryMenuList" resultType="Menu"> SELECT id,`name`,parent_id,url,icon,`order` FROM jrbac_menu ORDER BY `order` ASC </select> </mapper>

 

測試程序的運行結果，對輸出的json進行個格式化后的對比