递归实现目录树，菜单树

页面实现效果：Tree 树形控件

1. 定义菜单类

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方法
}

2. 定义数据库，并插入菜单数据

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');

3. 编写测试方法与递归方法

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

参考博客： https://www.cnblogs.com/zhifengge/p/6910881.html

4. Python代码实现：

# 获取机构树
def getOrgList(page_num, page_size):

    result_list = []
	# 获取机构列表
    org_data = selectOrgList()
    if len(org_data) > 0:
        result_list = listToTree(org_data)  # 列表转为树形结构

    retdata = {
        'total':len(result_list),

        'data': result_list,
        'code': 20000
    }

    return Response(json.dumps(retdata), mimetype="application/json")

def listToTree(org_data):
    result_data = []
    # 先找到所有的一级机构
    for org_obj in org_data:
        # 根的parent_id为SYSTEM
        if org_obj['parent_id'] == 'SYSTEM':
            result_data.append(org_obj)

    # 为一级菜单设置子机构，getChild是递归调用的
    for org_obj in result_data:
        org_obj['children'] = getChild(org_obj['id'], org_data)

    return result_data

# 递归查找子机构
def getChild(id, org_data):

    # 子机构
    child_list = []
    for org_object in org_data:
        # 遍历所有节点，将父机构org_level_code与传过来的org_level_code比较
        if org_object['parent_id'] != '' and org_object['parent_id'] is not None:
            if org_object['parent_id'] == id:
                child_list.append(org_object)

    # 把子机构的子机构再循环一遍
    for org_obj in child_list:
        # 暂时没有标识哪个机构是有子机构的
        org_obj['children'] = getChild(org_obj['id'], org_data)

    # 递归退出条件
    if len(org_data) == 0:
        return None

    return child_list

    pass