package com.ruoyi.project.system.menu.service; import java.text.MessageFormat; import java.util.ArrayList; import java.util.Arrays; import java.util.HashSet; import java.util.LinkedHashMap; import java.util.LinkedList; import java.util.List; import java.util.Set; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.stereotype.Service; import com.ruoyi.common.constant.UserConstants; import com.ruoyi.common.utils.StringUtils; import com.ruoyi.common.utils.TreeUtils; import com.ruoyi.common.utils.security.ShiroUtils; import com.ruoyi.framework.web.domain.Ztree; import com.ruoyi.project.system.menu.domain.Menu; import com.ruoyi.project.system.menu.mapper.MenuMapper; import com.ruoyi.project.system.role.domain.Role; import com.ruoyi.project.system.role.mapper.RoleMenuMapper; import com.ruoyi.project.system.user.domain.User; /** * 菜单 业务层处理 * * @author ruoyi */ @Service public class MenuServiceImpl implements IMenuService { public static final String PREMISSION_STRING = "perms[\"{0}\"]"; @Autowired private MenuMapper menuMapper; @Autowired private RoleMenuMapper roleMenuMapper; /** * 根据用户查询菜单 * * @param user 用户信息 * @return 菜单列表 */ @Override public List selectMenusByUser(User user) { List menus = new LinkedList(); // 管理员显示所有菜单信息 if (user.isAdmin()) { menus = menuMapper.selectMenuNormalAll(); } else { menus = menuMapper.selectMenusByUserId(user.getUserId()); } return TreeUtils.getChildPerms(menus, 0); } /** * 查询菜单集合 * * @return 所有菜单信息 */ @Override public List selectMenuList(Menu menu) { List menuList = null; User user = ShiroUtils.getSysUser(); if (user.isAdmin()) { menuList = menuMapper.selectMenuList(menu); } else { menu.getParams().put("userId", user.getUserId()); menuList = menuMapper.selectMenuListByUserId(menu); } return menuList; } /** * 查询菜单集合 * * @return 所有菜单信息 */ @Override public List selectMenuAll() { List menuList = null; User user = ShiroUtils.getSysUser(); if (user.isAdmin()) { menuList = menuMapper.selectMenuAll(); } else { menuList = menuMapper.selectMenuAllByUserId(user.getUserId()); } return menuList; } /** * 根据用户ID查询权限 * * @param userId 用户ID * @return 权限列表 */ @Override public Set selectPermsByUserId(Long userId) { List perms = menuMapper.selectPermsByUserId(userId); Set permsSet = new HashSet<>(); for (String perm : perms) { if (StringUtils.isNotEmpty(perm)) { permsSet.addAll(Arrays.asList(perm.trim().split(","))); } } return permsSet; } /** * 根据角色ID查询菜单 * * @param role 角色对象 * @return 菜单列表 */ @Override public List roleMenuTreeData(Role role) { Long roleId = role.getRoleId(); List ztrees = new ArrayList(); List menuList = selectMenuAll(); if (StringUtils.isNotNull(roleId)) { List roleMenuList = menuMapper.selectMenuTree(roleId); ztrees = initZtree(menuList, roleMenuList, true); } else { ztrees = initZtree(menuList, null, true); } return ztrees; } /** * 查询所有菜单 * * @return 菜单列表 */ @Override public List menuTreeData() { List menuList = selectMenuAll(); List ztrees = initZtree(menuList); return ztrees; } /** * 查询系统所有权限 * * @return 权限列表 */ @Override public LinkedHashMap selectPermsAll() { LinkedHashMap section = new LinkedHashMap<>(); List permissions = selectMenuAll(); if (StringUtils.isNotEmpty(permissions)) { for (Menu menu : permissions) { section.put(menu.getUrl(), MessageFormat.format(PREMISSION_STRING, menu.getPerms())); } } return section; } /** * 对象转菜单树 * * @param menuList 菜单列表 * @return 树结构列表 */ public List initZtree(List menuList) { return initZtree(menuList, null, false); } /** * 对象转菜单树 * * @param menuList 菜单列表 * @param roleMenuList 角色已存在菜单列表 * @param permsFlag 是否需要显示权限标识 * @return 树结构列表 */ public List initZtree(List menuList, List roleMenuList, boolean permsFlag) { List ztrees = new ArrayList(); boolean isCheck = StringUtils.isNotNull(roleMenuList); for (Menu menu : menuList) { Ztree ztree = new Ztree(); ztree.setId(menu.getMenuId()); ztree.setpId(menu.getParentId()); ztree.setName(transMenuName(menu, permsFlag)); ztree.setTitle(menu.getMenuName()); if (isCheck) { ztree.setChecked(roleMenuList.contains(menu.getMenuId() + menu.getPerms())); } ztrees.add(ztree); } return ztrees; } public String transMenuName(Menu menu, boolean permsFlag) { StringBuffer sb = new StringBuffer(); sb.append(menu.getMenuName()); if (permsFlag) { sb.append("   " + menu.getPerms() + ""); } return sb.toString(); } /** * 删除菜单管理信息 * * @param menuId 菜单ID * @return 结果 */ @Override public int deleteMenuById(Long menuId) { ShiroUtils.clearCachedAuthorizationInfo(); return menuMapper.deleteMenuById(menuId); } /** * 根据菜单ID查询信息 * * @param menuId 菜单ID * @return 菜单信息 */ @Override public Menu selectMenuById(Long menuId) { return menuMapper.selectMenuById(menuId); } /** * 查询子菜单数量 * * @param parentId 菜单ID * @return 结果 */ @Override public int selectCountMenuByParentId(Long parentId) { return menuMapper.selectCountMenuByParentId(parentId); } /** * 查询菜单使用数量 * * @param menuId 菜单ID * @return 结果 */ @Override public int selectCountRoleMenuByMenuId(Long menuId) { return roleMenuMapper.selectCountRoleMenuByMenuId(menuId); } /** * 新增保存菜单信息 * * @param menu 菜单信息 * @return 结果 */ @Override public int insertMenu(Menu menu) { menu.setCreateBy(ShiroUtils.getLoginName()); ShiroUtils.clearCachedAuthorizationInfo(); return menuMapper.insertMenu(menu); } /** * 修改保存菜单信息 * * @param menu 菜单信息 * @return 结果 */ @Override public int updateMenu(Menu menu) { menu.setUpdateBy(ShiroUtils.getLoginName()); ShiroUtils.clearCachedAuthorizationInfo(); return menuMapper.updateMenu(menu); } /** * 校验菜单名称是否唯一 * * @param menu 菜单信息 * @return 结果 */ @Override public String checkMenuNameUnique(Menu menu) { Long menuId = StringUtils.isNull(menu.getMenuId()) ? -1L : menu.getMenuId(); Menu info = menuMapper.checkMenuNameUnique(menu.getMenuName(), menu.getParentId()); if (StringUtils.isNotNull(info) && info.getMenuId().longValue() != menuId.longValue()) { return UserConstants.MENU_NAME_NOT_UNIQUE; } return UserConstants.MENU_NAME_UNIQUE; } }