From 8c5c5ce2938bd4f88f12722abe740570d9ed165e Mon Sep 17 00:00:00 2001
From: mitao <2763622819@qq.com>
Date: 星期五, 13 十二月 2024 18:03:19 +0800
Subject: [PATCH] 暂存间管理接口

---
 medicalWaste-system/src/main/java/com/sinata/system/service/impl/MwStagingRoomServiceImpl.java |  114 +++++++++++++++++++++++++++++++++++++++++++++++++++++----
 1 files changed, 106 insertions(+), 8 deletions(-)

diff --git a/medicalWaste-system/src/main/java/com/sinata/system/service/impl/MwStagingRoomServiceImpl.java b/medicalWaste-system/src/main/java/com/sinata/system/service/impl/MwStagingRoomServiceImpl.java
index f687b10..9676585 100644
--- a/medicalWaste-system/src/main/java/com/sinata/system/service/impl/MwStagingRoomServiceImpl.java
+++ b/medicalWaste-system/src/main/java/com/sinata/system/service/impl/MwStagingRoomServiceImpl.java
@@ -1,5 +1,6 @@
 package com.sinata.system.service.impl;
 
+import com.alibaba.excel.EasyExcel;
 import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
 import com.sinata.common.entity.PageDTO;
@@ -8,16 +9,22 @@
 import com.sinata.system.domain.MwStagingRoom;
 import com.sinata.system.domain.SysDepartment;
 import com.sinata.system.domain.dto.MwStagingRoomDTO;
+import com.sinata.system.domain.query.CheckoutRecordQuery;
 import com.sinata.system.domain.query.MwStagingRoomQuery;
 import com.sinata.system.domain.query.StorageRecordQuery;
-import com.sinata.system.domain.vo.MwCollectRecordVO;
+import com.sinata.system.domain.vo.MwCheckoutRecordVO;
 import com.sinata.system.domain.vo.MwStagingRoomVO;
+import com.sinata.system.domain.vo.MwStorageRecordVO;
 import com.sinata.system.mapper.MwStagingRoomMapper;
 import com.sinata.system.service.MwStagingRoomService;
 import com.sinata.system.service.SysDepartmentService;
 import lombok.RequiredArgsConstructor;
 import org.springframework.stereotype.Service;
 
+import javax.servlet.http.HttpServletResponse;
+import java.io.IOException;
+import java.net.URLEncoder;
+import java.util.List;
 import java.util.Objects;
 
 /**
@@ -42,14 +49,16 @@
     @Override
     public PageDTO<MwStagingRoomVO> pageList(MwStagingRoomQuery query) {
         String treeCode = "";
-        if (Objects.nonNull(query.getDepartmentId())) {
-            SysDepartment department = sysDepartmentService.getById(query.getDepartmentId());
-            if (Objects.isNull(department)) {
+        if (Objects.isNull(query.getDepartmentId())) {
+            SysDepartment myDepartment = sysDepartmentService.getMyDepartment();
+            if (Objects.isNull(myDepartment)) {
                 return PageDTO.empty(0L, 0L);
             }
-            treeCode = department.getTreeCode();
+            treeCode = myDepartment.getTreeCode();
+        } else if (Objects.nonNull(query.getDepartmentId())) {
+            treeCode = sysDepartmentService.getById(query.getDepartmentId()).getTreeCode();
         }
-        Page<MwStagingRoomVO> page = baseMapper.pagelist(new Page<>(query.getPageCurr(), query.getPageSize()), query.getDepartmentId(), treeCode);
+        Page<MwStagingRoomVO> page = baseMapper.pagelist(new Page<>(query.getPageCurr(), query.getPageSize()), treeCode);
         return PageDTO.of(page);
     }
 
@@ -101,7 +110,96 @@
      * @return
      */
     @Override
