From 1c3c68d3dc45e4d519088de17001284858de5eb9 Mon Sep 17 00:00:00 2001
From: luxiaotao1123 <t1341870251@163.com>
Date: 星期六, 06 六月 2020 10:50:02 +0800
Subject: [PATCH] #导出

---
 src/main/java/com/zy/asrs/controller/MatCodeController.java |   41 +++++++++++++++++++++++++++++------------
 1 files changed, 29 insertions(+), 12 deletions(-)

diff --git a/src/main/java/com/zy/asrs/controller/MatCodeController.java b/src/main/java/com/zy/asrs/controller/MatCodeController.java
index e5fd20f..b36c779 100644
--- a/src/main/java/com/zy/asrs/controller/MatCodeController.java
+++ b/src/main/java/com/zy/asrs/controller/MatCodeController.java
@@ -13,14 +13,16 @@
 import com.core.common.R;
 import com.zy.asrs.entity.MatCode;
 import com.zy.asrs.service.MatCodeService;
-import com.zy.common.utils.excel.EnInitData;
-import com.zy.common.utils.excel.EnInitDataListener;
+import com.zy.common.utils.excel.matcode.MatCodeExcel;
+import com.zy.common.utils.excel.matcode.MatCodeExcelListener;
 import com.zy.common.web.BaseController;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.web.bind.annotation.*;
 import org.springframework.web.multipart.MultipartFile;
 
+import javax.servlet.http.HttpServletResponse;
 import java.io.IOException;
+import java.net.URLEncoder;
 import java.util.*;
 
 @RestController
@@ -99,21 +101,36 @@
 
     // 瀵煎嚭
     @RequestMapping(value = "/matCode/export/auth")
-    @ManagerAuth
-    public R export(@RequestBody JSONObject param){
-        List<String> fields = JSONObject.parseArray(param.getJSONArray("fields").toJSONString(), String.class);
-        EntityWrapper<MatCode> wrapper = new EntityWrapper<>();
-        Map<String, Object> map = excludeTrash(param.getJSONObject("matCode"));
-        convert(map, wrapper);
-        List<MatCode> list = matCodeService.selectList(wrapper);
-        return R.ok(exportSupport(list, fields));
+    public void export(@RequestParam(required = false) String fileName,
+                       @RequestParam(required = false) Integer rowCount,
+                       HttpServletResponse response) throws Exception {
+        List<MatCode> list = matCodeService.selectList(new EntityWrapper<>());
+        List<MatCodeExcel> excels = new ArrayList<>();
+        for (MatCode matCode : list) {
+            MatCodeExcel excel = new MatCodeExcel();
+            excel.setMatNo(matCode.getMatNo());
+            excel.setBarcode(matCode.getBarcode());
+            excel.setMatName(matCode.getMatName());
+            excel.setStr1(matCode.getStr1());
+            excel.setStr2(matCode.getStr2());
+            excels.add(excel);
+            if (rowCount != null && excels.size() >= rowCount){
+                break;
+            }
+        }
+        response.setContentType("application/vnd.ms-excel");
+        response.setCharacterEncoding("utf-8");
+        fileName = URLEncoder.encode(Cools.isEmpty(fileName)?"鐗╂枡缂栫爜":fileName, "UTF-8");
+        response.setHeader("Content-disposition", "attachment;filename=" + fileName + ".xlsx");
+        EasyExcel.write(response.getOutputStream(), MatCodeExcel.class).sheet("琛�1").doWrite(excels);
     }
 
     // 瀵煎叆
     @RequestMapping(value = "/matCode/import/auth")
     @ManagerAuth
-    public R matCodeImport(MultipartFile file) throws IOException {
-        EasyExcel.read(file.getInputStream(), EnInitData.class, new EnInitDataListener()).sheet().doRead();
+    public R matCodeImport(MultipartFile file) throws IOException, InterruptedException {
+        Thread.sleep(2000);
+        EasyExcel.read(file.getInputStream(), MatCodeExcel.class, new MatCodeExcelListener()).sheet().doRead();
         return R.ok();
     }
 

--
Gitblit v1.9.1