SpringBoot Java将数据封装成树形结构返回给前端

红太狼 2022-12-16 09:07 258阅读 0赞

> 根据level\_code列封装树形数据结构  
> 每四位为一个节点  
> 0001、0002…为根节点，诸如此类四位为第一层根节点  
> 00010001、00010002为0001子节点

## 数据库表（无parent\_id列） ##

![在这里插入图片描述][20201019170354603.png_pic_center]

## 实体类（只列部分代码） ##

/** * 由于数据库表中未定义父级id,实体类中使用@TableField(exist = false)标记 * 父级id */
    	@TableField(exist = false)
    	@ApiModelProperty(value = "父级id")
    	private Integer parentId;

## 新建实体类用来封装树形数据 ##

/** * 物料类型树对象 * 这里具体TreeNode可以自己查一下 */
    @Data
    @ApiModel(value = "物料类型树")
    @EqualsAndHashCode(callSuper = true)
    public class MaterialTypeTree extends TreeNode { 
    
    	@ApiModelProperty(value = "物料类型名称")
    	private String name;
    
    }

## Service层（返回树形菜单集合） ##

/** * 查询全部物料类型树 * @return 树 */
    	@Override
    	public List<MaterialTypeTree> selectTree() { 
    		// 查询全部物料类型树
    		List<MaterialType> materialTypeAllList = materialTypeMapper.selectList(Wrappers.emptyWrapper());
    
    		// 物料类型
    		List<MaterialTypeTree> collect = materialTypeAllList.stream().map(materialType -> { 
    			MaterialTypeTree node = new MaterialTypeTree();
    			node.setId(materialType.getId());
    			//如果是根节点 父id为0
    			if (materialType.getLevelCode().length() == 4) { 
    				materialType.setParentId(0);
    			} else { 
    				//不是根节点 截掉最后四位 剩余为父节点的levelCode 根据levelCode查找对应的id 为父id
    				String temp = materialType.getLevelCode();
    				temp = temp.substring(0 , temp.length()-4);
    				Integer id = baseMapper.QueryIdByLevelCode(temp);
    				materialType.setParentId(id);
    			}
    			node.setParentId(materialType.getParentId());
    			node.setName(materialType.getTypeName());
    
    			return node;
    		}).collect(Collectors.toList());
    		return TreeUtil.build(collect, 0);
    	}

## 返回到前端的树形数据结构 ##

![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2thX3hpbmds_size_16_color_FFFFFF_t_70_pic_center]

[20201019170354603.png_pic_center]: /images/20221123/60543f6312bc4494939570f9dfa07be7.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2thX3hpbmds_size_16_color_FFFFFF_t_70_pic_center]: /images/20221123/7fdcb801f9b3417f92efb5bd0a2cab2d.png