From 7e10622ead5bd8b3ce0c9fda39d22ae0e938e030 Mon Sep 17 00:00:00 2001
From: zwl <1051256694@qq.com>
Date: 星期五, 12 九月 2025 13:44:10 +0800
Subject: [PATCH] 完善订单组托可用数据显示功能

---
 zy-asrs-wms/src/main/java/com/zy/asrs/wms/controller/PriorController.java |  104 ++++++++++++++++++++++++++++++++++++++++++++++++++--
 1 files changed, 100 insertions(+), 4 deletions(-)

diff --git a/zy-asrs-wms/src/main/java/com/zy/asrs/wms/controller/PriorController.java b/zy-asrs-wms/src/main/java/com/zy/asrs/wms/controller/PriorController.java
index d86e811..960e9be 100644
--- a/zy-asrs-wms/src/main/java/com/zy/asrs/wms/controller/PriorController.java
+++ b/zy-asrs-wms/src/main/java/com/zy/asrs/wms/controller/PriorController.java
@@ -1,20 +1,36 @@
 package com.zy.asrs.wms.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.core.conditions.query.LambdaQueryWrapper;
+import com.baomidou.mybatisplus.core.metadata.IPage;
 import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
+import com.zy.asrs.common.domain.dto.SafeStoDo;
+import com.zy.asrs.common.domain.dto.SafeStoDto;
+import com.zy.asrs.common.domain.entity.LocInPrintMatExcel;
+import com.zy.asrs.common.domain.entity.MatExcel;
+import com.zy.asrs.common.domain.entity.SafeStoExcel;
+import com.zy.asrs.common.utils.MatExcelListener;
+import com.zy.asrs.common.utils.SafeStoExcelListener;
+import com.zy.asrs.common.wms.entity.Mat;
 import com.zy.asrs.common.wms.entity.Prior;
+import com.zy.asrs.common.wms.service.MatService;
 import com.zy.asrs.common.wms.service.PriorService;
 import com.zy.asrs.framework.annotations.ManagerAuth;
-import com.zy.asrs.framework.common.Cools;
-import com.zy.asrs.framework.common.R;
+import com.zy.asrs.framework.common.*;
 import com.zy.asrs.framework.domain.KeyValueVo;
-import com.zy.asrs.framework.common.DateUtils;
 import com.zy.asrs.common.web.BaseController;
+import com.zy.asrs.framework.exception.CoolException;
 import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.transaction.annotation.Transactional;
 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
@@ -22,6 +38,10 @@
 
     @Autowired
     private PriorService priorService;
+    @Autowired
+    private MatService matService;
+    @Autowired
+    private SnowflakeIdWorker snowflakeIdWorker;
 
     @RequestMapping(value = "/prior/{id}/auth")
     @ManagerAuth
@@ -115,7 +135,83 @@
         if (hostId != null) {
             param.put("host_id", hostId);
         }
-        return R.ok(priorService.getSafeQtyPage(curr, limit, param));
+        IPage<SafeStoDo> safeQtyPage = priorService.getSafeQtyPage(curr, limit, param);
+        return R.ok(safeQtyPage);
     }
 
+    @RequestMapping(value = "/safeSto/add/auth")
+    @ManagerAuth(memo = "娣诲姞瀹夊叏搴撳瓨")
+    public R del(@RequestBody SafeStoDto dto){
+        if (Cools.isEmpty(dto.getMatId(), dto.getSafeQua())) {
+            return R.parse(BaseRes.PARAM);
+        }
+        if (dto.getSafeQua() <= 0.0D) {
+            return R.error("瀹夊叏搴撳瓨閲忓繀椤诲ぇ浜庨浂");
+        }
+        Mat mat = matService.getOne(new LambdaQueryWrapper<Mat>().eq(Mat::getId, dto.getMatId()).eq(Mat::getHostId, getHostId()));
+        if (mat != null) {
+            Prior prior = new Prior();
+            prior.setHostId(getHostId());
+            prior.setUuid("SP" + snowflakeIdWorker.nextId());
+            prior.setMatId(mat.getId());
+            prior.setMatnr(mat.getMatnr());
+            prior.setMaktx(mat.getMaktx());
+            prior.setLocNo(dto.getLocNo());
+            prior.setSafeQua(dto.getSafeQua());
+            prior.setCreateBy(getUserId());
+            prior.setCreateTime(new Date());
+            prior.setUpdateBy(getUserId());
+            prior.setUpdateTime(new Date());
+            prior.setStatus(1);
+            prior.setMemo(dto.getMemo());
+            if (!priorService.save(prior)) {
+                throw new CoolException("鏁版嵁淇濆瓨澶辫触");
+            }
+        } else {
+            throw new CoolException("鏈嶅姟鍣ㄥ唴閮ㄩ敊璇�");
+        }
+        return R.ok();
+    }
+
+    @RequestMapping(value = "/safeSto/delete/auth")
+    @ManagerAuth(memo = "鍒犻櫎搴撳瓨棰勮")
+    @Transactional
+    public R del(@RequestBody List<SafeStoDto> dtos){
+        if (Cools.isEmpty(dtos)) {
+            return R.parse(BaseRes.PARAM);
+        }
+        for (SafeStoDto dto : dtos) {
+            if (!priorService.remove(new LambdaQueryWrapper<Prior>().eq(Prior::getId, dto.getId()).eq(Prior::getHostId, getHostId()))) {
+                throw new CoolException("鍒犻櫎瀹夊叏搴撳瓨澶辫触锛岃閲嶆柊灏濊瘯");
+            }
+        }
+        return R.ok();
+    }
+
+
+    @PostMapping(value = "/safeSto/excel/import/auth")
+    @ManagerAuth(memo = "瀹夊叏搴撳瓨鏁版嵁瀵煎叆")
+    @Transactional
+    public R matExcelImport(MultipartFile file) throws IOException {
+        SafeStoExcelListener listener = new SafeStoExcelListener(getUserId(), getHostId());
+        EasyExcel.read(file.getInputStream(), SafeStoExcel.class, listener).sheet().doRead();
+        return R.ok("鎴愬姛鍚屾"+listener.getTotal()+"鏉″畨鍏ㄥ簱瀛樻暟鎹�");
+    }
+
+
+    /**
+     * excel瀵煎叆妯℃澘涓嬭浇
+     */
+    @RequestMapping(value = "/safeSto/excel/import/mould")
+    public void matExcelImportMould(HttpServletResponse response) throws IOException {
+        List<SafeStoExcel> excels = new ArrayList<>();
+        response.setContentType("application/vnd.ms-excel");
+        response.setCharacterEncoding("utf-8");
+        String fileName = URLEncoder.encode("瀹夊叏搴撳瓨Excel瀵煎叆妯℃澘", "UTF-8");
+        response.setHeader("Content-disposition", "attachment;filename=" + fileName + ".xlsx");
+        EasyExcel.write(response.getOutputStream(), SafeStoExcel.class)
+                .registerWriteHandler(new LongestMatchColumnWidthStyleStrategy())
+                .sheet("sheet1")
+                .doWrite(excels);
+    }
 }

--
Gitblit v1.9.1