-    public PageDTO<MwCollectRecordVO> storageRecord(StorageRecordQuery query) {
-        return null;
+    public PageDTO<MwStorageRecordVO> storageRecord(StorageRecordQuery query) {
+        String treeCode = "";
+        if (Objects.isNull(query.getDepartmentId()) && Objects.isNull(query.getStagingRoomId())) {
+            SysDepartment myDepartment = sysDepartmentService.getMyDepartment();
+            if (Objects.isNull(myDepartment)) {
+                return PageDTO.empty(0L, 0L);
+            }
+            treeCode = myDepartment.getTreeCode();
+        } else if (Objects.nonNull(query.getDepartmentId()) && Objects.isNull(query.getStagingRoomId())) {
+            treeCode = sysDepartmentService.getById(query.getDepartmentId()).getTreeCode();
+        }
+        Page<MwStorageRecordVO> page = baseMapper.storageRecordPage(new Page<>(query.getPageCurr(), query.getPageSize()), query, treeCode);
+        return PageDTO.of(page);
+    }
+
+    /**
+     * 暂存间入库记录导出
+     *
+     * @param query
+     * @param response
+     */
+    @Override
+    public void storageRecordExport(StorageRecordQuery query, HttpServletResponse response) throws IOException {
+        String treeCode = "";
+        if (Objects.isNull(query.getDepartmentId()) && Objects.isNull(query.getStagingRoomId())) {
+            SysDepartment myDepartment = sysDepartmentService.getMyDepartment();
+            if (Objects.isNull(myDepartment)) {
+                return;
+            }
+            treeCode = myDepartment.getTreeCode();
+        } else if (Objects.nonNull(query.getDepartmentId()) && Objects.isNull(query.getStagingRoomId())) {
+            treeCode = sysDepartmentService.getById(query.getDepartmentId()).getTreeCode();
+        }
+        List<MwStorageRecordVO> list = baseMapper.storageRecordList(query, treeCode);
+        // 这里注意 有同学反应使用swagger 会导致各种问题,请直接用浏览器或者用postman
+        response.setContentType("application/vnd.openxmlformats-officedocument.spreadsheetml.sheet");
+        response.setCharacterEncoding("utf-8");
+        // 这里URLEncoder.encode可以防止中文乱码 当然和easyexcel没有关系
+        String fileName = URLEncoder.encode("暂存间入库记录", "UTF-8").replaceAll("\\+", "%20");
+        response.setHeader("Content-disposition", "attachment;filename*=utf-8''" + fileName + ".xlsx");
+        EasyExcel.write(response.getOutputStream(), MwStorageRecordVO.class).sheet("暂存间入库记录").doWrite(list);
+    }
+
+    /**
+     * 暂存间出库记录
+     *
+     * @param query
+     * @return
+     */
+    @Override
+    public PageDTO<MwCheckoutRecordVO> checkoutRecord(CheckoutRecordQuery query) {
+        String treeCode = "";
+        if (Objects.isNull(query.getDepartmentId()) && Objects.isNull(query.getStagingRoomId())) {
+            SysDepartment myDepartment = sysDepartmentService.getMyDepartment();
+            if (Objects.isNull(myDepartment)) {
+                return PageDTO.empty(0L, 0L);
+            }
+            treeCode = myDepartment.getTreeCode();
+        } else if (Objects.nonNull(query.getDepartmentId()) && Objects.isNull(query.getStagingRoomId())) {
+            treeCode = sysDepartmentService.getById(query.getDepartmentId()).getTreeCode();
+        }
+        Page<MwCheckoutRecordVO> page = baseMapper.checkoutRecordPage(new Page<>(query.getPageCurr(), query.getPageSize()), query, treeCode);
+        return PageDTO.of(page);
+    }
+
+    /**
+     * 暂存间出库记录导出
+     *
+     * @param query
+     * @param response
+     */
+    @Override
+    public void checkoutRecordExport(CheckoutRecordQuery query, HttpServletResponse response) throws IOException {
+        String treeCode = "";
+        if (Objects.isNull(query.getDepartmentId()) && Objects.isNull(query.getStagingRoomId())) {
+            SysDepartment myDepartment = sysDepartmentService.getMyDepartment();
+            if (Objects.isNull(myDepartment)) {
+                return;
+            }
+            treeCode = myDepartment.getTreeCode();
+        } else if (Objects.nonNull(query.getDepartmentId()) && Objects.isNull(query.getStagingRoomId())) {
+            treeCode = sysDepartmentService.getById(query.getDepartmentId()).getTreeCode();
+        }
+        List<MwCheckoutRecordVO> list = baseMapper.checkoutRecordList(query, treeCode);
+        // 这里注意 有同学反应使用swagger 会导致各种问题,请直接用浏览器或者用postman
+        response.setContentType("application/vnd.openxmlformats-officedocument.spreadsheetml.sheet");
+        response.setCharacterEncoding("utf-8");
+        // 这里URLEncoder.encode可以防止中文乱码 当然和easyexcel没有关系
+        String fileName = URLEncoder.encode("暂存间出库记录", "UTF-8").replaceAll("\\+", "%20");
+        response.setHeader("Content-disposition", "attachment;filename*=utf-8''" + fileName + ".xlsx");
+        EasyExcel.write(response.getOutputStream(), MwCheckoutRecordVO.class).sheet("暂存间出库记录").doWrite(list);
     }
 }

--
Gitblit v1.7.1