java 把DataTable數據類型轉換爲樹形結構(多叉樹)

問題分析:一個關係數據庫的表,如圖所示:


可以看到後面四個字段:Country,Province,City,Street 具有邏輯上的從屬結構,現在要把這種數據搞成一個樹形結構,如圖所示:


不是原來的數據轉換而成的,大致就是這個意思,可以想象成,dataTable裏面相同的數據進行單元格合併,然後找到所有的從根到葉子節點的路徑,就算完成任務。JS裏面似乎有很多插件可以實現,但Java中我暫時還沒找到,沒辦法只能自己寫了。從結構上看,應該是一個多叉多級樹形結構,所以在轉換的時候必須具備一定的靈活性,節點的層級也要分明。

首先定義一個node類,描述節點:

public class Node {
	private String id;
	private String pId;
	private String text;
	private Map<String, Object> nodeValue;
	private String path;
	public Node() {
	
	}
	public Node(String id,String pId,String text,String path){
		this.id = id;
		this.pId = pId;
		this.text = text;
		this.path = path;
	}
	public String getId() {
		return id;
	}
	public void setId(String id) {
		this.id = id;
	}
	public String getpId() {
		return pId;
	}
	public void setpId(String pId) {
		this.pId = pId;
	}
	public String getText() {
		return text;
	}
	public void setText(String text) {
		this.text = text;
	}
	public Map<String, Object> getNodeValue() {
		return nodeValue;
	}
	public void setNodeValue(Map<String, Object> nodeValue) {
		this.nodeValue = nodeValue;
	}
	public String getPath() {
		return path;
	}
	public void setPath(String path) {
		this.path = path;
	}
	@Override
	public String toString() {
		String str = "";
		if(this.nodeValue!=null){
			Set<Entry<String,Object>> entrySet = this.nodeValue.entrySet();
			for (Entry<String, Object> entry : entrySet) {
				str+=entry.getKey()+"="+entry.getValue();
			}
		}
		return str;
	}
}

簡單說明一下設計初衷:

1,id和pid就不說了,明眼人一眼就看穿了。text表示的是節點當前顯示內容。

2,nodeValue是Map結構,包含從當前節點到根節點的text,比如:三級節點City=QingDao的nodeValue包含說明什麼呢?答案:{city=QingDao,province=ShanDong,country=China}

3,path屬性表示節點的地址,或者叫做路徑,用來標識某個節點是否已存在,樣式舉例:/China/ShanDong/QingDao

看具體實現類:

public class MultiTree {
	private List<Node> nodeList;
	private Node rootNode;

	public List<Node> getNodeList() {
		return nodeList;
	}

	public void setNodeList(List<Node> nodeList) {
		this.nodeList = nodeList;
	}

	public MultiTree() {
		init();
	}

	public MultiTree(List<Node> nodeList, Node rootNode) {
		this();
		if (nodeList != null) {
			this.nodeList = nodeList;
		}
		if (rootNode != null) {
			this.rootNode = rootNode;
		}
	}

	private void init() {
		nodeList = new ArrayList<Node>();
		rootNode = new Node("0", "-1", "0", "/");
	}

	/**
	 * 把DataTable數據轉換爲DataTree,保證path唯一
	 * @param listMaps
	 * @param args
	 */
	public void convertListMapToTree(List<Map<String, Object>> listMaps,
			String... args) {
		Object value = null;
		String path = "";
		Node pNode = null;
		Node node = null;
		Map<String, Object> nodeValue = new HashMap<String, Object>();
		nodeList.add(rootNode);
		for (Map<String, Object> map : listMaps) {
			path = "";
			pNode = getRoot();
			for (int i = 0;i < args.length;i++) {
				String key = args[i];
				value = map.get(key);
				path += "/" + value;
				node = findNodeByPath(path);
				if (node == null) {
					node = new Node(IdGenerator.uuidGenerator(), pNode.getId(),
							String.valueOf(value), path);
					if(i==args.length-1){
						nodeValue = map;
					}else{
						nodeValue = getNodeValueByPath(path,args);
					}
					node.setNodeValue(nodeValue);
					nodeList.add(node);
				} else {
					pNode = node;
				}
			}
		}
	}
	/**
	 * 根據node path node應該有nodeValue
	 * nodeValue 應該包含父節點的Text,而不應該包含子節點的text,葉子節點應該包含所有的值
	 * @param path
	 * @param args
	 * @return
	 */
	private Map<String, Object> getNodeValueByPath(String path, String[] args) {
		Map<String, Object> nodeValue = new HashMap<String, Object>();
		String[] values = path.split("/");
		
		for (int i = 1;i < values.length;i++) {
			nodeValue.put(args[i-1], values[i]);
		}
		return nodeValue;
	}

	public Node getRoot() {
		return rootNode;
	}
	/**
	 * 某個節點的所有子節點
	 * @param pNode
	 * @return
	 */
	public List<Node> getChildNodes(Node pNode) {
		List<Node> childNodes = new ArrayList<Node>();
		if (pNode == null || pNode.getId() == null) {
			return childNodes;
		}
		for (Node node : nodeList) {
			if (pNode.getId().equals(node.getpId())) {
				childNodes.add(node);
			}
		}
		return childNodes;
	}
	/**
	 * 根據path查找node是否存在(因path唯一)
	 * @param path
	 * @return 找到node返回,否則返回null
	 */
	public Node findNodeByPath(String path) {
		for (Node node : nodeList) {
			if (path.equals(node.getPath())) {
				return node;
			}
		}
		return null;
	}
	/**
	 * 從某個節點開始進行深度度遞歸遍歷
	 * @param pNode
	 */
	public void recursionTraversal(Node pNode){
		List<Node> childNodes = getChildNodes(pNode);
		for (Node node : childNodes) {
			System.out.println(node.toString());
			if(getChildNodes(node).size()>0){
				recursionTraversal(node);
			}
		}
	}
}

此類的核心方法是:  convertListMapToTree 參數,是數據源和節點的字段名稱。

調用方式:

tree.convertListMapToTree(listMaps, "COUNTRY","PROVINCE","CITY","STREET");

執行結果:

/
/China
/China/HeBei
/China/HeBei/BaoDing
/China/HeBei/BaoDing/street1
/China/HeBei/HengShui
/China/HeBei/HengShui/street1
/China/ShanDong
/China/ShanDong/Jian
/China/ShanDong/Jian/street1
/China/ShanDong/QingDao
/China/ShanDong/QingDao/street1
/China/ShanDong/YanTai
/China/ShanDong/YanTai/street1
/Japan
/Japan/JiuZhou
/Japan/JiuZhou/ChangQi
/Japan/JiuZhou/ChangQi/street2
/America
/America/California
/America/California/Los Angeles
/America/California/Los Angeles/street3
/England
/England/Norwich
/England/Norwich/Any
/England/Norwich/Any/street4


此處有幾個點需要注意:

1,字段名稱參數傳遞的順序就是節點的層級順序,從高到低,若是寫錯,則結果不準確。

2,一定要有一個根節點,這是樹形結構的必備,程序中已給出默認根節點,也給出了自定義的接口。

3,本程序中,nodeValue中只包含(葉子節點除外)從當前節點到根節點的字段值,葉子節點包含所有的字段值,比如本例,葉子節點中也包含ID=1這樣的數據,雖然沒有被應用到節點層級中。

4,判斷path是否存在是關鍵一步,如果該步驟不能準確,則整個程序就以失敗告終。

不足之處:

很多地方都在全局查找,效率較低,期待後續改進。

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章