();
- /**
- * 假设这个是一个DAO，当然有业务逻辑这个也可以是一个service。当然如果不用存储这个对象没用。
- */
- private UploadDAO uploadDAO;
-
- public UploadDataListener() {
- // 这里是demo，所以随便new一个。实际使用如果到了spring,请使用下面的有参构造函数
- uploadDAO = new UploadDAO();
- }
-
- /**
- * 如果使用了spring,请使用这个构造方法。每次创建Listener的时候需要把spring管理的类传进来
- *
- * @param uploadDAO
- */
- public UploadDataListener(UploadDAO uploadDAO) {
- this.uploadDAO = uploadDAO;
- }
-
- /**
- * 这个每一条数据解析都会来调用
- *
- * @param data
- * one row value. Is is same as {@link AnalysisContext#readRowHolder()}
- * @param context
- */
- @Override
- public void invoke(UploadData data, AnalysisContext context) {
- LOGGER.info("解析到一条数据:{}", JSON.toJSONString(data));
- list.add(data);
- // 达到BATCH_COUNT了，需要去存储一次数据库，防止数据几万条数据在内存，容易OOM
- if (list.size() >= BATCH_COUNT) {
- saveData();
- // 存储完成清理 list
- list.clear();
- }
- }
-
- /**
- * 所有数据解析完成了 都会来调用
- *
- * @param context
- */
- @Override
- public void doAfterAllAnalysed(AnalysisContext context) {
- // 这里也要保存数据，确保最后遗留的数据也存储到数据库
- saveData();
- LOGGER.info("所有数据解析完成！");
- }
-
- /**
- * 加上存储数据库
- */
- private void saveData() {
- LOGGER.info("{}条数据，开始存储数据库！", list.size());
- uploadDAO.save(list);
- LOGGER.info("存储数据库成功！");
- }
-}
diff --git a/src/test/java/com/taover/easyexcel/test/demo/web/WebTest.java b/src/test/java/com/taover/easyexcel/test/demo/web/WebTest.java
deleted file mode 100644
index ffe633f..0000000
--- a/src/test/java/com/taover/easyexcel/test/demo/web/WebTest.java
+++ /dev/null
@@ -1,110 +0,0 @@
-package com.taover.easyexcel.test.demo.web;
-
-import java.io.IOException;
-import java.net.URLEncoder;
-import java.util.ArrayList;
-import java.util.Date;
-import java.util.HashMap;
-import java.util.List;
-import java.util.Map;
-
-import javax.servlet.http.HttpServletResponse;
-
-import org.springframework.beans.factory.annotation.Autowired;
-import org.springframework.stereotype.Controller;
-import org.springframework.web.bind.annotation.GetMapping;
-import org.springframework.web.bind.annotation.PostMapping;
-import org.springframework.web.bind.annotation.ResponseBody;
-import org.springframework.web.multipart.MultipartFile;
-
-import com.alibaba.fastjson.JSON;
-import com.taover.easyexcel.EasyExcel;
-
-/**
- * web读写案例
- *
- * @author Jiaju Zhuang
- **/
-@Controller
-public class WebTest {
-
- @Autowired
- private UploadDAO uploadDAO;
-
- /**
- * 文件下载（失败了会返回一个有部分数据的Excel）
- *
- * 1. 创建excel对应的实体对象 参照{@link DownloadData}
- *
- * 2. 设置返回的 参数
- *
- * 3. 直接写，这里注意，finish的时候会自动关闭OutputStream,当然你外面再关闭流问题不大
- */
- @GetMapping("download")
- public void download(HttpServletResponse response) throws IOException {
- // 这里注意 有同学反应使用swagger 会导致各种问题，请直接用浏览器或者用postman
- response.setContentType("application/vnd.ms-excel");
- response.setCharacterEncoding("utf-8");
- // 这里URLEncoder.encode可以防止中文乱码 当然和easyexcel没有关系
- String fileName = URLEncoder.encode("测试", "UTF-8");
- response.setHeader("Content-disposition", "attachment;filename=" + fileName + ".xlsx");
- EasyExcel.write(response.getOutputStream(), DownloadData.class).sheet("模板").doWrite(data());
- }
-
- /**
- * 文件下载并且失败的时候返回json（默认失败了会返回一个有部分数据的Excel）
- *
- * @since 2.1.1
- */
- @GetMapping("downloadFailedUsingJson")
- public void downloadFailedUsingJson(HttpServletResponse response) throws IOException {
- // 这里注意 有同学反应使用swagger 会导致各种问题，请直接用浏览器或者用postman
- try {
- response.setContentType("application/vnd.ms-excel");
- response.setCharacterEncoding("utf-8");
- // 这里URLEncoder.encode可以防止中文乱码 当然和easyexcel没有关系
- String fileName = URLEncoder.encode("测试", "UTF-8");
- response.setHeader("Content-disposition", "attachment;filename=" + fileName + ".xlsx");
- // 这里需要设置不关闭流
- EasyExcel.write(response.getOutputStream(), DownloadData.class).autoCloseStream(Boolean.FALSE).sheet("模板")
- .doWrite(data());
- } catch (Exception e) {
- // 重置response
- response.reset();
- response.setContentType("application/json");
- response.setCharacterEncoding("utf-8");
- Map map = new HashMap();
- map.put("status", "failure");
- map.put("message", "下载文件失败" + e.getMessage());
- response.getWriter().println(JSON.toJSONString(map));
- }
- }
-
- /**
- * 文件上传
- *
- * 1. 创建excel对应的实体对象 参照{@link UploadData}
- *
- * 2. 由于默认一行行的读取excel，所以需要创建excel一行一行的回调监听器，参照{@link UploadDataListener}
- *
- * 3. 直接读即可
- */
- @PostMapping("upload")
- @ResponseBody
- public String upload(MultipartFile file) throws IOException {
- EasyExcel.read(file.getInputStream(), UploadData.class, new UploadDataListener(uploadDAO)).sheet().doRead();
- return "success";
- }
-
- private List data() {
- List list = new ArrayList();
- for (int i = 0; i < 10; i++) {
- DownloadData data = new DownloadData();
- data.setString("字符串" + 0);
- data.setDate(new Date());
- data.setDoubleData(0.56);
- list.add(data);
- }
- return list;
- }
-}
--
libgit2 0.21.2