java list转tree的工具类

今天帮前端做一个下拉框，关于行政区划的，表中是这种格式的，前端想要具有层次结构的tree类型，

这里需要处理下，这里可以看出districtCode 和parentCode具有父子关系。这样就好处理一些。伪代码如下：

 
import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.JSONArray;
import com.alibaba.fastjson.JSONObject;
 
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
 
/**
 * 通用的list转json tree的工具类
 */
public class CommonTreeUtil {
    /**
     - listToTree
     - <p>方法说明<p>
     - 将JSONArray数组转为树状结构
     - @param arr 需要转化的数据
     - @param id 数据唯一的标识键值
     - @param pid 父id唯一标识键值
     - @param child 子节点键值
     - @return JSONArray
     */
    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();
                    ch.add(aVal);
                    hashVP.put(child, ch);
                }
            }else{
                r.add(aVal);
            }
        }
        return r;
    }
 
 
    public static void main(String[] args) {
        List list = new ArrayList();
        Map map = new HashMap();
        map.put("id","t1");
        map.put("parentId","t0");
        map.put("name","tom");
        Map map1 = new HashMap();
        map1.put("id","t2");
        map1.put("parentId","t1");
        map1.put("name","cat");
        Map map2 = new HashMap();
        map2.put("id","t3");
        map2.put("parentId","t2");
        map2.put("name","jerry");
        Map map3 = new HashMap();
        map3.put("id","t4");
        map3.put("parentId","t0");
        map3.put("name","jeck");
 
        list.add(map);
        list.add(map1);
        list.add(map2);
        list.add(map3);
        JSONArray result = CommonTreeUtil.listToTree(JSONArray.parseArray(JSON.toJSONString(list)),"id","parentId","children");
        System.out.println(result);
 
    }
 
}

这个demo我测试了一下是可以跑通的，所以回到项目中把查询出来的数据list 转成JSONArray 再利用这个工具类做下递归处理就可以了。返回格式如下：

这里需要注意的是调用的时候传的参数："id","parentId","children"  ；id是唯一标识【看这个表中唯一标识，我的是districtCode】，parentId是父节点【parentCode】，children是自己命名的;如下：