From c22b5eb143260a98b9aab8fc3281ffa2a5ca2eed Mon Sep 17 00:00:00 2001
From: mg <maokecheng@163.com>
Date: 星期五, 28 十月 2022 18:07:42 +0800
Subject: [PATCH] 修改

---
 ycl-common/src/main/java/com/ycl/service/user/impl/UmsRoleServiceImpl.java |   23 +++++++++++++++++------
 1 files changed, 17 insertions(+), 6 deletions(-)

diff --git a/ycl-common/src/main/java/com/ycl/service/user/impl/UmsRoleServiceImpl.java b/ycl-common/src/main/java/com/ycl/service/user/impl/UmsRoleServiceImpl.java
index d610315..68dde62 100644
--- a/ycl-common/src/main/java/com/ycl/service/user/impl/UmsRoleServiceImpl.java
+++ b/ycl-common/src/main/java/com/ycl/service/user/impl/UmsRoleServiceImpl.java
@@ -5,16 +5,15 @@
 import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
 import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
+import com.ycl.dto.UmsMenuNode;
 import com.ycl.entity.dict.DataDictionary;
 import com.ycl.entity.user.*;
 import com.ycl.mapper.dict.DataDictionaryMapper;
 import com.ycl.mapper.user.UmsMenuMapper;
 import com.ycl.mapper.user.UmsResourceMapper;
 import com.ycl.mapper.user.UmsRoleMapper;
-import com.ycl.service.user.UmsAdminCacheService;
-import com.ycl.service.user.UmsRoleMenuRelationService;
-import com.ycl.service.user.UmsRoleResourceRelationService;
-import com.ycl.service.user.UmsRoleService;
+import com.ycl.service.user.*;
+import org.springframework.beans.BeanUtils;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
 
@@ -36,6 +35,13 @@
     private UmsRoleMenuRelationService roleMenuRelationService;
     @Autowired
     private UmsRoleResourceRelationService roleResourceRelationService;
+
+    private UmsMenuService umsMenuService;
+    @Autowired
+    public void setUmsMenuService(UmsMenuService umsMenuService) {
+        this.umsMenuService = umsMenuService;
+    }
+
     @Resource
     private UmsMenuMapper umsMenuMapper;
     @Resource
@@ -83,10 +89,15 @@
     }
 
     @Override
-    public List<UmsMenu> getMenuList(Long adminId) {
-        return umsMenuMapper.getMenuList(adminId);
+    public List<UmsMenuNode> getMenuList(Long adminId) {
+        List<UmsMenu> menuList = umsMenuMapper.getMenuList(adminId);
+        List<UmsMenuNode> result = menuList.stream()
+                .filter(menu -> menu.getParentId().equals(0L))
+                .map(menu -> umsMenuService.covertMenuNode(menu, menuList)).collect(Collectors.toList());
+        return result;
     }
 
+
     @Override
     public List<UmsMenu> listMenu(Long roleId) {
         return umsMenuMapper.getMenuListByRoleId(roleId);

--
Gitblit v1.8.0