简单list转化为tree结构
1 @Override 2 public List<Module> getTree() { 3 List<Module> list = moduleMapper.queryList(); 4 // List<Module> modules = makeTree(list); 5 return list; 6 } 7 8 private List<Module> makeTree(List<Module> list) { 9 List<Module> parent = new ArrayList<Module>(); 10 // get parentId = null; 11 for (Module e : list) { 12 if (e.getParent_Module() == null) { 13 e.setChildren(new ArrayList<Module>(0)); 14 parent.add(e); 15 } 16 } 17 // 删除parentId = null; 18 list.removeAll(parent); 19 20 makeChildren(parent, list); 21 22 return parent; 23 } 24 25 private void makeChildren(List<Module> parent, List<Module> children) { 26 if (children.isEmpty()) { 27 return; 28 } 29 30 List<Module> tmp = new ArrayList<Module>(); 31 for (Module c1 : parent) { 32 for (Module c2 : children) { 33 c2.setChildren(new ArrayList<Module>(0)); 34 if (c1.getId().equals(c2.getParent_Module().getId())) { 35 c1.getChildren().add(c2); 36 tmp.add(c2); 37 } 38 } 39 } 40 41 children.removeAll(tmp); 42 43 makeChildren(tmp, children); 44 }
补充po类:
package com.zktx.platform.entity.tb;
import java.util.ArrayList;
import java.util.List;
public class Module {
private Integer id;
private String class_name;
private String description;
private String name;
private Integer priority;
private String sn;
private String url;
private Integer parent_id;
private Module parent_Module;
private List<Module> children = new ArrayList<Module>();
public Integer getId() {
return id;
}
public void setId(Integer id) {
this.id = id;
}
public String getClass_name() {
return class_name;
}
public void setClass_name(String class_name) {
this.class_name = class_name;
}
public String getDescription() {
return description;
}
public void setDescription(String description) {
this.description = description;
}
public String getName() {
return name;
}
public void setName(String name) {
this.name = name;
}
public Integer getPriority() {
return priority;
}
public void setPriority(Integer priority) {
this.priority = priority;
}
public String getSn() {
return sn;
}
public void setSn(String sn) {
this.sn = sn;
}
public String getUrl() {
return url;
}
public void setUrl(String url) {
this.url = url;
}
public Integer getParent_id() {
return parent_id;
}
public void setParent_id(Integer parent_id) {
this.parent_id = parent_id;
}
public Module getParent_Module() {
return parent_Module;
}
public void setParent_Module(Module parent_Module) {
this.parent_Module = parent_Module;
}
public List<Module> getChildren() {
return children;
}
public void setChildren(List<Module> children) {
this.children = children;
}
}
浙公网安备 33010602011771号