From 35b2e532ef54f40586a00698e2795f1c8d787af4 Mon Sep 17 00:00:00 2001
From: luxiaotao1123 <t1341870251@163.com>
Date: 星期二, 26 三月 2024 14:16:06 +0800
Subject: [PATCH] #

---
 zy-asrs-wcs/src/main/java/com/zy/asrs/wcs/core/map/service/MapService.java |   90 ++++++++++++++++++++++++++++++++++++---------
 1 files changed, 72 insertions(+), 18 deletions(-)

diff --git a/zy-asrs-wcs/src/main/java/com/zy/asrs/wcs/core/map/service/MapService.java b/zy-asrs-wcs/src/main/java/com/zy/asrs/wcs/core/map/service/MapService.java
index ee3728c..03965db 100644
--- a/zy-asrs-wcs/src/main/java/com/zy/asrs/wcs/core/map/service/MapService.java
+++ b/zy-asrs-wcs/src/main/java/com/zy/asrs/wcs/core/map/service/MapService.java
@@ -4,14 +4,17 @@
 import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
 import com.zy.asrs.framework.common.Cools;
 import com.zy.asrs.framework.exception.CoolException;
+import com.zy.asrs.wcs.common.domain.enums.DictType;
 import com.zy.asrs.wcs.core.map.controller.param.MapDataParam;
 import com.zy.asrs.wcs.core.map.controller.param.MapQueryParam;
 import com.zy.asrs.wcs.system.entity.Dict;
-import com.zy.asrs.wcs.system.entity.User;
 import com.zy.asrs.wcs.system.service.DictService;
 import com.zy.asrs.wcs.system.service.UserService;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
+
+import java.util.Date;
+import java.util.Optional;
 
 /**
  * Created by vincent on 3/15/2024
@@ -24,29 +27,80 @@
     @Autowired
     private DictService dictService;
 
-    public MapDataParam getMapData(MapQueryParam param, Long userId) {
-
-        String dictFlag = "map-" + param.getFloor();
-        Dict dict = dictService.getOne(new LambdaQueryWrapper<Dict>().eq(Dict::getFlag, dictFlag));
-        if (Cools.isEmpty(dict)) {
-            return null;
-        } else {
-            return JSON.parseObject(dict.getValue(), MapDataParam.class) ;
+    public String getMapFloorList(Long userId) {
+        String floorKey = "map-floor-list";
+        Dict dict = dictService.getOne(new LambdaQueryWrapper<Dict>().eq(Dict::getFlag, floorKey));
+        if (null == dict) {
+            dict = new Dict();
+            dict.setName(floorKey);
+            dict.setFlag(floorKey);
+            dict.setType(DictType.JSON.flag);
+            dict.setValue("[\n" +
+                    "    {\n" +
+                    "        label: '1F',\n" +
+                    "        value: 1\n" +
+                    "    },\n" +
+                    "    {\n" +
+                    "        label: '2F',\n" +
+                    "        value: 2\n" +
+                    "    },\n" +
+                    "    {\n" +
+                    "        label: '3F',\n" +
+                    "        value: 3\n" +
+                    "    },\n" +
+                    "    {\n" +
+                    "        label: '4F',\n" +
+                    "        value: 4\n" +
+                    "    },\n" +
+                    "    {\n" +
+                    "        label: '5F',\n" +
+                    "        value: 5\n" +
+                    "    },\n" +
+                    "]");
+            dict.setCreateBy(userId);
+            dict.setUpdateBy(userId);
+            if (!dictService.save(dict)) {
+                throw new CoolException("鏈嶅姟鍣ㄥ唴閮ㄩ敊璇�");
+            }
         }
+        return Optional.ofNullable(dict).map(Dict::getValue).orElse(null);
+    }
 
-//        User user = userService.getById(userId);
-//        if (Cools.isEmpty(user.getMemo())) {
-//            return new MapDataParam();
-//        }
-//        return JSON.parseObject(user.getMemo(), MapDataParam.class);
+    public String getMapData(MapQueryParam param, Long userId) {
+        String mapKey = getMapKey(param.getFloor());
+        Dict dict = dictService.getOne(new LambdaQueryWrapper<Dict>().eq(Dict::getFlag, mapKey));
+        return Optional.ofNullable(dict).map(Dict::getValue).orElse(null);
     }
 
     public void saveMapData(MapDataParam param, Long userId) {
-        User user = userService.getById(userId);
-        user.setMemo(JSON.toJSONString(param));
-        if (!userService.updateById(user)) {
-            throw new CoolException("鏈嶅姟鍣ㄥ唴閮ㄩ敊璇�");
+        String mapKey = getMapKey(param.getFloor());
+        Dict dict = dictService.getOne(new LambdaQueryWrapper<Dict>().eq(Dict::getFlag, mapKey));
+        if (Cools.isEmpty(dict)) {
+           dict = new Dict();
+           dict.setName(mapKey);
+           dict.setFlag(mapKey);
+           dict.setType(DictType.JSON.flag);
+           dict.setValue(JSON.toJSONString(param.getItemList()));
+           dict.setCreateBy(userId);
+           dict.setUpdateBy(userId);
+           if (!dictService.save(dict)) {
+               throw new CoolException("鏈嶅姟鍣ㄥ唴閮ㄩ敊璇�");
+           }
+        } else {
+            dict.setValue(JSON.toJSONString(param.getItemList()));
+            dict.setUpdateTime(new Date());
+            dict.setUpdateBy(userId);
+            if (!dictService.updateById(dict)) {
+                throw new CoolException("鏈嶅姟鍣ㄥ唴閮ㄩ敊璇�");
+            }
         }
     }
 
+    private String getMapKey(Integer floor) {
+        if (null != floor) {
+            return "map-" + floor;
+        }
+        return "map";
+    }
+
 }

--
Gitblit v1.9.1