From 4ea00a26f7773e0ad27aeee512f34ff45fe0f18b Mon Sep 17 00:00:00 2001
From: luxiaotao1123 <t1341870251@163.com>
Date: 星期二, 16 六月 2020 15:21:29 +0800
Subject: [PATCH] #

---
 src/main/java/com/zy/asrs/controller/MatCodeController.java |   78 ++++++++++++++++++++++++++++++++-------
 1 files changed, 64 insertions(+), 14 deletions(-)

diff --git a/src/main/java/com/zy/asrs/controller/MatCodeController.java b/src/main/java/com/zy/asrs/controller/MatCodeController.java
index eb80c17..a3669e4 100644
--- a/src/main/java/com/zy/asrs/controller/MatCodeController.java
+++ b/src/main/java/com/zy/asrs/controller/MatCodeController.java
@@ -1,6 +1,7 @@
 package com.zy.asrs.controller;
 
 import com.alibaba.excel.EasyExcel;
+import com.alibaba.excel.write.style.column.LongestMatchColumnWidthStyleStrategy;
 import com.alibaba.fastjson.JSONArray;
 import com.alibaba.fastjson.JSONObject;
 import com.baomidou.mybatisplus.mapper.EntityWrapper;
@@ -12,7 +13,11 @@
 import com.core.common.DateUtils;
 import com.core.common.R;
 import com.zy.asrs.entity.MatCode;
+import com.zy.asrs.entity.MatCodePrint;
 import com.zy.asrs.service.MatCodeService;
+import com.zy.common.CodeRes;
+import com.zy.common.utils.BarcodeUtils;
+import com.zy.common.utils.QrCode;
 import com.zy.common.utils.excel.matcode.MatCodeExcel;
 import com.zy.common.utils.excel.matcode.MatCodeExcelListener;
 import com.zy.common.web.BaseController;
@@ -20,7 +25,9 @@
 import org.springframework.web.bind.annotation.*;
 import org.springframework.web.multipart.MultipartFile;
 
+import javax.imageio.ImageIO;
 import javax.servlet.http.HttpServletResponse;
+import java.awt.image.BufferedImage;
 import java.io.IOException;
 import java.net.URLEncoder;
 import java.util.*;
@@ -101,18 +108,11 @@
 
     // 瀵煎嚭
     @RequestMapping(value = "/matCode/export/auth")
-    @ManagerAuth
-    public void export(@RequestBody JSONObject param, HttpServletResponse response) throws Exception{
-//        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);
-
-
+    @ManagerAuth(value = ManagerAuth.Auth.NONE, memo = "鐗╂枡缂栫爜鏁版嵁瀵煎嚭")
+    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();
@@ -122,23 +122,73 @@
             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");
-        String fileName = URLEncoder.encode("娴嬭瘯", "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("妯℃澘").doWrite(excels);
+        EasyExcel.write(response.getOutputStream(), MatCodeExcel.class)
+                .registerWriteHandler(new LongestMatchColumnWidthStyleStrategy())
+                .sheet("琛�1")
+                .doWrite(excels);
     }
 
     // 瀵煎叆
     @RequestMapping(value = "/matCode/import/auth")
-    @ManagerAuth
+    @ManagerAuth(memo = "鐗╂枡缂栫爜鏁版嵁瀵煎叆")
     public R matCodeImport(MultipartFile file) throws IOException, InterruptedException {
         Thread.sleep(2000);
         EasyExcel.read(file.getInputStream(), MatCodeExcel.class, new MatCodeExcelListener()).sheet().doRead();
         return R.ok();
     }
 
+    // 鎵撳嵃
+    @RequestMapping(value = "/macCode/print/auth")
+    @ManagerAuth(memo = "鐗╂枡缂栫爜鎵撳嵃")
+    public R matCodePrint(@RequestParam(value = "param[]") String[] param) {
+        if(Cools.isEmpty(param)) {
+            return R.parse(CodeRes.EMPTY);
+        }
+        List<MatCodePrint> res = new ArrayList<>();
+        for (String matNo : param){
+            MatCode matCode = matCodeService.selectById(matNo);
+            // 鎵撳嵃鏁版嵁娉ㄥ叆
+            MatCodePrint print = new MatCodePrint();
+            print.setMatNo(matCode.getMatNo());
+            print.setBarcode(matCode.getBarcode());
+            print.setMatName(matCode.getMatName());
+            print.setStr1(matCode.getStr1());
+            print.setStr2(matCode.getStr2());
+            res.add(print);
+        }
+        return R.ok().add(res);
+    }
+
+    @RequestMapping(value = "/macCode/code/auth")
+//    @ManagerAuth(memo = "鐗╂枡缂栫爜鏉″舰鐮佽幏鍙�(type:1(鏉″舰鐮�);2(浜岀淮鐮�)")
+    public R matCodeBarcode(@RequestParam(defaultValue = "1") Integer type
+                        , @RequestParam String param
+                        , HttpServletResponse response) throws Exception {
+        if (Cools.isEmpty(param)){
+            return R.parse(BaseRes.EMPTY);
+        }
+        BufferedImage img;
+        if (type == 1) {
+            img = BarcodeUtils.encode(param);
+        } else {
+            img = QrCode.createImg(param);
+        }
+        if (!ImageIO.write(img, "jpg", response.getOutputStream())) {
+            throw new IOException("Could not write an image of format jpg");
+        }
+        response.getOutputStream().flush();
+        response.getOutputStream().close();
+        return R.ok();
+    }
+
     @RequestMapping(value = "/matCodeQuery/auth")
     @ManagerAuth
     public R query(String condition) {

--
Gitblit v1.9.1