From 3481b78afafe5ff83d68a4f58741b7c15efcaeb1 Mon Sep 17 00:00:00 2001 From: kaiyu <404897439@qq.com> Date: Tue, 11 May 2021 15:20:13 +0800 Subject: [PATCH] screen-manage 组织分配菜单界面读取当前组织已有菜单接口 --- screen-manage/src/main/java/com/moral/api/service/impl/ManageRoleServiceImpl.java | 55 ++++++++++++++++++++++++++++++++++++++----------------- 1 files changed, 38 insertions(+), 17 deletions(-) diff --git a/screen-manage/src/main/java/com/moral/api/service/impl/ManageRoleServiceImpl.java b/screen-manage/src/main/java/com/moral/api/service/impl/ManageRoleServiceImpl.java index aca8fc5..5efe435 100644 --- a/screen-manage/src/main/java/com/moral/api/service/impl/ManageRoleServiceImpl.java +++ b/screen-manage/src/main/java/com/moral/api/service/impl/ManageRoleServiceImpl.java @@ -2,14 +2,8 @@ import com.alibaba.fastjson.JSON; import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; -import com.moral.api.entity.ManageAccount; -import com.moral.api.entity.ManageAccountRole; -import com.moral.api.entity.ManageLog; -import com.moral.api.entity.ManageRole; -import com.moral.api.mapper.ManageAccountRoleMapper; -import com.moral.api.mapper.ManageLogMapper; -import com.moral.api.mapper.ManageRoleMapper; -import com.moral.api.mapper.ManageRoleMenuMapper; +import com.moral.api.entity.*; +import com.moral.api.mapper.*; import com.moral.api.pojo.dto.login.AccountInfoDTO; import com.moral.api.service.ManageRoleService; import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; @@ -51,6 +45,9 @@ @Autowired(required = false) private ManageRoleMapper manageRoleMapper; + + @Autowired(required = false) + private ManageMenuMapper manageMenuMapper; @Autowired(required = false) private ManageRoleMenuMapper manageRoleMenuMapper; @@ -112,15 +109,22 @@ manageRoleMapper.insertOne(manageRole); ManageRole manageRoleIns = manageRoleMapper.getManageRoleByName(manageRole.getName()); int role_id = manageRoleIns.getId(); - List insertList = new ArrayList(); - for (Object temp: list) { - Map<String,Integer> insertMap = new HashMap<>(); - insertMap.put("role_id",role_id); - insertMap.put("menu_id",Integer.parseInt(temp.toString())); - insertList.add(insertMap); - } - //��������������������������������������������� - if (insertList.size()>0){ + if (list.size()>0){ + //��������������� + List supplementList = new ArrayList(); + supplementList.addAll(list); + for (int i=0;i<supplementList.size();i++) { + supplementList.addAll(supplementParentMenus(Integer.parseInt(supplementList.get(i).toString()))); + } + LinkedHashSet<Object> hashSet = new LinkedHashSet<>(supplementList); + ArrayList<Object> menuList = new ArrayList<>(hashSet); + List insertList = new ArrayList(); + for (Object temp: menuList) { + Map<String,Integer> insertMap = new HashMap<>(); + insertMap.put("role_id",role_id); + insertMap.put("menu_id",Integer.parseInt(temp.toString())); + insertList.add(insertMap); + } manageRoleMenuMapper.insertBatch(insertList); } //������������������ @@ -233,4 +237,21 @@ } return resultMap; } + + /** + * @Description: ��������������� + * @Param: [list, menuId]list���menuId������ + * @return: void + * @Author: ��������� + * @Date: 2021/4/28 + */ + private List supplementParentMenus(int menuId) { + List list = new ArrayList(); + ManageMenu manageMenu = manageMenuMapper.getManageMenuById(menuId); + if (manageMenu.getParentId()!=0){ + list.add(manageMenu.getParentId()); + list.addAll(supplementParentMenus(manageMenu.getParentId())); + } + return list; + } } -- Gitblit v1.8.0