更新角色,菜单接口

whycxzp

2021-09-24 b7226efbe9f5fdcfe33b152baf91eac6ab44b30a

 src/main/java/com/whyc/service/RoleMenuService.java

@@ -1,11 +1,16 @@
package com.whyc.service;

import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.whyc.dto.Response;
import com.whyc.mapper.RoleMenuMapper;
import com.whyc.pojo.Menu;
import com.whyc.pojo.RoleMenu;
import org.springframework.stereotype.Service;

import javax.annotation.Resource;
import java.util.List;
import java.util.*;
import java.util.stream.Collectors;

/**
 * 角色-菜单
@@ -19,4 +24,31 @@
    public List<Menu> getUserMenu(Integer id) {
        return mapper.getUserMenu(id);
    }

    public List<Menu> getMenuByRoleId(Integer roleId) {

        List<Menu> menuList = mapper.getMenuByRoleId(roleId);

        for (int i = 0; i < menuList.size(); i++) {
            Integer parentId = menuList.get(i).getParentId();
            if(parentId !=0){
                //将这个menu插入到parent中
                for (int j = 0; j < menuList.size(); j++) {
                    if(menuList.get(j).getId()==parentId){
                        List<Menu> menuListTemp = menuList.get(j).getMenuList();
                        if(menuListTemp==null) {
                            menuListTemp = new LinkedList<>();
                        }
                        menuListTemp.add(menuList.get(i));
                        menuList.get(j).setMenuList(menuListTemp);
                        menuList.remove(i);
                        i--;
                        break;
                    }
                }
            }

        }
        return menuList;
    }
}