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
如果该代码对你有帮助,可请我喝咖啡