1、通過轉化成json封裝數據
原始數據如下
[
{
"name":"甘肅省", "pid":0, "id":1 }, { "name":"天水市", "pid":1, "id":2 }, { "name":"秦州區", "pid":2, "id":3 }, { "name":"北京市", "pid":0, "id":4 }, { "name":"昌平區", "pid":4, "id":5 } ] 現需要是使用java將以上數據轉為樹形結構，轉化后下的結構如下
[
{
"children":[ { "children":[ { "name":"秦州區", "pid":2, "id":3 } ], "name":"天水市", "pid":1, "id":2 } ], "name":"甘肅省", "pid":0, "id":1 }, { "children":[ { "name":"昌平區", "pid":4, "id":5 } ], "name":"北京市", "pid":0, "id":4 } ] 代碼如下
/** - listToTree - <p>方法說明<p> - 將JSONArray數組轉為樹狀結構 - @param arr 需要轉化的數據 - @param id 數據唯一的標識鍵值 - @param pid 父id唯一標識鍵值 - @param child 子節點鍵值 - @return JSONArray */ 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(); ch.add(aVal); hashVP.put(child, ch); } }else{ r.add(aVal); } } return r; } 測試代碼如下 public static void main(String[] args){ List<Map<String,Object>> data = new ArrayList<>(); Map<String,Object> map = new HashMap<>(); map.put("id",1); map.put("pid",0); map.put("name","甘肅省"); data.add(map); Map<String,Object> map2 = new HashMap<>(); map2.put("id",2); map2.put("pid",1); map2.put("name","天水市"); data.add(map2); Map<String,Object> map3 = new HashMap<>(); map3.put("id",3); map3.put("pid",2); map3.put("name","秦州區"); data.add(map3); Map<String,Object> map4 = new HashMap<>(); map4.put("id",4); map4.put("pid",0); map4.put("name","北京市"); data.add(map4); Map<String,Object> map5 = new HashMap<>(); map5.put("id",5); map5.put("pid",4); map5.put("name","昌平區"); data.add(map5); System.out.println(JSON.toJSONString(data)); JSONArray result = listToTree(JSONArray.parseArray(JSON.toJSONString(data)),"id","pid","children"); System.out.println(JSON.toJSONString(result)); } 2、通過java8 stream轉換
我在網上找了很多方法，自己寫的這個思路清晰，代碼量少，希望能找到志同道合的朋友，看看還有沒有優化的地方。
import lombok.AllArgsConstructor; import lombok.Data; import lombok.NoArgsConstructor; @Data @AllArgsConstructor @NoArgsConstructor public class ZhField { @Id @GeneratedValue(strategy = GenerationType.IDENTITY) private String id; /** * 上級領域id */ @Column(name = "parent_id") private String parentId; /** * 領域名稱 */ private String name; /** * 排序 */ private Integer sort; } import lombok.AllArgsConstructor; import lombok.Data; import lombok.NoArgsConstructor; import java.io.Serializable; import java.util.List; import java.util.Map; @Data @AllArgsConstructor @NoArgsConstructor public class TreeMenuNode implements Serializable { private String id; private String parentId; private String name; private Integer sort; private List<TreeMenuNode> children; private Boolean isAble; /**20180929zhw添加 樹的額外屬性(至少含有父節點ID:"parentId")**/ private Map<String,Object> attributes; } package com.egaoqi.service.impl.company; import com.egaoqi.entity.ZhField; import com.egaoqi.util.TreeMenuNode; import java.util.ArrayList; import java.util.Comparator; import java.util.List; import java.util.Map; import java.util.stream.Collectors; /** * @author zhizhao * @description * @create 2018-11-14 9:07 */ public class toModel { private static void forEach(Map<String, List<TreeMenuNode>> collect, TreeMenuNode treeMenuNode) { List<TreeMenuNode> treeMenuNodes = collect.get(treeMenuNode.getId()); if(collect.get(treeMenuNode.getId())!=null){ //排序 treeMenuNodes.sort((u1, u2) -> u1.getSort().compareTo(u2.getSort())); treeMenuNodes.stream().sorted(Comparator.comparing(TreeMenuNode::getSort)).collect(Collectors.toList()); treeMenuNode.setChildren(treeMenuNodes); treeMenuNode.getChildren().forEach(t->{ forEach(collect,t); }); } } public static void main(String[] args) { List<ZhField> zhFields = new ArrayList<>(); List<TreeMenuNode> treeNodeList = new ArrayList<>(); //轉換數據,這個是前端需要的格式。 zhFields.forEach(t->{ TreeMenuNode treeMenuNode = new TreeMenuNode(); treeMenuNode.setId(t.getId()); treeMenuNode.setParentId(t.getParentId()); treeMenuNode.setName(t.getName()); treeMenuNode.setSort(t.getSort()); treeNodeList.add(treeMenuNode); }); //分組 Map<String, List<TreeMenuNode>> collect = treeNodeList.stream().collect(Collectors.groupingBy(TreeMenuNode::getParentId)); //樹形結構 肯定有一個根部，我的這個根部的就是parentId.euqal("0"),而且只有一個就get（"0"） TreeMenuNode treeMenuNode = collect.get("0").get(0); //拼接數據 forEach(collect, treeMenuNode); } }
my
package com.hainei.service.impl.wp.jgzz; import com.hainei.common.constants.BaseConstant; import com.hainei.common.enums.YesOrNo; import com.hainei.common.token.JwtTokenUtil; import com.hainei.common.tree.TreeObject; import com.hainei.common.tree.TreeService; import com.hainei.common.tree.myTree.ToModel; import com.hainei.common.tree.myTree.TreeMenuNode; import com.hainei.common.tree.myTree.ZhField; import com.hainei.mapper.base.BaseUserMapper; import com.hainei.mapper.wp.jgzz.WpJgzzAqglwlMapper; import com.hainei.mapper.wp.sbgl.WpYfbyjlMapper; import com.hainei.pojo.bo.wp.jgzz.WpJgzzAqglwlBO; import com.hainei.pojo.bo.wp.sbgl.WpYfbyBO; import com.hainei.pojo.model.wp.jgzz.WpJgzzAqglwl; import com.hainei.pojo.model.wp.sbgl.WpYfby; import com.hainei.pojo.model.wp.sbgl.WpYfbyjl; import com.hainei.pojo.vo.base.BaseTreeOrganizationWithRolesVO; import com.hainei.pojo.vo.wp.jgzz.WpAqglwlVO; import com.hainei.pojo.vo.wp.sbgl.WpYfbyVO; import com.hainei.service.base.BaseDataTableService; import com.hainei.service.base.BaseOrganizationService; import com.hainei.service.wp.jgzz.WpAqglwlService; import com.hainei.service.wp.zyaqgl.WpZypNumService; import org.apache.commons.collections.CollectionUtils; import org.springframework.beans.BeanUtils; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.stereotype.Service; import org.springframework.transaction.annotation.Transactional; import javax.servlet.http.HttpServletRequest; import java.util.*; import java.util.stream.Collectors; /** * Created with IntelliJ IDEA. * User：wq * Date：2021/1/8 * Time: 15:53 * Description: 目標職責 */ @Service @Transactional public class WpAqglwlServiceImpl implements WpAqglwlService { @Autowired private WpJgzzAqglwlMapper jgzzAqglwlMapper; @Autowired private WpYfbyjlMapper wpYfbyjlMapper; @Autowired private HttpServletRequest request; @Autowired private BaseOrganizationService organizationService; @Autowired private BaseDataTableService dataTableService; @Autowired private WpZypNumService zypNumService; @Autowired private BaseUserMapper userMapper; @Override public void addRecord(WpJgzzAqglwlBO wpJgzzAqglwlBO) { String token = request.getHeader(BaseConstant.ACCESS_TOKEN); String userId = JwtTokenUtil.getUserId(token); String id = UUID.randomUUID().toString().replaceAll("-", ""); WpJgzzAqglwl wpJgzzAqglwl = new WpJgzzAqglwl(); BeanUtils.copyProperties(wpJgzzAqglwlBO, wpJgzzAqglwl); wpJgzzAqglwl.setId(id); //wpYfby.setFileNum(zypNumService.getZypNum("AQWJ")); wpJgzzAqglwl.setIsDeleted(YesOrNo.NO.type); wpJgzzAqglwl.setGmtCreatedOn(new Date()); wpJgzzAqglwl.setGmtCreatedBy(userId); //添加數據權限表 Set<String> organizationIdsByUserId = organizationService.getOrganizationIdsByUserId(userId); //3.插入數據權限 dataTableService.insert(organizationIdsByUserId, id); jgzzAqglwlMapper.insertSelective(wpJgzzAqglwl); } @Override public void deleteByIds(List<String> ids) { for (String id : ids) { WpJgzzAqglwl wpJgzzAqglwl = new WpJgzzAqglwl(); wpJgzzAqglwl.setId(id); wpJgzzAqglwl.setIsDeleted(YesOrNo.YES.type); jgzzAqglwlMapper.updateByPrimaryKeySelective(wpJgzzAqglwl); } } @Override public void updateRecord(WpJgzzAqglwlBO wpJgzzAqglwlBO) { String token = request.getHeader(BaseConstant.ACCESS_TOKEN); String userId = JwtTokenUtil.getUserId(token); WpJgzzAqglwl wpJgzzAqglwl = new WpJgzzAqglwl(); wpJgzzAqglwl.setGmtUpdatedBy(userId); BeanUtils.copyProperties(wpJgzzAqglwlBO, wpJgzzAqglwl); jgzzAqglwlMapper.updateByPrimaryKeySelective(wpJgzzAqglwl); } @Override public Map<String,Object> getById(String id) { Map byId = jgzzAqglwlMapper.getById(id); // putWfValue(byId); return byId; } @Override public List<TreeMenuNode> listAll(WpJgzzAqglwlBO wpJgzzAqglwlBO) { String token = request.getHeader(BaseConstant.ACCESS_TOKEN); String userId = JwtTokenUtil.getUserId(token); Collection<TreeObject<WpAqglwlVO>> treeRootList = null; WpAqglwlVO parent =null; TreeMenuNode treeMenuNode = new TreeMenuNode(); wpJgzzAqglwlBO.setGmtCreatedBy(userId); List<WpAqglwlVO> wpAqglwlVOS = jgzzAqglwlMapper.listAll(wpJgzzAqglwlBO); WpAqglwlVO wpAqglwlVO = wpAqglwlVOS.stream().filter(p->p.getIsDeleted()==0).filter(p -> p.getParentId() == null||"".equals(p.getParentId())).collect(Collectors.toList()).get(0); BeanUtils.copyProperties(wpAqglwlVO,treeMenuNode); List<TreeMenuNode> stringListMap = transTree(treeMenuNode, wpAqglwlVOS); return stringListMap; } public List<TreeMenuNode> transTree(TreeMenuNode treeMenuNode1,List<WpAqglwlVO> wpAqglwlVOS) { List<TreeMenuNode> treeNodeList = new ArrayList<>(); //轉換數據,這個是前端需要的格式。 wpAqglwlVOS.forEach(t->{ TreeMenuNode treeMenuNode = new TreeMenuNode(); treeMenuNode.setId(t.getId()); treeMenuNode.setParentId(t.getParentId()); treeMenuNode.setManageName(t.getName()); treeMenuNode.setSortIndex(t.getSortIndex()); treeMenuNode.setId(t.getId()); treeMenuNode.setPrincipal(t.getPrincipal()); treeMenuNode.setPrincipalPhone(t.getPrincipalPhone()); treeMenuNode.setRemark(t.getRemark()); treeMenuNode.setSafeDuty(t.getSafeDuty()); treeNodeList.add(treeMenuNode); }); //分組 Map<String, List<TreeMenuNode>> collect = treeNodeList.stream().collect(Collectors.groupingBy(TreeMenuNode::getParentId)); //樹形結構 肯定有一個根部，我的這個根部的就是parentId.euqal("0"),而且只有一個就get（"0"） //TreeMenuNode treeMenuNode = collect.get("0").get(0); //拼接數據 List<TreeMenuNode> treeMenuNodes = ToModel.forEach(collect, treeMenuNode1); return treeMenuNodes; } }
TreeMenuNode
package com.hainei.common.tree.myTree; import io.swagger.annotations.ApiModelProperty; import lombok.AllArgsConstructor; import lombok.Data; import lombok.NoArgsConstructor; import javax.persistence.Column; import javax.persistence.Id; import java.io.Serializable; import java.util.List; import java.util.Map; @Data @AllArgsConstructor @NoArgsConstructor public class TreeMenuNode implements Serializable { @Id private String id; /** * 父節點id */ @Column(name = "parent_id") @ApiModelProperty(value = "父節點id",example = "父節點id") private String parentId; /** * 子節點排序 */ @Column(name = "sort_index") @ApiModelProperty(value = "子節點排序",example = "子節點排序") private Integer sortIndex; /** * 名稱 */ @Column(name = "manage_name") @ApiModelProperty(value = "名稱",example = "名稱") private String manageName; /** * 負責人 */ @ApiModelProperty(value = "負責人",example = "負責人") private String principal; /** * 負責人聯系方式 */ @Column(name = "principal_phone") @ApiModelProperty(value = "負責人聯系方式",example = "負責人聯系方式") private String principalPhone; /** * 備注 */ @ApiModelProperty(value = "備注",example = "備注") private String remark; @ApiModelProperty(value = "安全職責",example = "安全職責") private String safeDuty; // private String name; // private Integer sort; private List<TreeMenuNode> children; private Boolean isAble; /**20180929zhw添加 樹的額外屬性(至少含有父節點ID:"parentId")**/ private Map<String,Object> attributes; }
每天進步一步步，堅持堅持堅持