From ba7168283fadf8e13761de353c51998fee09bd81 Mon Sep 17 00:00:00 2001
From: value <z1292839451@163.com>
Date: 星期日, 07 四月 2024 00:09:10 +0800
Subject: [PATCH] 新增仓库删除和仓库修改;修复检验保存功能

---
 inspect-server/src/main/java/com/yuanchu/mom/service/impl/WarehouseServiceImpl.java |   35 +++++++++++++++++++++++++++++++++++
 1 files changed, 35 insertions(+), 0 deletions(-)

diff --git a/inspect-server/src/main/java/com/yuanchu/mom/service/impl/WarehouseServiceImpl.java b/inspect-server/src/main/java/com/yuanchu/mom/service/impl/WarehouseServiceImpl.java
index 77b514b..3fb7e19 100644
--- a/inspect-server/src/main/java/com/yuanchu/mom/service/impl/WarehouseServiceImpl.java
+++ b/inspect-server/src/main/java/com/yuanchu/mom/service/impl/WarehouseServiceImpl.java
@@ -1,5 +1,6 @@
 package com.yuanchu.mom.service.impl;
 
+import com.baomidou.mybatisplus.core.toolkit.Wrappers;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
 import com.yuanchu.mom.dto.WarehouseDto;
 import com.yuanchu.mom.mapper.WarehouseCellMapper;
@@ -7,11 +8,15 @@
 import com.yuanchu.mom.mapper.WarehouseMapper;
 import com.yuanchu.mom.mapper.WarehouseShelfMapper;
 import com.yuanchu.mom.pojo.Warehouse;
+import com.yuanchu.mom.pojo.WarehouseCell;
 import com.yuanchu.mom.pojo.WarehouseHistory;
+import com.yuanchu.mom.pojo.WarehouseShelf;
 import com.yuanchu.mom.service.WarehouseService;
 import lombok.AllArgsConstructor;
 import org.springframework.stereotype.Service;
+import org.springframework.transaction.annotation.Transactional;
 
+import java.util.ArrayList;
 import java.util.List;
 
 /**
@@ -43,6 +48,36 @@
     public List<WarehouseDto> selectWarehouse() {
         return warehouseMapper.selectWarehouseList();
     }
+
+    @Override
+    @Transactional(rollbackFor = Exception.class)
+    public int addShelf(WarehouseShelf warehouseShelf) {
+        warehouseShelfMapper.insert(warehouseShelf);
+        List<WarehouseCell> cells = new ArrayList<>();
+        for (Integer i = 0; i < warehouseShelf.getRow(); i++) {
+            for (Integer a = 0; a < warehouseShelf.getCol(); a++) {
+                WarehouseCell cell = new WarehouseCell();
+                cell.setRow(i);
+                cell.setCol(a);
+                cell.setState(1);
+                cells.add(cell);
+            }
+        }
+        warehouseCellMapper.saveBatch(cells);
+        return 1;
+    }
+
+    @Override
+    @Transactional(rollbackFor = Exception.class)
+    public int delWarehouse(Integer id) {
+        warehouseShelfMapper.delete(Wrappers.<WarehouseShelf>lambdaUpdate().eq(WarehouseShelf::getWarehouseId, id));
+        return warehouseMapper.deleteById(id);
+    }
+
+    @Override
+    public int upWarehouse(Warehouse warehouse) {
+        return warehouseMapper.updateById(warehouse);
+    }
 }
 
 

--
Gitblit v1.9.3