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));
}
}
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;
}
private List<CategoryDto> children;
,这个也就是一个自己的集合,代表自己的孩子ResponseEntity<Object>
不用管,是一个通用的返回数据封装类,然后中间那行就是最里面使用了QueryHelp工具,可以不写SQL语句进行条件查询,然后convert就是一个复制方法,可以类似于BeanUtils里面的copy等等,这就是先将查询到的list复制给Dto类,然后我们进入接下来的Service方法:buildTree:/**
* 构建分类树
* @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;
}
}
版权说明 : 本文为转载文章, 版权归原作者所有 版权申明
原文链接 : https://blog.csdn.net/justleavel/article/details/125584029
内容来源于网络,如有侵权,请联系作者删除!