kaiyu
2021-05-11 3481b78afafe5ff83d68a4f58741b7c15efcaeb1
screen-manage/src/main/java/com/moral/api/service/impl/ManageRoleMenuServiceImpl.java
@@ -16,14 +16,13 @@
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.util.ObjectUtils;
import org.springframework.web.context.request.RequestContextHolder;
import org.springframework.web.context.request.ServletRequestAttributes;
import javax.servlet.http.HttpServletRequest;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.*;
import java.util.stream.Collectors;
/**
 * <p>
@@ -50,20 +49,29 @@
    @Override
    @Transactional
    public Map<String, Object> updateRoleMenu(List list, int id) {
    public Map<String, Object> updateRoleMenu(List menuList, int id) {
        //补充父菜单
        List supplementList = new ArrayList();
        supplementList.addAll(menuList);
        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> list = new ArrayList<>(hashSet);
        //return null;
        Map<String,Object> resultMap = new HashMap<>();
        List<ManageMenu> manageMenusOld = manageMenuMapper.getManageMenuByRoleId(id);
        if (list.size()==0){
            resultMap.put("code",ResponseCodeEnum.PARAMETERS_IS_MISSING.getCode());
            resultMap.put("msg",ResponseCodeEnum.PARAMETERS_IS_MISSING.getMsg());
            return resultMap;
        }
        QueryWrapper<ManageRoleMenu> wrapper = new QueryWrapper();
        wrapper.eq("role_id",id);
        wrapper.eq("is_delete",0);
        List<Map> result = manageRoleMenuMapper.getDataByMenuIds(list,id);
        List<Map> result = new ArrayList<>();
        if (list.size()==0){
            result = result;
        }else {
            result = manageRoleMenuMapper.getDataByMenuIds(list,id);
        }
        List<Map> result1 = manageRoleMenuMapper.getDataByRoleId(id);
        if (manageRoleMenuMapper.selectCount(wrapper)==0 || result.size()==0){
        if ((manageRoleMenuMapper.selectCount(wrapper)==0 || result.size()==0)&&list.size()!=0){
            List<Map> insertList = new ArrayList<>();
            for (Object temp:list) {
                Map<String,Object> map = new HashMap<>();
@@ -71,48 +79,51 @@
                map.put("menu_id",Integer.parseInt(temp.toString()));
                insertList.add(map);
            }
            manageRoleMenuMapper.insertBatch(insertList);
        }
        List updateList = new ArrayList();
        for (Map manageRoleMenu:result1) {
            int have = 1;
            for (Object temp:list){
                if (manageRoleMenu.get("menu_id").equals(Integer.parseInt(temp.toString()))){
                    have = 2;
            if (insertList.size()!=0){
                manageRoleMenuMapper.insertBatch(insertList);
            }
        }else {
            List updateList = new ArrayList();
            for (Map manageRoleMenu:result1) {
                int have = 1;
                for (Object temp:list){
                    if (manageRoleMenu.get("menu_id").equals(Integer.parseInt(temp.toString()))){
                        have = 2;
                        continue;
                    }
                }
                if (have == 2){
                    continue;
                }
                updateList.add(manageRoleMenu.get("menu_id"));
            }
            if (have == 2){
                continue;
            }
            updateList.add(manageRoleMenu.get("menu_id"));
        }
        List insertList = new ArrayList();
        for (Object temp:list) {
            int have = 1;
            for (Map manageRoleMenu:result1){
                if (manageRoleMenu.get("menu_id").equals(Integer.parseInt(temp.toString()))){
                    have = 2;
            List insertList = new ArrayList();
            for (Object temp:list) {
                int have = 1;
                for (Map manageRoleMenu:result1){
                    if (manageRoleMenu.get("menu_id").equals(Integer.parseInt(temp.toString()))){
                        have = 2;
                        continue;
                    }
                }
                if (have == 2){
                    continue;
                }
                insertList.add(temp);
            }
            if (have == 2){
                continue;
            if (updateList.size()>0){
                manageRoleMenuMapper.updateDeleteStateByRoleIdMenuIds(updateList,id);
            }
            insertList.add(temp);
        }
        if (updateList.size()>0){
            manageRoleMenuMapper.updateDeleteStateByRoleIdMenuIds(updateList,id);
        }
        if (insertList.size()>0){
            List mapList = new ArrayList();
            for (Object temp:insertList) {
                Map<String,Integer> insertMap = new HashMap<>();
                insertMap.put("role_id",id);
                insertMap.put("menu_id",Integer.parseInt(temp.toString()));
                mapList.add(insertMap);
            if (insertList.size()>0){
                List mapList = new ArrayList();
                for (Object temp:insertList) {
                    Map<String,Integer> insertMap = new HashMap<>();
                    insertMap.put("role_id",id);
                    insertMap.put("menu_id",Integer.parseInt(temp.toString()));
                    mapList.add(insertMap);
                }
                manageRoleMenuMapper.insertBatch(mapList);
            }
            manageRoleMenuMapper.insertBatch(mapList);
        }
        List<ManageMenu> manageMenusNew = manageMenuMapper.getManageMenuByRoleId(id);
        List manageMenuOldList = new ArrayList();
@@ -140,4 +151,21 @@
        resultMap.put("msg",ResponseCodeEnum.SUCCESS.getMsg());
        return resultMap;
    }
    /**
     * @Description: 补充父菜单
     * @Param: [list, menuId]list:menuId集合
     * @return: void
     * @Author: 李子杰
     * @Date: 2021/4/7
     */
    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;
    }
}