优化

mes/hd/nladmin-system/src/main/java/org/nl/wms/pa/rest/PerformancemstController.java

@@ -16,9 +16,13 @@ import org.springframework.security.access.prepost.PreAuthorize;
 import org.springframework.validation.annotation.Validated;
 import org.springframework.web.bind.annotation.*;
 import io.swagger.annotations.*;
+
+import java.io.IOException;
 import java.util.Map;
 import lombok.extern.slf4j.Slf4j;
 
+import javax.servlet.http.HttpServletResponse;
+
 /**
 * @author Liuxy
 * @date 2021-12-28
@@ -48,6 +52,12 @@ public class PerformancemstController {
         return new ResponseEntity<>(performancemstService.workLoadStatQuery(whereJson,page),HttpStatus.OK);
     }
 
+    @ApiOperation("导出数据")
+    @GetMapping(value = "/workloadstat/download")
+    public void download(HttpServletResponse response, @RequestParam Map whereJson) throws IOException {
+        performancemstService.download(whereJson, response);
+    }
+
     @PostMapping
     @Log("新增工作量汇报")
     @ApiOperation("新增工作量汇报")

mes/hd/nladmin-system/src/main/java/org/nl/wms/pa/service/PerformancemstService.java

@@ -94,4 +94,11 @@ public interface PerformancemstService {
      * 获取工作量统计表表头
      */
     JSONArray getHeader();
+
+    /**
+     * 导出数据
+     *
+     * @param whereJson /
+     */
+    void download(Map whereJson, HttpServletResponse response) throws IOException;
 }

mes/hd/nladmin-system/src/main/java/org/nl/wms/pa/service/impl/PerformancemstServiceImpl.java

@@ -16,6 +16,7 @@ import lombok.extern.slf4j.Slf4j;
 import org.nl.exception.BadRequestException;
 import org.nl.modules.security.service.dto.JwtUserDto;
 import org.nl.modules.system.util.CodeUtil;
+import org.nl.utils.FileUtil;
 import org.nl.utils.SecurityUtils;
 import org.nl.wms.pa.Enum.TestStatusEnum;
 import org.nl.wms.pa.service.PerformancemstService;
@@ -27,7 +28,10 @@ import org.springframework.data.domain.Pageable;
 import org.springframework.stereotype.Service;
 import org.springframework.transaction.annotation.Transactional;
 
+import javax.servlet.http.HttpServletResponse;
+import java.io.IOException;
 import java.math.BigDecimal;
+import java.util.ArrayList;
 import java.util.HashMap;
 import java.util.List;
 import java.util.Map;
@@ -317,4 +321,11 @@ public class PerformancemstServiceImpl implements PerformancemstService {
         return jsonArrResult;
     }
 
+    @Override
+    @Transactional(rollbackFor = Exception.class)
+    public void download(Map whereJson, HttpServletResponse response) throws IOException {
+        List<Map<String, Object>> list = new ArrayList<>();
+        FileUtil.downloadExcel(list, response);
+    }
+
 }

mes/hd/nladmin-system/src/main/java/org/nl/wms/statistics/rest/PowderOrderQueryController.java

@@ -14,6 +14,8 @@ import org.springframework.web.bind.annotation.RequestMapping;
 import org.springframework.web.bind.annotation.RequestParam;
 import org.springframework.web.bind.annotation.RestController;
 
+import javax.servlet.http.HttpServletResponse;
+import java.io.IOException;
 import java.util.Map;
 
 /**
@@ -41,4 +43,10 @@ public class PowderOrderQueryController {
     public ResponseEntity<Object> getHeader() {
         return new ResponseEntity<>(powderOrderQueryService.getHeader(), HttpStatus.OK);
     }
+
+    @ApiOperation("导出数据")
+    @GetMapping(value = "/download")
+    public void download(HttpServletResponse response, @RequestParam Map whereJson) throws IOException {
+        powderOrderQueryService.download(whereJson, response);
+    }
 }

mes/hd/nladmin-system/src/main/java/org/nl/wms/statistics/service/PowderOrderQueryService.java

@@ -3,6 +3,8 @@ package org.nl.wms.statistics.service;
 import com.alibaba.fastjson.JSONArray;
 import org.springframework.data.domain.Pageable;
 
+import javax.servlet.http.HttpServletResponse;
+import java.io.IOException;
 import java.util.Map;
 
 public interface PowderOrderQueryService {
@@ -19,4 +21,11 @@ public interface PowderOrderQueryService {
      * 获取表头
      */
     JSONArray getHeader();
+
+    /**
+     * 导出数据
+     *
+     * @param whereJson /
+     */
+    void download(Map whereJson, HttpServletResponse response) throws IOException;
 }

mes/hd/nladmin-system/src/main/java/org/nl/wms/statistics/service/impl/PowderOrderQueryServiceImpl.java

@@ -1,12 +1,16 @@
 package org.nl.wms.statistics.service.impl;
 
+import cn.hutool.core.date.DateUnit;
+import cn.hutool.core.date.DateUtil;
 import cn.hutool.core.map.MapUtil;
 import cn.hutool.core.util.NumberUtil;
 import cn.hutool.core.util.ObjectUtil;
+import cn.hutool.core.util.StrUtil;
 import lombok.RequiredArgsConstructor;
 import lombok.extern.slf4j.Slf4j;
 import com.alibaba.fastjson.JSONArray;
 import com.alibaba.fastjson.JSONObject;
+import org.nl.utils.FileUtil;
 import org.nl.wms.basedata.master.constant.MaterOptTypeEnum;
 import org.nl.wms.basedata.master.service.ClassstandardService;
 import org.nl.wms.statistics.service.PowderOrderQueryService;
@@ -15,9 +19,11 @@ import org.nl.wql.core.bean.WQLObject;
 import org.nl.wql.util.WqlUtil;
 import org.springframework.data.domain.Pageable;
 import org.springframework.stereotype.Service;
+import org.springframework.transaction.annotation.Transactional;
 
-import java.util.HashMap;
-import java.util.Map;
+import javax.servlet.http.HttpServletResponse;
+import java.io.IOException;
+import java.util.*;
 
 @Service
 @RequiredArgsConstructor
@@ -112,10 +118,6 @@ public class PowderOrderQueryServiceImpl implements PowderOrderQueryService {
             jsonResult4.put("prop",json.getString("material_id"));
             jsonResult4.put("label","物料" + NumberUtil.add(String.valueOf(i), "1") + "");
             jsonResultArr.add(jsonResult4);
-//            JSONObject jsonResult5 = new JSONObject();
-//            jsonResult5.put("prop",json.getString("material_code")+"");
-//            jsonResult5.put("label","物料编码");
-//            jsonResultArr.add(jsonResult5);
             JSONObject jsonResult6 = new JSONObject();
             jsonResult6.put("prop",json.getString("material_id")+"pcsn");
             jsonResult6.put("label","批次");
@@ -149,4 +151,11 @@ public class PowderOrderQueryServiceImpl implements PowderOrderQueryService {
 
         return jsonResultArr;
     }
+
+    @Override
+    @Transactional(rollbackFor = Exception.class)
+    public void download(Map whereJson, HttpServletResponse response) throws IOException {
+        List<Map<String, Object>> list = new ArrayList<>();
+        FileUtil.downloadExcel(list, response);
+    }
 }