Java 數據庫樹形查詢生成菜單結構

Java 數據庫樹形查詢

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='菜單表';

插入模擬數據

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;
}

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進行個格式化后的對比。