From 8ba5c432338b400eae32180d6ece584601d7947b Mon Sep 17 00:00:00 2001
From: wl <173@qq.com>
Date: 星期五, 04 十一月 2022 09:55:06 +0800
Subject: [PATCH] 修改bug

---
 ycl-common/src/main/java/com/ycl/controller/user/UmsAdminController.java |   15 ++++++++-------
 1 files changed, 8 insertions(+), 7 deletions(-)

diff --git a/ycl-common/src/main/java/com/ycl/controller/user/UmsAdminController.java b/ycl-common/src/main/java/com/ycl/controller/user/UmsAdminController.java
index 46976df..a7516b2 100644
--- a/ycl-common/src/main/java/com/ycl/controller/user/UmsAdminController.java
+++ b/ycl-common/src/main/java/com/ycl/controller/user/UmsAdminController.java
@@ -4,6 +4,7 @@
 import com.baomidou.mybatisplus.core.metadata.IPage;
 import com.ycl.annotation.LogSave;
 import com.ycl.api.CommonResult;
+import com.ycl.bo.AdminUserDetails;
 import com.ycl.controller.BaseController;
 import com.ycl.dto.UmsAdminLoginParam;
 import com.ycl.dto.UmsAdminParam;
@@ -12,7 +13,7 @@
 import com.ycl.entity.user.UmsRole;
 import com.ycl.service.redis.RedisService;
 import com.ycl.service.user.UmsAdminService;
-import com.ycl.service.user.UmsRoleService;
+import com.ycl.service.user.UmsMenuService;
 import com.ycl.utils.redis.RedisKey;
 import com.ycl.vo.user.UserVO;
 import io.swagger.annotations.Api;
@@ -21,6 +22,7 @@
 import io.swagger.annotations.ApiOperation;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.beans.factory.annotation.Value;
+import org.springframework.security.core.context.SecurityContextHolder;
 import org.springframework.stereotype.Controller;
 import org.springframework.validation.annotation.Validated;
 import org.springframework.web.bind.annotation.*;
@@ -47,7 +49,7 @@
     @Autowired
     private UmsAdminService adminService;
     @Autowired
-    private UmsRoleService roleService;
+    UmsMenuService umsMenuService;
     @Resource
     private RedisService redisService;
 
@@ -101,13 +103,12 @@
         }
         String username = principal.getName();
         UmsAdmin umsAdmin = adminService.getAdminByUsername(username);
+        AdminUserDetails user = (AdminUserDetails) SecurityContextHolder.getContext().getAuthentication().getPrincipal();
         Map<String, Object> data = new HashMap<>();
-        data.put("userId", umsAdmin.getId());
-        data.put("username", umsAdmin.getUsername());
-        data.put("menus", roleService.getMenuList(umsAdmin.getId()));
-        data.put("icon", umsAdmin.getIcon());
+        data.put("user", umsAdmin);
+        data.put("menus", umsMenuService.ListTreeMyself(user.getUserId()));
         List<UmsRole> roleList = adminService.getRoleList(umsAdmin.getId());
-        if (CollUtil.isNotEmpty(roleList)) {    
+        if (CollUtil.isNotEmpty(roleList)) {
             List<String> roles = roleList.stream().map(UmsRole::getName).collect(Collectors.toList());
             data.put("roles", roles);
         }

--
Gitblit v1.8.0