From 0a68f96c9079645ac8b6f3543f0f5e85e854422b Mon Sep 17 00:00:00 2001
From: Amjacks <1932302177@qq.com>
Date: Fri, 21 Mar 2025 08:48:51 +0800
Subject: [PATCH] =?UTF-8?q?feat=20=E5=BA=93=E4=BD=8D=E7=BB=B4=E6=8A=A4?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../cim/idm/controller/StorageController.java | 79 +++++++++++++++++--
 1 file changed, 74 insertions(+), 5 deletions(-)

diff --git a/zi-wms-pda/src/main/java/com/cim/idm/controller/StorageController.java b/zi-wms-pda/src/main/java/com/cim/idm/controller/StorageController.java
index a0d1779..c73a9de 100644
--- a/zi-wms-pda/src/main/java/com/cim/idm/controller/StorageController.java
+++ b/zi-wms-pda/src/main/java/com/cim/idm/controller/StorageController.java
@@ -7,7 +7,11 @@ import com.cim.idm.framework.IDMFrameServiceProxy;
 import com.cim.idm.model.BSERPlocation;
 import com.cim.idm.model.CheckPlan1;
 import com.cim.idm.model.StorageSpec;
+import com.cim.idm.response.BaseResponse;
+import com.cim.idm.response.RespGenerator;
 import com.cim.idm.utils.AjaxResult;
+import com.cim.idm.wmspackage.storage.StorageServiceProxy;
+import com.cim.idm.wmspackage.storage.management.data.StorageSpecKey;
 import org.springframework.boot.autoconfigure.EnableAutoConfiguration;
 import org.springframework.web.bind.annotation.RequestBody;
 import org.springframework.web.bind.annotation.RequestMapping;
@@ -28,10 +32,8 @@ import java.util.Objects;
 @EnableAutoConfiguration
 @RequestMapping("/api/storage")
 public class StorageController {
-	
-	
-	
-	
+
+
 	@RequestMapping(value = "/getORG", method = RequestMethod.POST)
 	public AjaxResult getORG(@RequestBody JSONObject in ){
 
@@ -225,5 +227,72 @@ public class StorageController {
 		List<Map<String,Object>> list = IDMFrameServiceProxy.getSqlTemplate().queryForList(sql, bindMap);
 		return AjaxResult.me().setResultObj(list);
 	}
-	
+
+	@RequestMapping(value = "/addStorage", method = RequestMethod.POST)
+	public BaseResponse<Object> addStorage(@RequestBody Map<String, Object> param) {
+		// 获取参数
+		String storageName = (String) param.get("STORAGENAME");
+		String description = (String) param.get("DESCRIPTION");
+		String siteName = (String) param.get("SITENAME");
+		String erpFactory = (String) param.get("ERPFACTORY");
+		// 异常
+		StorageSpecKey storageSpecKey = new StorageSpecKey();
+		storageSpecKey.setStorageName(storageName);
+		storageSpecKey.setSiteName(siteName);
+		com.cim.idm.wmspackage.storage.management.data.StorageSpec storageSpec = null;
+		try {
+			storageSpec = StorageServiceProxy.getStorageSpecService().selectByKey(storageSpecKey);
+			if (Objects.nonNull(storageSpec)) {
+				return RespGenerator.returnError("已存在该货位！");
+			}
+		} catch (Exception ignored) {
+
+		}
+		storageSpec = new com.cim.idm.wmspackage.storage.management.data.StorageSpec();
+		// 构建数据
+		storageSpec.setKey(storageSpecKey);
+		storageSpec.setDescription(description);
+		storageSpec.setErpFactory(erpFactory);
+		// 插入数据
+		StorageServiceProxy.getStorageSpecService().insert(storageSpec);
+		return RespGenerator.returnOK(param);
+	}
+
+	@RequestMapping(value = "/editStorage", method = RequestMethod.POST)
+	public BaseResponse<Object> editStorage(@RequestBody Map<String, Object> param) {
+		// 获取参数
+		String storageName = (String) param.get("STORAGENAME");
+		String siteName = (String) param.get("SITENAME");
+		String description = (String) param.get("DESCRIPTION");
+		// 获取数据
+		StorageSpecKey storageSpecKey = new StorageSpecKey();
+		storageSpecKey.setStorageName(storageName);
+		storageSpecKey.setSiteName(siteName);
+		com.cim.idm.wmspackage.storage.management.data.StorageSpec storageSpec = null;
+		try {
+			storageSpec = StorageServiceProxy.getStorageSpecService().selectByKey(storageSpecKey);
+		} catch (Exception e) {
+			return RespGenerator.returnError("未找到该货位！");
+		}
+		// 构建数据
+		storageSpec.setDescription(description);
+		// 更新数据
+		StorageServiceProxy.getStorageSpecService().update(storageSpec);
+		return RespGenerator.returnOK(param);
+
+	}
+
+	@RequestMapping(value = "/delStorage", method = RequestMethod.POST)
+	public BaseResponse<Map<String, Object>> delStorage(@RequestBody Map<String, Object> param) {
+		// 获取参数
+		String storageName = (String) param.get("STORAGENAME");
+		String siteName = (String) param.get("SITENAME");
+		// 组建数据
+		StorageSpecKey storageSpec = new StorageSpecKey();
+		storageSpec.setStorageName(storageName);
+		storageSpec.setSiteName(siteName);
+		// 删除数据
+		StorageServiceProxy.getStorageSpecService().delete(storageSpec);
+		return RespGenerator.returnOK(param);
+	}
 }