一、程式碼

package utils;

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

import entity.Auth;

public class MenuUtil {
    private static List<Auth> tree = new ArrayList<Auth>();
    /**
     * 獲取導航選單
     * @param data
     * @return
     */
    public static List<Auth> getMenu(List<Auth> data) {
        
 
// 獲取頂級選單
        ArrayList<Auth> rootNode = new ArrayList<Auth>();
        for (Auth auth : data) {
            if (auth.getParent_id() == 0) {
                rootNode.add(auth);
            }
        }
        //System.out.println("rootNode = "+rootNode);
        // 獲取子級選單
        for (Auth auth : rootNode) {
            List
 
<Auth> child = getChild(auth.getId(), data);
            auth.setChildren(child);
        }
        return rootNode;
    }

    private static List<Auth> getChild(int id, List<Auth> data) {
        //存放子選單的集合
        ArrayList<Auth> childNode = new ArrayList<Auth>();
        
 
for (Auth auth : data) {
            System.out.println(id == auth.getParent_id());
            if(id == auth.getParent_id()) {
                childNode.add(auth);
            }
        }
        //遞迴子級
        for (Auth auth : childNode) {
            auth.setChildren(getChild(auth.getId(), data));
        }
        if(childNode.size() == 0) {
            return null;
        }
        return childNode;
    }
    
    /**
     * 獲取無限極分類
     * @param data
     * @param parent_id
     * @param level
     * @return
     */
    public static List<Auth> getAuthList(List<Auth> data, int parent_id, int level) {
        for(Auth auth : data) {
            if(auth.getParent_id() == parent_id) {
                auth.setLevel(level);
                tree.add(auth);
                getAuthList(data, auth.getId(), level + 1);
            }
        }
        return tree;
    }
}