Excel文档的处理

Excel文档的导出

Excel文档导出可以分为两种情况

情况一：模板类型（不会随意变动）

情况二：数据动态添加（数据库查询后写入到excel文件中）

方法一：提前准备好Excel文件（适用于模板类型）

private final FileController fileController;

@ApiOperation("导出信息模板")
    @AnonymousGetMapping("/download-template")
    public ResponseEntity<Resource> downloadLoanExcel(HttpServletRequest request, HttpServletResponse response) {
        // 这儿直接调用的是FileController中的方法。
        // 模板名称是后端写死的值
        return fileController.downloadFile("template.xlsx",request,response);
    }


// 也可以调用Service层中的方法
@ApiOperation("导出信息模板")
    @AnonymousGetMapping("/download-template")
    public ResponseEntity<Resource> downloadLoanExcel(HttpServletRequest request, HttpServletResponse response) throws IOException {
        // 调用fileService层方法 模板是后端写死的值
        Resource resource = fileService.loadFileAsResource("template.xlsx");
        FileUtils.downloadFile(request,response,resource.getFile(),"template1.xlsx",false);
        // 第四个参数表示传给前端显示的名称
        return null;
    }

方法二：通过FileUtils中的downloadExcel生成（适用于模板类型/动态添加）

// 摘取的DeptServiceImpl中的导出方法
@Override
public void download(List<DeptDTO> deptDtos, HttpServletResponse response) throws IOException {
    List<Map<String, Object>> list = new ArrayList<>();
    for (DeptDTO deptDto : deptDtos) {
        Map<String, Object> map = new LinkedHashMap<>();
        map.put("部门名称", deptDto.getName());
        map.put("部门状态", deptDto.getEnabled() ? "启用" : "停用");
        map.put("创建日期", ObjectUtil.isNull(deptDto.getCreateTime()) ? null :
                LocalDateTimeUtil.format(deptDto.getCreateTime()
                        , DatePattern.NORM_DATETIME_FORMATTER));
        list.add(map);
    }
    FileUtils.downloadExcel(list, response);
}

Excel文档的读取

使用alibaba提供的EasyExcel

文档地址：EasyExcel · 语雀 (yuque.com)

准备好一个写好的excel文件

[图片上传失败...(image-d06045-1650208801959)]

步骤一：引入依赖

<dependency>
    <groupId>com.alibaba</groupId>
    <artifactId>easyexcel</artifactId>
    <version>3.0.5</version>
</dependency>

步骤二：对应的实体类

@Getter
@Setter
@EqualsAndHashCode
public class StudentExcel {

    private String name;

    private String num;

    private String sex;


}

步骤三：监听器

package marchsoft.modules.tiku.listener;

import com.alibaba.excel.context.AnalysisContext;
import com.alibaba.excel.read.listener.ReadListener;
import com.alibaba.excel.util.ListUtils;
import com.alibaba.fastjson.JSON;
import lombok.extern.slf4j.Slf4j;
import marchsoft.modules.tiku.entity.Dto.student.StudentExcel;

import java.util.List;

/**
 * Description：
 *
 * @author mfei
 * Date: 2022/4/14 10:07
 **/
@Slf4j
public class StudentDataListener implements ReadListener<StudentExcel> {
    /**
     * 每隔5条存储数据库，实际使用中可以100条，然后清理list ，方便内存回收
     */
    private static final int BATCH_COUNT = 2;
    /**
     * 缓存的数据
     */
    private List<StudentExcel> cachedDataList = ListUtils.newArrayListWithExpectedSize(BATCH_COUNT);
    /**
     * 假设这个是一个DAO，当然有业务逻辑这个也可以是一个service。当然如果不用存储这个对象没用。
     */
    private StudentExcel studentExcel;

    /**
     * 如果使用了spring,请使用这个构造方法。每次创建Listener的时候需要把spring管理的类传进来
     *
     * @param studentExcel
     */
//    public StudentDataListener(StudentExcel studentExcel) {
//        this.studentExcel = studentExcel;
//    }

    /**
     * 这个每一条数据解析都会来调用
     *
     * @param data    one row value. Is is same as {@link AnalysisContext#readRowHolder()}
     * @param context
     */
    @Override
    public void invoke(StudentExcel data, AnalysisContext context) {
        log.info("解析到一条数据:{}", JSON.toJSONString(data));
        cachedDataList.add(data);
        // 达到BATCH_COUNT了，需要去存储一次数据库，防止数据几万条数据在内存，容易OOM
        if (cachedDataList.size() >= BATCH_COUNT) {
            saveData();
            // 存储完成清理 list
            cachedDataList = ListUtils.newArrayListWithExpectedSize(BATCH_COUNT);
        }
    }

    /**
     * 所有数据解析完成了 都会来调用
     *
     * @param context
     */
    @Override
    public void doAfterAllAnalysed(AnalysisContext context) {
        // 这里也要保存数据，确保最后遗留的数据也存储到数据库
        saveData();
        log.info("所有数据解析完成！");
    }

    /**
     * 加上存储数据库
     */
    private void saveData() {
        log.info("{}条数据，开始存储数据库！", cachedDataList.size());
        System.out.println(cachedDataList.toString());
        log.info("存储数据库成功！");
    }

}

步骤四：上传调用

@AnonymousPostMapping("/upload")
@ResponseBody
public Result<Object> importLoad(@RequestPart MultipartFile file) throws Exception {
    // easyExcel解析
    EasyExcel.read(file.getInputStream(), StudentExcel.class,new StudentDataListener()).sheet().doRead();

    return Result.success();
}