From 1c47b33225c57a13b65fc3bc2e43a7822b23b6a3 Mon Sep 17 00:00:00 2001
From: Junjie <540245094@qq.com>
Date: 星期五, 05 一月 2024 17:00:01 +0800
Subject: [PATCH] #

---
 zy-asrs-common/src/main/java/com/zy/asrs/common/utils/TreeUtils.java |   28 ++++++++++++++++++++++------
 1 files changed, 22 insertions(+), 6 deletions(-)

diff --git a/zy-asrs-common/src/main/java/com/zy/asrs/common/utils/TreeUtils.java b/zy-asrs-common/src/main/java/com/zy/asrs/common/utils/TreeUtils.java
index f1db360..43a8b3f 100644
--- a/zy-asrs-common/src/main/java/com/zy/asrs/common/utils/TreeUtils.java
+++ b/zy-asrs-common/src/main/java/com/zy/asrs/common/utils/TreeUtils.java
@@ -2,7 +2,7 @@
 
 import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
 import com.zy.asrs.common.wms.entity.Tag;
-import com.zy.asrs.common.wms.service.ManTagService;
+import com.zy.asrs.common.wms.service.TagService;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.cache.annotation.Cacheable;
 import org.springframework.stereotype.Component;
@@ -17,17 +17,17 @@
 public class TreeUtils {
 
     @Autowired
-    private ManTagService tagService;
+    private TagService tagService;
 
     /******************************** 褰掔被鏍� *********************************/
 
     /**
      * 鑾峰彇鏍戝浘鏁版嵁缁撴瀯
      */
-    @Cacheable(cacheNames="tagTree",key="#id")
-    public ArrayList<Map> getTree(String id){
+    @Cacheable(cacheNames = "tagTree", key = "#id")
+    public ArrayList<Map> getTree(Long id, Long hostId) {
         ArrayList<Map> result = new ArrayList<>();
-        Tag tag = tagService.getById(id);
+        Tag tag = tagService.getOne(new LambdaQueryWrapper<Tag>().eq(Tag::getId, id).eq(Tag::getHostId, hostId));
         // 涓昏妭鐐�
         Map<String, Object> map = new HashMap<>();
         map.put("title", tag.getName());
@@ -49,7 +49,7 @@
         List<Tag> tags = tagService.list(
                 new LambdaQueryWrapper<Tag>()
                         .eq(Tag::getParentId, parent.getId())
-                        .eq(Tag::getStatus, "1"));
+                        .eq(Tag::getStatus, 1));
         for (Tag tag : tags) {
             Map<String, Object> map = new HashMap<>();
             map.put("title", tag.getName());
@@ -62,6 +62,22 @@
         }
     }
 
+    /**
+     * 閫掑綊鑾峰彇鑺傜偣浠ュ強瀛愯妭鐐规暟鎹�
+     */
+    public void getTagIdList(Long id, List<Long> nodes) {
+        if (!nodes.contains(id)) {
+            nodes.add(id);
+        }
+        List<Tag> tags = tagService.list(new LambdaQueryWrapper<Tag>().eq(Tag::getParentId, id).eq(Tag::getStatus, 1));
+        for (Tag tag : tags) {
+            if (!nodes.contains(tag.getId())) {
+                nodes.add(tag.getId());
+            }
+            getTagIdList(tag.getId(), nodes);
+        }
+    }
+
 
     // -------------------------------------------------------------------------------------------------------
 

--
Gitblit v1.9.1