java后端把数据转换为树,map递归生成json树,返回给前端(后台转换)

//第一个参数，需要生成树的数组，第二个参数为树的根节点
public JSONObject getJsontree(JSONArray json,JSONObject job){
 JSONArray tempJson = JSONArray.fromObject("[]");
 //筛选出父id等于job里面id的科室
 for(int i = 0;i < json.size();i++) 
 { 
 //这里可以使用Iterator 
 if(json.getJSONObject(i).get("parent_id").equals(job.get("unit_sn"))) {  
  tempJson.add(json.getJSONObject(i));
 }; 
 }
 // 优化，减少科室集合的数量，避免重复查询，有再优化的方法，希望告知。。
 json.removeAll(tempJson);
 for(int i = 0;i < tempJson.size(); i ++) {
 //对第二层进行递归，此处类推
 getJsontree(json, tempJson.getJSONObject(i));
 }
 //生成完的树结构map集合加到根节点
 if(tempJson.size()!=0)
 job.put("children", tempJson);
 return job; 
}

[
 { text: '节点1', children: [
  { text: '节点1.1' },
  { text: '节点1.2' },
  { text: '节点1.3', children: [
  { text: '节点1.3.1' },
  { text: '节点1.3.2' }
  ]
  },
  { text: '节点1.4' }
 ]
 }
 ]

[
 {
 "name":"甘肃省",
 "pid":0,
 "id":1
 },
 {
 "name":"天水市",
 "pid":1,
 "id":2
 },
 {
 "name":"秦州区",
 "pid":2,
 "id":3
 },
 {
 "name":"北京市",
 "pid":0,
 "id":4
 },
 {
 "name":"昌平区",
 "pid":4,
 "id":5
 }
]

[
 {
 "children":[
  {
  "children":[
   {
   "name":"秦州区",
   "pid":2,
   "id":3
   }
  ],
  "name":"天水市",
  "pid":1,
  "id":2
  }
 ],
 "name":"甘肃省",
 "pid":0,
 "id":1
 },
 {
 "children":[
  {
  "name":"昌平区",
  "pid":4,
  "id":5
  }
 ],
 "name":"北京市",
 "pid":0,
 "id":4
 }
]

 import com.alibaba.fastjson.JSON;
 import com.alibaba.fastjson.JSONArray;
 import com.alibaba.fastjson.JSONObject;
 
 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的值
  String pidStr = "";
  Object pidObj = aVal.get(pid);
  if (aVal.get(pid) != null) {
  pidStr = aVal.get(pid).toString();
  }
  JSONObject hashVP = (JSONObject) hash.get(pidStr);
  //如果记录的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<Map<String,Object>> data = new ArrayList<>();
 Map<String,Object> map = new HashMap<>();
 map.put("id",1);
 map.put("pid",0);
 map.put("name","甘肃省");
 data.add(map);
 Map<String,Object> map2 = new HashMap<>();
 map2.put("id",2);
 map2.put("pid",1);
 map2.put("name","天水市");
 data.add(map2);
 Map<String,Object> map3 = new HashMap<>();
 map3.put("id",3);
 map3.put("pid",2);
 map3.put("name","秦州区");
 data.add(map3);
 Map<String,Object> map4 = new HashMap<>();
 map4.put("id",4);
 map4.put("pid",0);
 map4.put("name","北京市");
 data.add(map4);
 Map<String,Object> map5 = new HashMap<>();
 map5.put("id",5);
 map5.put("pid",4);
 map5.put("name","昌平区");
 data.add(map5);
 System.out.println(JSON.toJSONString(data));
 JSONArray result = 
 listToTree(JSONArray.parseArray(JSON.toJSONString(data)),"id","pid","children");
 System.out.println(JSON.toJSONString(result));
}