SysMenuServiceImpl.java 14 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441
  1. package com.ruoyi.system.service.impl;
  2. import cn.hutool.core.util.StrUtil;
  3. import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
  4. import com.ruoyi.common.constant.Constants;
  5. import com.ruoyi.common.constant.UserConstants;
  6. import com.ruoyi.common.core.domain.TreeSelect;
  7. import com.ruoyi.common.core.domain.entity.SysMenu;
  8. import com.ruoyi.common.core.domain.entity.SysRole;
  9. import com.ruoyi.common.core.domain.entity.SysUser;
  10. import com.ruoyi.common.core.mybatisplus.core.ServicePlusImpl;
  11. import com.ruoyi.common.utils.SecurityUtils;
  12. import com.ruoyi.common.utils.StringUtils;
  13. import com.ruoyi.system.domain.SysRoleMenu;
  14. import com.ruoyi.system.domain.vo.MetaVo;
  15. import com.ruoyi.system.domain.vo.RouterVo;
  16. import com.ruoyi.system.mapper.SysMenuMapper;
  17. import com.ruoyi.system.mapper.SysRoleMapper;
  18. import com.ruoyi.system.mapper.SysRoleMenuMapper;
  19. import com.ruoyi.system.service.ISysMenuService;
  20. import org.springframework.beans.factory.annotation.Autowired;
  21. import org.springframework.stereotype.Service;
  22. import java.util.*;
  23. import java.util.stream.Collectors;
  24. /**
  25. * 菜单 业务层处理
  26. *
  27. * @author ruoyi
  28. */
  29. @Service
  30. public class SysMenuServiceImpl extends ServicePlusImpl<SysMenuMapper, SysMenu, SysMenu> implements ISysMenuService {
  31. public static final String PREMISSION_STRING = "perms[\"{0}\"]";
  32. @Autowired
  33. private SysRoleMapper roleMapper;
  34. @Autowired
  35. private SysRoleMenuMapper roleMenuMapper;
  36. /**
  37. * 根据用户查询系统菜单列表
  38. *
  39. * @param userId 用户ID
  40. * @return 菜单列表
  41. */
  42. @Override
  43. public List<SysMenu> selectMenuList(String userId) {
  44. return selectMenuList(new SysMenu(), userId);
  45. }
  46. /**
  47. * 查询系统菜单列表
  48. *
  49. * @param menu 菜单信息
  50. * @return 菜单列表
  51. */
  52. @Override
  53. public List<SysMenu> selectMenuList(SysMenu menu, String userId) {
  54. List<SysMenu> menuList = null;
  55. // 管理员显示所有菜单信息
  56. if (SysUser.isAdmin(userId)) {
  57. menuList = list(new LambdaQueryWrapper<SysMenu>()
  58. .like(StringUtils.isNotBlank(menu.getMenuName()),SysMenu::getMenuName,menu.getMenuName())
  59. .eq(StringUtils.isNotBlank(menu.getVisible()),SysMenu::getVisible,menu.getVisible())
  60. .eq(StringUtils.isNotBlank(menu.getStatus()),SysMenu::getStatus,menu.getStatus())
  61. .orderByAsc(SysMenu::getParentId)
  62. .orderByAsc(SysMenu::getOrderNum));
  63. } else {
  64. menu.getParams().put("userId", userId);
  65. menuList = baseMapper.selectMenuListByUserId(menu);
  66. }
  67. return menuList;
  68. }
  69. /**
  70. * 根据用户ID查询权限
  71. *
  72. * @param userId 用户ID
  73. * @return 权限列表
  74. */
  75. @Override
  76. public Set<String> selectMenuPermsByUserId(String userId) {
  77. List<String> perms = baseMapper.selectMenuPermsByUserId(userId);
  78. Set<String> permsSet = new HashSet<>();
  79. for (String perm : perms) {
  80. if (StringUtils.isNotEmpty(perm)) {
  81. permsSet.addAll(Arrays.asList(perm.trim().split(",")));
  82. }
  83. }
  84. return permsSet;
  85. }
  86. /**
  87. * 根据用户ID查询菜单
  88. *
  89. * @param userId 用户名称
  90. * @return 菜单列表
  91. */
  92. @Override
  93. public List<SysMenu> selectMenuTreeByUserId(String userId) {
  94. List<SysMenu> menus = null;
  95. if (SecurityUtils.isAdmin(userId)) {
  96. menus = baseMapper.selectMenuTreeAll();
  97. } else {
  98. menus = baseMapper.selectMenuTreeByUserId(userId);
  99. }
  100. return getChildPerms(menus, 0);
  101. }
  102. /**
  103. * 根据角色ID查询菜单树信息
  104. *
  105. * @param roleId 角色ID
  106. * @return 选中菜单列表
  107. */
  108. @Override
  109. public List<Integer> selectMenuListByRoleId(String roleId) {
  110. SysRole role = roleMapper.selectById(roleId);
  111. return baseMapper.selectMenuListByRoleId(roleId, role.isMenuCheckStrictly());
  112. }
  113. /**
  114. * 构建前端路由所需要的菜单
  115. *
  116. * @param menus 菜单列表
  117. * @return 路由列表
  118. */
  119. @Override
  120. public List<RouterVo> buildMenus(List<SysMenu> menus) {
  121. List<RouterVo> routers = new LinkedList<RouterVo>();
  122. for (SysMenu menu : menus) {
  123. RouterVo router = new RouterVo();
  124. router.setHidden("1".equals(menu.getVisible()));
  125. router.setName(getRouteName(menu));
  126. router.setPath(getRouterPath(menu));
  127. router.setComponent(getComponent(menu));
  128. router.setMeta(new MetaVo(menu.getMenuName(), menu.getIcon(), StringUtils.equals("1", menu.getIsCache()), menu.getPath()));
  129. List<SysMenu> cMenus = menu.getChildren();
  130. if (!cMenus.isEmpty() && UserConstants.TYPE_DIR.equals(menu.getMenuType())) {
  131. router.setAlwaysShow(true);
  132. router.setRedirect("noRedirect");
  133. router.setChildren(buildMenus(cMenus));
  134. } else if (isMenuFrame(menu)) {
  135. router.setMeta(null);
  136. List<RouterVo> childrenList = new ArrayList<RouterVo>();
  137. RouterVo children = new RouterVo();
  138. children.setPath(menu.getPath());
  139. children.setComponent(menu.getComponent());
  140. children.setName(StringUtils.capitalize(menu.getPath()));
  141. children.setMeta(new MetaVo(menu.getMenuName(), menu.getIcon(), StringUtils.equals("1", menu.getIsCache()), menu.getPath()));
  142. childrenList.add(children);
  143. router.setChildren(childrenList);
  144. } else if (menu.getParentId().intValue() == 0 && isInnerLink(menu)) {
  145. router.setMeta(new MetaVo(menu.getMenuName(), menu.getIcon()));
  146. router.setPath("/inner");
  147. List<RouterVo> childrenList = new ArrayList<RouterVo>();
  148. RouterVo children = new RouterVo();
  149. String routerPath = StringUtils.replaceEach(menu.getPath(), new String[] { Constants.HTTP, Constants.HTTPS }, new String[] { "", "" });
  150. children.setPath(routerPath);
  151. children.setComponent(UserConstants.INNER_LINK);
  152. children.setName(StringUtils.capitalize(routerPath));
  153. children.setMeta(new MetaVo(menu.getMenuName(), menu.getIcon(), menu.getPath()));
  154. childrenList.add(children);
  155. router.setChildren(childrenList);
  156. }
  157. routers.add(router);
  158. }
  159. return routers;
  160. }
  161. /**
  162. * 构建前端所需要树结构
  163. *
  164. * @param menus 菜单列表
  165. * @return 树结构列表
  166. */
  167. @Override
  168. public List<SysMenu> buildMenuTree(List<SysMenu> menus) {
  169. List<SysMenu> returnList = new ArrayList<SysMenu>();
  170. List<String> tempList = new ArrayList<>();
  171. for (SysMenu dept : menus) {
  172. tempList.add(dept.getMenuId());
  173. }
  174. for (SysMenu menu : menus) {
  175. // 如果是顶级节点, 遍历该父节点的所有子节点
  176. if (!tempList.contains(menu.getParentId())) {
  177. recursionFn(menus, menu);
  178. returnList.add(menu);
  179. }
  180. }
  181. if (returnList.isEmpty()) {
  182. returnList = menus;
  183. }
  184. return returnList;
  185. }
  186. /**
  187. * 构建前端所需要下拉树结构
  188. *
  189. * @param menus 菜单列表
  190. * @return 下拉树结构列表
  191. */
  192. @Override
  193. public List<TreeSelect> buildMenuTreeSelect(List<SysMenu> menus) {
  194. List<SysMenu> menuTrees = buildMenuTree(menus);
  195. return menuTrees.stream().map(TreeSelect::new).collect(Collectors.toList());
  196. }
  197. /**
  198. * 根据菜单ID查询信息
  199. *
  200. * @param menuId 菜单ID
  201. * @return 菜单信息
  202. */
  203. @Override
  204. public SysMenu selectMenuById(String menuId) {
  205. return getById(menuId);
  206. }
  207. /**
  208. * 是否存在菜单子节点
  209. *
  210. * @param menuId 菜单ID
  211. * @return 结果
  212. */
  213. @Override
  214. public boolean hasChildByMenuId(String menuId) {
  215. long result = count(new LambdaQueryWrapper<SysMenu>().eq(SysMenu::getParentId,menuId));
  216. return result > 0;
  217. }
  218. /**
  219. * 查询菜单使用数量
  220. *
  221. * @param menuId 菜单ID
  222. * @return 结果
  223. */
  224. @Override
  225. public boolean checkMenuExistRole(String menuId) {
  226. long result = roleMenuMapper.selectCount(new LambdaQueryWrapper<SysRoleMenu>().eq(SysRoleMenu::getMenuId,menuId));
  227. return result > 0;
  228. }
  229. /**
  230. * 新增保存菜单信息
  231. *
  232. * @param menu 菜单信息
  233. * @return 结果
  234. */
  235. @Override
  236. public int insertMenu(SysMenu menu) {
  237. return baseMapper.insert(menu);
  238. }
  239. /**
  240. * 修改保存菜单信息
  241. *
  242. * @param menu 菜单信息
  243. * @return 结果
  244. */
  245. @Override
  246. public int updateMenu(SysMenu menu) {
  247. return baseMapper.updateById(menu);
  248. }
  249. /**
  250. * 删除菜单管理信息
  251. *
  252. * @param menuId 菜单ID
  253. * @return 结果
  254. */
  255. @Override
  256. public int deleteMenuById(String menuId) {
  257. return baseMapper.deleteById(menuId);
  258. }
  259. /**
  260. * 校验菜单名称是否唯一
  261. *
  262. * @param menu 菜单信息
  263. * @return 结果
  264. */
  265. @Override
  266. public String checkMenuNameUnique(SysMenu menu) {
  267. String menuId = StrUtil.isBlank(menu.getMenuId()) ? "-1" : menu.getMenuId();
  268. SysMenu info = getOne(new LambdaQueryWrapper<SysMenu>()
  269. .eq(SysMenu::getMenuName,menu.getMenuName())
  270. .eq(SysMenu::getParentId,menu.getParentId())
  271. .last("limit 1"));
  272. if (StringUtils.isNotNull(info) && !info.getMenuId().equals(menuId)) {
  273. return UserConstants.NOT_UNIQUE;
  274. }
  275. return UserConstants.UNIQUE;
  276. }
  277. /**
  278. * 获取路由名称
  279. *
  280. * @param menu 菜单信息
  281. * @return 路由名称
  282. */
  283. public String getRouteName(SysMenu menu) {
  284. String routerName = StringUtils.capitalize(menu.getPath());
  285. // 非外链并且是一级目录(类型为目录)
  286. if (isMenuFrame(menu)) {
  287. routerName = StringUtils.EMPTY;
  288. }
  289. return routerName;
  290. }
  291. /**
  292. * 获取路由地址
  293. *
  294. * @param menu 菜单信息
  295. * @return 路由地址
  296. */
  297. public String getRouterPath(SysMenu menu) {
  298. String routerPath = menu.getPath();
  299. // 内链打开外网方式
  300. if (menu.getParentId().intValue() != 0 && isInnerLink(menu)) {
  301. routerPath = StringUtils.replaceEach(routerPath, new String[] { Constants.HTTP, Constants.HTTPS }, new String[] { "", "" });
  302. }
  303. // 非外链并且是一级目录(类型为目录)
  304. if (0 == menu.getParentId().intValue() && UserConstants.TYPE_DIR.equals(menu.getMenuType())
  305. && UserConstants.NO_FRAME.equals(menu.getIsFrame())) {
  306. routerPath = "/" + menu.getPath();
  307. }
  308. // 非外链并且是一级目录(类型为菜单)
  309. else if (isMenuFrame(menu)) {
  310. routerPath = "/";
  311. }
  312. return routerPath;
  313. }
  314. /**
  315. * 获取组件信息
  316. *
  317. * @param menu 菜单信息
  318. * @return 组件信息
  319. */
  320. public String getComponent(SysMenu menu) {
  321. String component = UserConstants.LAYOUT;
  322. if (StringUtils.isNotEmpty(menu.getComponent()) && !isMenuFrame(menu)) {
  323. component = menu.getComponent();
  324. } else if (StringUtils.isEmpty(menu.getComponent()) && menu.getParentId().intValue() != 0 && isInnerLink(menu)) {
  325. component = UserConstants.INNER_LINK;
  326. } else if (StringUtils.isEmpty(menu.getComponent()) && isParentView(menu)) {
  327. component = UserConstants.PARENT_VIEW;
  328. }
  329. return component;
  330. }
  331. /**
  332. * 是否为菜单内部跳转
  333. *
  334. * @param menu 菜单信息
  335. * @return 结果
  336. */
  337. public boolean isMenuFrame(SysMenu menu) {
  338. return menu.getParentId().intValue() == 0 && UserConstants.TYPE_MENU.equals(menu.getMenuType())
  339. && menu.getIsFrame().equals(UserConstants.NO_FRAME);
  340. }
  341. /**
  342. * 是否为内链组件
  343. *
  344. * @param menu 菜单信息
  345. * @return 结果
  346. */
  347. public boolean isInnerLink(SysMenu menu) {
  348. return menu.getIsFrame().equals(UserConstants.NO_FRAME) && StringUtils.ishttp(menu.getPath());
  349. }
  350. /**
  351. * 是否为parent_view组件
  352. *
  353. * @param menu 菜单信息
  354. * @return 结果
  355. */
  356. public boolean isParentView(SysMenu menu) {
  357. return menu.getParentId().intValue() != 0 && UserConstants.TYPE_DIR.equals(menu.getMenuType());
  358. }
  359. /**
  360. * 根据父节点的ID获取所有子节点
  361. *
  362. * @param list 分类表
  363. * @param parentId 传入的父节点ID
  364. * @return String
  365. */
  366. public List<SysMenu> getChildPerms(List<SysMenu> list, int parentId) {
  367. List<SysMenu> returnList = new ArrayList<SysMenu>();
  368. for (SysMenu t : list) {
  369. // 一、根据传入的某个父节点ID,遍历该父节点的所有子节点
  370. if (t.getParentId() == parentId) {
  371. recursionFn(list, t);
  372. returnList.add(t);
  373. }
  374. }
  375. return returnList;
  376. }
  377. /**
  378. * 递归列表
  379. *
  380. * @param list
  381. * @param t
  382. */
  383. private void recursionFn(List<SysMenu> list, SysMenu t) {
  384. // 得到子节点列表
  385. List<SysMenu> childList = getChildList(list, t);
  386. t.setChildren(childList);
  387. for (SysMenu tChild : childList) {
  388. if (hasChild(list, tChild)) {
  389. recursionFn(list, tChild);
  390. }
  391. }
  392. }
  393. /**
  394. * 得到子节点列表
  395. */
  396. private List<SysMenu> getChildList(List<SysMenu> list, SysMenu t) {
  397. List<SysMenu> tlist = new ArrayList<SysMenu>();
  398. for (SysMenu n : list) {
  399. if (n.getParentId().equals(t.getMenuId())) {
  400. tlist.add(n);
  401. }
  402. }
  403. return tlist;
  404. }
  405. /**
  406. * 判断是否有子节点
  407. */
  408. private boolean hasChild(List<SysMenu> list, SysMenu t) {
  409. return getChildList(list, t).size() > 0;
  410. }
  411. }