java组装json_java 组装树形结构

package cn.thinkjoy.kidscare.utils;

import com.alibaba.fastjson.JSON;

import com.alibaba.fastjson.JSONArray;

import com.alibaba.fastjson.JSONObject;

import com.alibaba.fastjson.serializer.SerializeFilter;

import com.alibaba.fastjson.serializer.SimplePropertyPreFilter;

import org.apache.commons.lang3.ArrayUtils;

import org.springframework.util.CollectionUtils;

import java.util.*;

/***

*

* @ClassName: TreeUtils

* @Description: 组织树结构

* @author liyuhang

*

*/

public class TreeUtils {

/****

*

* @Title: getTreeByStack

* @Description: 获取树形结构字符串

* @param jsonRootList 所有根结点

* @return String 树形字符串

*/

public static String getTreeByStack(JSONArray jsonRootList, Map parentMap){

Stack stack=new Stack();

for(int i=0;i

JSONObject root=jsonRootList.getJSONObject(i);

stack.push(root);

}

while (!stack.isEmpty()) {

JSONObject parentNode=stack.pop();

if (!CollectionUtils.isEmpty(parentMap.get(parentNode.getLong("id")))) {

JSONArray list = parentMap.get(parentNode.getLong("id"));

parentNode.put("values", list);

for(int k=0;k

stack.push(list.getJSONObject(k));

}

}else{

parentNode.put("values", new JSONArray());

}

}

String tree=jsonRootList.toString();

return tree;

}

/****

*

* @Title: getTreeByStack

* @Description: 获取树形结构字符串

* @param nodeAll 所有结点

* @param selectedIds 选中ids

* @param filters

* @return String 树形字符串

*/

public static String getTreeByStack(List> nodeAll, List selectedIds, SerializeFilter[] filters){

if (ArrayUtils.isEmpty(filters)) {

SimplePropertyPreFilter simplePropertyPreFilter = new SimplePropertyPreFilter();

filters = new SerializeFilter[]{simplePropertyPreFilter};

}

Map parentMap = new HashMap<>();

JSONArray rootList = new JSONArray();

if (!CollectionUtils.isEmpty(nodeAll)) {

JSONArray jsonListAll = JSONArray.parseArray(JSON.toJSONString(nodeAll,filters));

for(int i=0;i

JSONObject jsonObject = jsonListAll.getJSONObject(i);

if (!CollectionUtils.isEmpty(selectedIds)) {

if (selectedIds.contains(jsonObject.getLong("id"))) {

jsonObject.put("flag", 1);

}else {

jsonObject.put("flag", 0);

}

}

if (jsonObject.get("parentId")==null||jsonObject.getLongValue("parentId")==0) {

rootList.add(jsonObject);

continue;

}

Long parentId = jsonObject.getLong("parentId");

if (CollectionUtils.isEmpty(parentMap.get(parentId))) {

JSONArray list = new JSONArray();

parentMap.put(parentId, list);

}

parentMap.get(parentId).add(jsonObject);

}

}

String tree=getTreeByStack( rootList,parentMap);

return tree;

}

}