diff --git a/spzx-manager/src/main/java/com/atguigu/spzx/manger/excel/CategoryExcelListener.java b/spzx-manager/src/main/java/com/atguigu/spzx/manger/excel/CategoryExcelListener.java new file mode 100644 index 0000000..7eedc59 --- /dev/null +++ b/spzx-manager/src/main/java/com/atguigu/spzx/manger/excel/CategoryExcelListener.java @@ -0,0 +1,46 @@ +package com.atguigu.spzx.manger.excel; + +import com.alibaba.excel.context.AnalysisContext; +import com.alibaba.excel.event.AnalysisEventListener; +import com.alibaba.excel.util.ListUtils; +import com.atguigu.spzx.manger.mapper.CategoryMapper; +import com.atguigu.spzx.model.vo.product.CategoryExcelVo; +import lombok.Data; +import lombok.EqualsAndHashCode; +import lombok.RequiredArgsConstructor; + +import java.util.List; + +@EqualsAndHashCode(callSuper = true) +@Data +@RequiredArgsConstructor +public class CategoryExcelListener extends AnalysisEventListener { + // 每个5条存储数据库，方便垃圾回收 + private static final int BATCH_COUNT = 100; + // 缓存的数据 + private final List cachedDataList = ListUtils.newArrayListWithExpectedSize(BATCH_COUNT); + private CategoryMapper categoryMapper; + + @Override + public void invoke(T t, AnalysisContext analysisContext) { + CategoryExcelVo data = (CategoryExcelVo) t; + cachedDataList.add(data); + if (cachedDataList.size() >= BATCH_COUNT) { + saveData(); + } + } + + @Override + public void doAfterAllAnalysed(AnalysisContext analysisContext) { + // excel解析完毕以后需要执行的代码 + // 这里也要保存数据，确保最后遗留的数据也存储到数据库 + saveData(); + } + + /** + * 写入数据库 + */ + private void saveData() { + categoryMapper.batchInsert(cachedDataList); + } +} diff --git a/spzx-common/common-util/src/main/java/com/atguigu/excel/ExcelListener.java b/spzx-manager/src/main/java/com/atguigu/spzx/manger/excel/ExcelListener.java similarity index 95% rename from spzx-common/common-util/src/main/java/com/atguigu/excel/ExcelListener.java rename to spzx-manager/src/main/java/com/atguigu/spzx/manger/excel/ExcelListener.java index d11cae7..d308942 100644 --- a/spzx-common/common-util/src/main/java/com/atguigu/excel/ExcelListener.java +++ b/spzx-manager/src/main/java/com/atguigu/spzx/manger/excel/ExcelListener.java @@ -1,4 +1,4 @@ -package com.atguigu.excel; +package com.atguigu.spzx.manger.excel; import com.alibaba.excel.context.AnalysisContext; import com.alibaba.excel.event.AnalysisEventListener; diff --git a/spzx-manager/src/main/java/com/atguigu/spzx/manger/mapper/CategoryMapper.java b/spzx-manager/src/main/java/com/atguigu/spzx/manger/mapper/CategoryMapper.java index 6e785e7..7b9bcf7 100644 --- a/spzx-manager/src/main/java/com/atguigu/spzx/manger/mapper/CategoryMapper.java +++ b/spzx-manager/src/main/java/com/atguigu/spzx/manger/mapper/CategoryMapper.java @@ -1,6 +1,7 @@ package com.atguigu.spzx.manger.mapper; import com.atguigu.spzx.model.entity.product.Category; +import com.atguigu.spzx.model.vo.product.CategoryExcelVo; import org.apache.ibatis.annotations.Mapper; import java.util.List; @@ -22,4 +23,18 @@ public interface CategoryMapper { * @return int */ int countByParentId(Long id); + + /** + * 读取所以分类数据 + * + * @return 分类实体类列表 + */ + List selectAll(); + + /** + * 批量插入数据库 + * + * @param cachedDataList Excel表品牌管理实体类 + */ + void batchInsert(List cachedDataList); } diff --git a/spzx-manager/src/main/java/com/atguigu/spzx/manger/service/impl/CategoryServiceImpl.java b/spzx-manager/src/main/java/com/atguigu/spzx/manger/service/impl/CategoryServiceImpl.java index 722ddce..233ce79 100644 --- a/spzx-manager/src/main/java/com/atguigu/spzx/manger/service/impl/CategoryServiceImpl.java +++ b/spzx-manager/src/main/java/com/atguigu/spzx/manger/service/impl/CategoryServiceImpl.java @@ -1,11 +1,15 @@ package com.atguigu.spzx.manger.service.impl; import com.alibaba.excel.EasyExcel; +import com.atguigu.exception.BunnyException; +import com.atguigu.spzx.manger.excel.ExcelListener; import com.atguigu.spzx.manger.mapper.CategoryMapper; import com.atguigu.spzx.manger.service.CategoryService; import com.atguigu.spzx.model.entity.product.Category; import com.atguigu.spzx.model.vo.product.CategoryExcelVo; +import com.atguigu.spzx.model.vo.result.ResultCodeEnum; import jakarta.servlet.http.HttpServletResponse; +import lombok.extern.slf4j.Slf4j; import org.springframework.beans.BeanUtils; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.stereotype.Service; @@ -19,6 +23,7 @@ import java.util.ArrayList; import java.util.List; @Service +@Slf4j public class CategoryServiceImpl implements CategoryService { @Autowired private CategoryMapper categoryMapper; @@ -50,7 +55,29 @@ public class CategoryServiceImpl implements CategoryService { */ @Override public void exportData(HttpServletResponse response) { - // TODO 导出数据 + // 设置响应结果类型 + response.setContentType("application/vnd.ms-excel"); + response.setCharacterEncoding("utf-8"); + // 这里URLEncoder.encode可以防止中文乱码 当然和easyexcel没有关系 + String filename = URLEncoder.encode("分类数据", StandardCharsets.UTF_8); + response.setHeader("Content-disposition", "attachment;filename=" + filename + ".xlsx"); + // 查询数据库中的数据 + List categoryList = categoryMapper.selectAll(); + ArrayList excelVoArrayList = new ArrayList<>(); + // 将从数据库中查询到的Category对象转换成CategoryExcelVo对象 + categoryList.forEach(category -> { + CategoryExcelVo vo = new CategoryExcelVo(); + BeanUtils.copyProperties(category, vo, CategoryExcelVo.class); + excelVoArrayList.add(vo); + }); + + try { + // 写入文件 + EasyExcel.write(response.getOutputStream(), CategoryExcelVo.class).sheet("分类数据").doWrite(excelVoArrayList); + } catch (IOException exception) { + log.error("文件写入失败：{}", exception.getMessage()); + throw new BunnyException(ResultCodeEnum.WRITE_FILE_ERROR); + } } /** @@ -60,6 +87,13 @@ public class CategoryServiceImpl implements CategoryService { */ @Override public void importData(MultipartFile file) { - // TODO 导入功能 + // 创建监听器对象，传递mapper对象 + ExcelListener listener = new ExcelListener<>(); + // 调用read方法读取excel数据 + try { + EasyExcel.read(file.getInputStream(), CategoryExcelVo.class, listener).sheet().doRead(); + } catch (IOException e) { + throw new BunnyException(ResultCodeEnum.DATA_ERROR); + } } } diff --git a/spzx-manager/src/main/resources/mapper/CategoryMapper.xml b/spzx-manager/src/main/resources/mapper/CategoryMapper.xml index 1a41c0d..f20139e 100644 --- a/spzx-manager/src/main/resources/mapper/CategoryMapper.xml +++ b/spzx-manager/src/main/resources/mapper/CategoryMapper.xml @@ -8,6 +8,17 @@ id,name,image_url,parent_id,status,order_num,create_time,update_time,is_deleted + + + insert into category (id, name, image_url, parent_id, status, order_num, create_time, update_time, is_deleted) + values + + (#{item.id}, #{item.name}, #{item.image_url}, #{item.parent_id}, #{item.status}, #{item.order_num}, + #{item.create_time}, #{item.update_time}, + #{item.is_deleted}) + + + + + + diff --git a/spzx-manager/src/test/java/com/atguigu/spzx/manger/EasyExcelTest.java b/spzx-manager/src/test/java/com/atguigu/spzx/manger/EasyExcelTest.java index 9c54703..6bdc9bc 100644 --- a/spzx-manager/src/test/java/com/atguigu/spzx/manger/EasyExcelTest.java +++ b/spzx-manager/src/test/java/com/atguigu/spzx/manger/EasyExcelTest.java @@ -1,7 +1,7 @@ package com.atguigu.spzx.manger; import com.alibaba.excel.EasyExcel; -import com.atguigu.excel.ExcelListener; +import com.atguigu.spzx.manger.excel.ExcelListener; import com.atguigu.spzx.model.vo.product.CategoryExcelVo; import org.junit.jupiter.api.Test; import org.springframework.boot.test.context.SpringBootTest; diff --git a/spzx-model/src/main/java/com/atguigu/spzx/model/entity/product/Category.java b/spzx-model/src/main/java/com/atguigu/spzx/model/entity/product/Category.java index a9d1fa0..0178d0a 100644 --- a/spzx-model/src/main/java/com/atguigu/spzx/model/entity/product/Category.java +++ b/spzx-model/src/main/java/com/atguigu/spzx/model/entity/product/Category.java @@ -3,32 +3,32 @@ package com.atguigu.spzx.model.entity.product; import com.atguigu.spzx.model.entity.base.BaseEntity; import io.swagger.v3.oas.annotations.media.Schema; import lombok.Data; +import lombok.EqualsAndHashCode; import java.util.List; +@EqualsAndHashCode(callSuper = true) @Data @Schema(description = "分类实体类") public class Category extends BaseEntity { + @Schema(description = "分类名称") + private String name; - @Schema(description = "分类名称") - private String name; + @Schema(description = "分类图片url") + private String imageUrl; - @Schema(description = "分类图片url") - private String imageUrl; + @Schema(description = "父节点id") + private Long parentId; - @Schema(description = "父节点id") - private Long parentId; + @Schema(description = "分类状态: 是否显示[0-不显示，1显示]") + private Integer status; - @Schema(description = "分类状态: 是否显示[0-不显示，1显示]") - private Integer status; + @Schema(description = "排序字段") + private Integer orderNum; - @Schema(description = "排序字段") - private Integer orderNum; - - @Schema(description = "是否存在子节点") - private Boolean hasChildren; - - @Schema(description = "子节点List集合") - private List children; + @Schema(description = "是否存在子节点") + private Boolean hasChildren; + @Schema(description = "子节点List集合") + private List children; } \ No newline at end of file diff --git a/spzx-model/src/main/java/com/atguigu/spzx/model/vo/product/CategoryExcelVo.java b/spzx-model/src/main/java/com/atguigu/spzx/model/vo/product/CategoryExcelVo.java index 45062a5..e0ed073 100644 --- a/spzx-model/src/main/java/com/atguigu/spzx/model/vo/product/CategoryExcelVo.java +++ b/spzx-model/src/main/java/com/atguigu/spzx/model/vo/product/CategoryExcelVo.java @@ -1,6 +1,7 @@ package com.atguigu.spzx.model.vo.product; import com.alibaba.excel.annotation.ExcelProperty; +import io.swagger.v3.oas.annotations.media.Schema; import lombok.AllArgsConstructor; import lombok.Data; import lombok.NoArgsConstructor; @@ -8,24 +9,23 @@ import lombok.NoArgsConstructor; @Data @AllArgsConstructor @NoArgsConstructor +@Schema(description = "Excel表品牌管理实体类") public class CategoryExcelVo { + @ExcelProperty(value = "id", index = 0) + private Long id; - @ExcelProperty(value = "id" ,index = 0) - private Long id; + @ExcelProperty(value = "名称", index = 1) + private String name; - @ExcelProperty(value = "名称" ,index = 1) - private String name; + @ExcelProperty(value = "图片url", index = 2) + private String imageUrl; - @ExcelProperty(value = "图片url" ,index = 2) - private String imageUrl ; + @ExcelProperty(value = "上级id", index = 3) + private Long parentId; - @ExcelProperty(value = "上级id" ,index = 3) - private Long parentId; - - @ExcelProperty(value = "状态" ,index = 4) - private Integer status; - - @ExcelProperty(value = "排序" ,index = 5) - private Integer orderNum; + @ExcelProperty(value = "状态", index = 4) + private Integer status; + @ExcelProperty(value = "排序", index = 5) + private Integer orderNum; } \ No newline at end of file diff --git a/spzx-model/src/main/java/com/atguigu/spzx/model/vo/result/ResultCodeEnum.java b/spzx-model/src/main/java/com/atguigu/spzx/model/vo/result/ResultCodeEnum.java index 78e5df2..2b83f24 100644 --- a/spzx-model/src/main/java/com/atguigu/spzx/model/vo/result/ResultCodeEnum.java +++ b/spzx-model/src/main/java/com/atguigu/spzx/model/vo/result/ResultCodeEnum.java @@ -8,15 +8,16 @@ import lombok.Getter; @Getter public enum ResultCodeEnum { SUCCESS(200, "成功"), + PERMISSION(401, "没有权限"), FAIL(500, "失败"), USE_NOT_EXIST(500, "失败"), - SERVICE_ERROR(2012, "服务异常"), - DATA_ERROR(204, "数据异常"), - LOGIN_AUTH(208, "需要先登陆"), - LOGIN_MOBLE_ERROR(208, "登录验证失败"), - ACCOUNT_DEACTIVATION(208, "账户停用"), - PERMISSION(209, "没有权限"), + SERVICE_ERROR(500, "服务异常"), + DATA_ERROR(500, "数据异常"), + LOGIN_AUTH(500, "需要先登陆"), + LOGIN_MOBLE_ERROR(500, "登录验证失败"), + ACCOUNT_DEACTIVATION(500, "账户停用"), USERNAME_IS_EXISTS(500, "用户名已存在"), + WRITE_FILE_ERROR(500, "写入文件失败"), ; private final Integer code;