java遞歸 處理權限管理菜單樹或分類

1.數據庫表設計

2.實體類設計

package com.ieou.capsule.dto.SystemPermissions;

import java.util.List;

/**
 * 功能菜單類
 */
public class SystemPermissionsTree {

    private String functionCode;//菜單碼

    private String parentFunctionCode;//父級菜單碼

    private String functionName;//菜單名

    private Boolean flag; // true:選中   false:未選中

    private List<SystemPermissionsTree> childrenList;

    public String getFunctionCode() {
        return functionCode;
    }

    public void setFunctionCode(String functionCode) {
        this.functionCode = functionCode;
    }

    public String getParentFunctionCode() {
        return parentFunctionCode;
    }

    public void setParentFunctionCode(String parentFunctionCode) {
        this.parentFunctionCode = parentFunctionCode;
    }

    public String getFunctionName() {
        return functionName;
    }

    public void setFunctionName(String functionName) {
        this.functionName = functionName;
    }

    public Boolean getFlag() {
        return flag;
    }

    public void setFlag(Boolean flag) {
        this.flag = flag;
    }

    public List<SystemPermissionsTree> getChildrenList() {
        return childrenList;
    }

    public void setChildrenList(List<SystemPermissionsTree> childrenList) {
        this.childrenList = childrenList;
    }
}

3.遞歸工具類

package com.ieou.capsule.util;

import com.ieou.capsule.dto.SystemPermissions.SystemPermissionsTree;

import java.util.ArrayList;
import java.util.List;

public class TreeUtil {
    /**
     * 作者：一沐楓一
     * 來源：CSDN
     * 原文：https://blog.csdn.net/gxgl8811/article/details/72803833
     * 版權聲明：本文為博主原創文章，轉載請附上博文鏈接！
     */

    public static List<SystemPermissionsTree> getTreeList(List<SystemPermissionsTree> entityList) {
        List<SystemPermissionsTree> resultList = new ArrayList<>();

        //獲取頂層元素集合
        String parentCode;
        for (SystemPermissionsTree entity : entityList) {
            parentCode = entity.getParentFunctionCode();
            //頂層元素的parentCode==null或者為0
            if (parentCode == null || "0".equals(parentCode)) {
                resultList.add(entity);
            }
        }

        //獲取每個頂層元素的子數據集合
        for (SystemPermissionsTree entity : resultList) {
            entity.setChildrenList(getSubList(entity.getFunctionCode(), entityList));
        }

        return resultList;
    }

    /**
     * 獲取子數據集合
     *
     * @param id
     * @param entityList
     * @return
     * @author jianda
     * @date 2017年5月29日
     */
    private static List<SystemPermissionsTree> getSubList(String id, List<SystemPermissionsTree> entityList) {
        List<SystemPermissionsTree> childList = new ArrayList<>();
        String parentId;

        //子集的直接子對象
        for (SystemPermissionsTree entity : entityList) {
            parentId = entity.getParentFunctionCode();
            if (id.equals(parentId)) {
                childList.add(entity);
            }
        }

        //子集的間接子對象
        for (SystemPermissionsTree entity : childList) {
            entity.setChildrenList(getSubList(entity.getFunctionCode(), entityList));
        }

        //遞歸退出條件
        if (childList.size() == 0) {
            return null;
        }

        return childList;
    }


}