Java 之无限级分类

一、代码

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