Springboot實現給前端返回一個tree結構方法

2022-07-06 14:05:36

1：首先我們看一下資料庫的表：

這裡的pid就是代表他的父節點id，如果沒有父節點，那麼pid就是0，上面的表就可以看作是一個tree結構，那麼我們怎樣去將這個tree結構返回給前端呢？

2：首先寫好資料庫對應的實體類和Dto層：

package com.wyr.modules.example.domain;
import com.baomidou.mybatisplus.annotation.FieldFill;
import com.baomidou.mybatisplus.annotation.TableField;
import com.baomidou.mybatisplus.annotation.TableId;
import lombok.Data;
import com.baomidou.mybatisplus.annotation.TableName;
import cn.hutool.core.bean.BeanUtil;
import cn.hutool.core.bean.copier.CopyOptions;
import javax.validation.constraints.*;
import java.sql.Timestamp;
import java.io.Serializable;
/**
 * @author jianyijun
 * @date 2022-07-02
 */
@Data
@TableName("store_category")
public class Category implements Serializable {
    /** 商品分類表ID */
    @TableId
    private Integer id;
    /** 父id */
    @NotNull
    private Integer pid;
    /** 分類名稱 */
    @NotBlank
    private String cateName;
    /** 排序 */
    private Integer sort;
    /** 圖示 */
    private String pic;
    /** 是否推薦 */
    private Integer isShow;
    /** 新增時間 */
    @TableField(fill= FieldFill.INSERT)
    private Timestamp createTime;
    /** 更新時間 */
    @TableField(fill= FieldFill.INSERT_UPDATE)
    private Timestamp updateTime;
    /** 刪除狀態 */
    private Integer isDel;
    public void copy(Category source){
        BeanUtil.copyProperties(source,this, CopyOptions.create().setIgnoreNullValue(true));
    }
}

Dto層：

package com.wyr.modules.example.service.dto;
import lombok.Data;
import java.sql.Timestamp;
import java.io.Serializable;
import java.util.List;
/**
 * @author jianyijun
 * @date 2022-07-02
 */
@Data
public class CategoryDto implements Serializable {
    /** 商品分類表ID */
    private Long id;
    /** 父id */
    private Long pid;
    /** 分類名稱 */
    private String cateName;
    /** 排序 */
    private Integer sort;
    /** 圖示 */
    private String pic;
    /** 是否推薦 */
    private Integer isShow;
    /** 新增時間 */
    private Timestamp createTime;
    /** 更新時間 */
    private Timestamp updateTime;
    /** 刪除狀態 */
    private Integer isDel;
    private List<CategoryDto> children;
}

這裡注意一下Dto層多餘的欄位：private List<CategoryDto> children;，這個也就是一個自己的集合，代表自己的孩子

3：這裡介紹一下什麼是Dto層，以及一些區別：

(1) entity 裡的每一個欄位，與資料庫相對應，

(2) vo 裡的每一個欄位，是和你前臺 html 頁面相對應，

(3) dto 這是用來轉換從 entity 到 vo，或者從 vo 到 entity 的中間的東西。（DTO中擁有的欄位應該是entity中或者是vo中的一個子集）

4:然後是controller層：

ResponseEntity<Object>不用管，是一個通用的返回資料封裝類，然後中間那行就是最裡面使用了QueryHelp工具，可以不寫SQL語句進行條件查詢，然後convert就是一個複製方法，可以類似於BeanUtils裡面的copy等等，這就是先將查詢到的list複製給Dto類，然後我們進入接下來的Service方法：buildTree：

5：業務層：

 /**
     * 構建分類樹
     * @param categoryDtos 原始資料
     * @return
     */
    @Override
    public Map<String, Object> buildTree(List<CategoryDto> categoryDtos) {
        List<CategoryDto> trees = new ArrayList<>();
        Set<Long> ids = new HashSet<>();
        for (CategoryDto categoryDto :categoryDtos) {
            if (categoryDto.getPid() == 0) {
                trees.add(categoryDto);
            }
            for (CategoryDto it : categoryDtos) {
                if (it.getPid().equals(categoryDto.getId())) {
                    if (categoryDto.getChildren() == null) {
                        categoryDto.setChildren(new ArrayList<>());
                    }
                    categoryDto.getChildren().add(it);
                    ids.add(it.getId());
                }
            }

        }
        Map<String, Object> map = new HashMap<>(2);
        if (trees.size() == 0){
            trees = categoryDtos.stream().filter(s -> !ids.contains(s.getId())).collect(Collectors.toList());
        }
        map.put("content",trees);
        map.put("totalElements", categoryDtos.size());
        return map;
    }
}

到此這篇關於Springboot實現給前端返回一個tree結構方法的文章就介紹到這了,更多相關Springboot tree結構內容請搜尋it145.com以前的文章或繼續瀏覽下面的相關文章希望大家以後多多支援it145.com！

Springboot實現給前端返回一個tree結構方法

熱門文章