java 生成树表菜单

public class treeUtils {
    /**
     * 生成树表
     * @param arr 查询出来的数据的json集合
     * @param id id名字
     * @param pid 父id名字
     * @param child 下级目录的名字（此参数可不用修改）
     * @return
     */
   // JSONArray array = listToTree(ARRAY, "tableid", "tableparenid", "children");
    public static JSONArray listToTree(JSONArray arr, String id, String pid, String child){
        JSONArray r = new JSONArray();
        JSONObject hash = new JSONObject();
        //将数组转为Object的形式，key为数组中的id
        for(int i=0;i<arr.size();i++){
            JSONObject json = (JSONObject) arr.get(i);
            hash.put(json.getString(id), json);
        }
        //遍历结果集
        for(int j=0;j<arr.size();j++){
            //单条记录
            JSONObject aVal = (JSONObject) arr.get(j);
            //在hash中取出key为单条记录中pid的值
            JSONObject hashVP = (JSONObject) hash.get(aVal.get(pid).toString());
            //如果记录的pid存在，则说明它有父节点，将她添加到孩子节点的集合中
            if(hashVP!=null){
                //检查是否有child属性
                if(hashVP.get(child)!=null){
                    JSONArray ch = (JSONArray) hashVP.get(child);
                    ch.add(aVal);
                    hashVP.put(child, ch);
                }else{
                    JSONArray ch = new JSONArray();
                    aVal.put(child,null);
                    ch.add(aVal);
                    hashVP.put(child, ch);
                }
            }else{
                aVal.put(child,null);
                r.add(aVal);
            }
        }
        return r;
    }
}

技术交流群：758585246

如果该代码对你有帮助，可请我喝咖啡