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