From 4b8e20b85c1e114aacdab2a31bd851012f71b755 Mon Sep 17 00:00:00 2001
From: lizijie <lzjiiie@163.com>
Date: Tue, 27 Apr 2021 17:11:20 +0800
Subject: [PATCH] 递归查询前端传入的所有子菜单的父菜单
---
screen-manage/src/main/java/com/moral/api/service/impl/MenuServiceImpl.java | 182 +++++++++++++++++++++++++++++++++++++++++++++
1 files changed, 182 insertions(+), 0 deletions(-)
diff --git a/screen-manage/src/main/java/com/moral/api/service/impl/MenuServiceImpl.java b/screen-manage/src/main/java/com/moral/api/service/impl/MenuServiceImpl.java
index 5047a8c..efcaec9 100644
--- a/screen-manage/src/main/java/com/moral/api/service/impl/MenuServiceImpl.java
+++ b/screen-manage/src/main/java/com/moral/api/service/impl/MenuServiceImpl.java
@@ -1,10 +1,28 @@
package com.moral.api.service.impl;
+import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
+import com.baomidou.mybatisplus.core.conditions.update.UpdateWrapper;
+import com.moral.api.entity.ManageMenu;
import com.moral.api.entity.Menu;
import com.moral.api.mapper.MenuMapper;
+import com.moral.api.pojo.dto.menu.MenuDTO;
+import com.moral.api.pojo.dto.menu.MenuQueryDTO;
+import com.moral.api.pojo.form.menu.MenuDeleteForm;
+import com.moral.api.pojo.form.menu.MenuInsertForm;
import com.moral.api.service.MenuService;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
+import com.moral.constant.Constants;
+import com.moral.constant.ResponseCodeEnum;
+import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
+import org.springframework.transaction.annotation.Transactional;
+import org.springframework.util.ObjectUtils;
+
+import java.util.ArrayList;
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
+import java.util.function.Predicate;
/**
* <p>
@@ -17,4 +35,168 @@
@Service
public class MenuServiceImpl extends ServiceImpl<MenuMapper, Menu> implements MenuService {
+ @Autowired
+ MenuMapper menuMapper;
+
+ @Override
+ public MenuQueryDTO queryAllMenu() {
+ //������������������
+ MenuQueryDTO dto = new MenuQueryDTO();
+ //������������������
+ List<Menu> menus = menuMapper.selectList(null);
+ //������menu������������
+ Map<Integer, Menu> menuMap = new HashMap<>();
+ for (Menu menu : menus) {
+ menuMap.put(menu.getId(), menu);
+ menu.setChildren(new ArrayList<>());//���������������
+ }
+ for (Menu menu : menus) {
+ combinationParentChildrenMenus(menuMap, menu);
+ }
+ //������������������
+ menus.removeIf(new Predicate<Menu>() {
+ @Override
+ public boolean test(Menu menu) {
+ if (menu.getParentId().equals(0))
+ return false;
+ return true;
+ }
+ });
+ //���������������DTO
+ List<MenuDTO> dtos = new ArrayList<>();
+ for (Menu menu : menus) {
+ MenuDTO menuDTO = new MenuDTO();
+ menuDTO.setMenu(menu);
+ dtos.add(menuDTO);
+ }
+ dto.setDtos(dtos);
+ dto.setCode(ResponseCodeEnum.SUCCESS.getCode());
+ dto.setMsg(ResponseCodeEnum.SUCCESS.getMsg());
+ return dto;
+ }
+
+ @Override
+ @Transactional
+ public MenuDTO insertMenu(MenuInsertForm form) {
+ //������������������
+ MenuDTO dto = new MenuDTO();
+ //������
+ Menu menu = form.formConvertEntity();
+ //������������������������
+ Menu existMenu = new Menu();
+ existMenu.setName(menu.getName());
+ existMenu.setIsDelete(Constants.NOT_DELETE);
+ QueryWrapper existWrapper = new QueryWrapper();
+ existWrapper.setEntity(existMenu);
+ Menu existMenuResult = menuMapper.selectOne(existWrapper);
+ if (!ObjectUtils.isEmpty(existMenuResult)) {
+ dto.setCode(ResponseCodeEnum.MENU_IS_EXIST.getCode());
+ dto.setMsg(ResponseCodeEnum.MENU_IS_EXIST.getMsg());
+ return dto;
+ }
+ //������������������
+ menuMapper.insert(menu);
+ //������������������
+ dto.setMenu(menu);
+ dto.setCode(ResponseCodeEnum.SUCCESS.getCode());
+ dto.setMsg(ResponseCodeEnum.SUCCESS.getMsg());
+ return dto;
+ }
+
+ @Override
+ public MenuDTO deleteMenu(MenuDeleteForm form) {
+ //������������������
+ MenuDTO dto = new MenuDTO();
+ //������
+ Integer id = form.getId();
+ //������������������������
+ Menu existMenu = menuMapper.selectById(id);
+ if (ObjectUtils.isEmpty(existMenu)) {
+ dto.setCode(ResponseCodeEnum.MENU_IS_NULL.getCode());
+ dto.setMsg(ResponseCodeEnum.MENU_IS_NULL.getMsg());
+ return dto;
+ }
+ //������������������
+ existMenu.setIsDelete(Constants.DELETE);
+ menuMapper.updateById(existMenu);
+ //���������������������
+ List<Integer> childrenIds = getChildrenIdsByParentId(id);
+ UpdateWrapper wrapper = new UpdateWrapper();
+ wrapper.in("id",childrenIds);
+ wrapper.set("is_delete",Constants.DELETE);
+ menuMapper.update(null,wrapper);
+ //������������������
+ dto.setCode(ResponseCodeEnum.SUCCESS.getCode());
+ dto.setMsg(ResponseCodeEnum.SUCCESS.getMsg());
+ dto.setMenu(existMenu);
+ return dto;
+ }
+
+ private void combinationParentChildrenMenus(Map<Integer, Menu> menuMap, Menu menu) {
+ Integer parentId = menu.getParentId();
+ Menu parentMenu = menuMap.get(parentId);
+ if (!ObjectUtils.isEmpty(parentMenu)) {
+ parentMenu.getChildren().add(menu);
+ }
+ }
+
+ /**
+ * @Description: ���������������id���������������children���menu������
+ * @Param: [id]
+ * @return: com.moral.api.entity.Menu
+ * @Author: ���������
+ * @Date: 2021/4/26
+ */
+ private Menu getMenuAndChildrenById(Integer id){
+ //������������������
+ List<Menu> menus = menuMapper.selectList(null);
+ //������menu������������
+ Map<Integer, Menu> menuMap = new HashMap<>();
+ for (Menu menu : menus) {
+ menuMap.put(menu.getId(), menu);
+ menu.setChildren(new ArrayList<>());//���������������
+ }
+ for (Menu menu : menus) {
+ combinationParentChildrenMenus(menuMap, menu);
+ }
+ Menu menu = menuMap.get(id);
+ return menu;
+ }
+
+ /**
+ * @Description: ������������������������������������id������
+ * @Param: [id]
+ * @return: java.util.List<java.lang.Integer>
+ * @Author: ���������
+ * @Date: 2021/4/26
+ */
+ private List<Integer> getChildrenIdsByParentId(Integer id){
+ Menu parentMenu = getMenuAndChildrenById(id);
+ List<Menu> children = parentMenu.getChildren();
+ List<Integer> childrenIds = new ArrayList<>();
+ for (Menu child : children) {
+ recursiveAccess(child,childrenIds);
+ }
+ return childrenIds;
+ }
+
+
+ /**
+ * @Description: ���������������������id���������������
+ * @Param: [menu, ids]
+ * @return: void
+ * @Author: ���������
+ * @Date: 2021/4/26
+ */
+ private void recursiveAccess(Menu menu,List<Integer> ids){
+ ids.add(menu.getId());
+ List<Menu> children = menu.getChildren();
+ if(!ObjectUtils.isEmpty(children)){
+ for (Menu child : children) {
+ recursiveAccess(child,ids);
+ }
+ }
+ }
+
}
+
--
Gitblit v1.8.0