From 25b114aae7e9efb2bc54cca8a5875237bc3da694 Mon Sep 17 00:00:00 2001 From: RuoYi <yzz_ivy@163.com> Date: 星期一, 18 五月 2020 14:27:15 +0800 Subject: [PATCH] 支持是否需要设置token属性,自定义返回码消息。 --- ruoyi/src/main/java/com/ruoyi/project/system/service/impl/SysMenuServiceImpl.java | 32 ++++++++++++++++++++++++++++---- 1 files changed, 28 insertions(+), 4 deletions(-) diff --git a/ruoyi/src/main/java/com/ruoyi/project/system/service/impl/SysMenuServiceImpl.java b/ruoyi/src/main/java/com/ruoyi/project/system/service/impl/SysMenuServiceImpl.java index 2b19546..31eb70b 100644 --- a/ruoyi/src/main/java/com/ruoyi/project/system/service/impl/SysMenuServiceImpl.java +++ b/ruoyi/src/main/java/com/ruoyi/project/system/service/impl/SysMenuServiceImpl.java @@ -15,6 +15,7 @@ import com.ruoyi.common.utils.StringUtils; import com.ruoyi.framework.web.domain.TreeSelect; import com.ruoyi.project.system.domain.SysMenu; +import com.ruoyi.project.system.domain.SysUser; import com.ruoyi.project.system.domain.vo.MetaVo; import com.ruoyi.project.system.domain.vo.RouterVo; import com.ruoyi.project.system.mapper.SysMenuMapper; @@ -38,15 +39,37 @@ private SysRoleMenuMapper roleMenuMapper; /** + * 根据用户查询系统菜单列表 + * + * @param userId 用户ID + * @return 菜单列表 + */ + @Override + public List<SysMenu> selectMenuList(Long userId) + { + return selectMenuList(new SysMenu(), userId); + } + + /** * 查询系统菜单列表 * * @param menu 菜单信息 * @return 菜单列表 */ @Override - public List<SysMenu> selectMenuList(SysMenu menu) + public List<SysMenu> selectMenuList(SysMenu menu, Long userId) { - List<SysMenu> menuList = menuMapper.selectMenuList(menu); + List<SysMenu> menuList = null; + // 管理员显示所有菜单信息 + if (SysUser.isAdmin(userId)) + { + menuList = menuMapper.selectMenuList(menu); + } + else + { + menu.getParams().put("userId", userId); + menuList = menuMapper.selectMenuListByUserId(menu); + } return menuList; } @@ -72,9 +95,9 @@ } /** - * 根据用户名称查询菜单 + * 根据用户ID查询菜单 * - * @param username 用户名称 + * @param userId 用户名称 * @return 菜单列表 */ @Override @@ -116,6 +139,7 @@ for (SysMenu menu : menus) { RouterVo router = new RouterVo(); + router.setHidden("1".equals(menu.getVisible())); router.setName(StringUtils.capitalize(menu.getPath())); router.setPath(getRouterPath(menu)); router.setComponent(StringUtils.isEmpty(menu.getComponent()) ? "Layout" : menu.getComponent()); -- Gitblit v1.9.2