com.tmsps.ne4spring.utils.TreeUtil Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of ne4Spring Show documentation
NE For Spring is good :)
There is a newer version: 999.0.0.0
package com.tmsps.ne4spring.utils;

import java.util.ArrayList;
import java.util.List;
import java.util.Map;

public class TreeUtil {
	// 预处理树节点
	public static List> handleTree(List> menuList) {
		return handleTree(menuList, false);
	}

	// 预处理树节点
	public static List> handleTree(List> menuList, boolean isChecked) {
		for (Map map : menuList) {
			map.put("id", map.get("code"));
			map.put("iconCls", map.get("url"));
			map.put("text", map.remove("name"));
			map.put("leaf", true);
			if (isChecked) {
				map.put("checked", false);
			}
		}

		return menuList;
	}

	/**
	 * 
	 *======================================================
	 * @author zhangwei [email protected] 
	 *------------------------------------------------------
	 *======================================================
	 */
	public static String turnListToTree(List> menuList) {
		// TODO 转换List为树形结构
		return turnListToTree(menuList, false);
	}
	
	/**
	 * 
	 *======================================================
	 * @author zhangwei [email protected] 
	 *------------------------------------------------------
	 *======================================================
	 */
	@SuppressWarnings("unchecked")
	public static String turnListToTree(List> menuList, boolean isChecked) {
		// TODO 转换List为树形结构
		menuList = handleTree(menuList, isChecked);

		List> nodeList = new ArrayList>();
		for (Map node1 : menuList) {
			String node1_code = (String) node1.get("code");
			String node1_parent_code = node1_code.substring(0, node1_code.length() - 3);

			boolean mark = false;
			for (Map node2 : menuList) {
				String node2_code = (String) node2.get("code");

				if (node1_parent_code != null && node1_parent_code.equals(node2_code)) {
					mark = true;
					if (node2.get("children") == null) {
						node2.put("children", new ArrayList>());
					}
					((List>) node2.get("children")).add(node1);
					node2.put("leaf", false);
					if (!isChecked) {
						node2.put("expanded", false);
					}
					break;
				}
			}
			if (!mark) {
				nodeList.add(node1);
			}
		}
		return JsonUtil.toJson(nodeList);
	}

	// 预处理树节点
	public static List> handleAreaTree(List> menuList, boolean isChecked,
			int deep) {
		for (Map map : menuList) {
			map.put("id", map.remove("code"));
			map.put("iconCls", map.remove("area_id"));
			map.put("text", map.remove("area_name"));
			if (deep == 3) {
				map.put("leaf", true);

			} else {
				map.put("leaf", false);
			}

			if (isChecked) {
				map.put("checked", false);
			}
		}
		return menuList;
	}

}