用户菜单接口更新

whyczh

2021-05-14 b179439062ee4e5160dad7e553d5cdff8714d5fa

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

@@ -2,6 +2,7 @@

import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.toolkit.ObjectUtils;
import com.baomidou.mybatisplus.extension.api.R;
import com.google.common.collect.Sets;
import com.whyc.dto.Response;
import com.whyc.mapper.MenuMapper;
@@ -36,6 +37,23 @@
    public List<Menu> getAllMenus(){
        List<Menu> menus = menuMapper.selectList(null);
        return menus;
    }

    public Response updateRoleMenu(int roleId,int[] menuIds){
        QueryWrapper<RoleMenu> queryWrapper = new QueryWrapper<RoleMenu>().eq("role_id",roleId);
        mapper.delete(queryWrapper);
        List<RoleMenu> roleMenus = new ArrayList<>();
        for (int menuId:menuIds) {
            RoleMenu roleMenu = new RoleMenu();
            roleMenu.setRoleId(roleId);
            roleMenu.setMenuId(menuId);
            roleMenus.add(roleMenu);
        }
        if (mapper.insertBatchSomeColumn(roleMenus)==roleMenus.size()){
            return new Response().setMsg(1,"更新成功");
        }else {
            return new Response().setMsg(0,"更新失败");
        }
    }

    public List<Menu> getRoleMenu(Integer roleId){
@@ -142,18 +160,7 @@



    public void updateRoleMenu(int roleId,List<Menu> menus){
        QueryWrapper<RoleMenu> queryWrapper = new QueryWrapper<RoleMenu>().eq("role_id",roleId);
        mapper.delete(queryWrapper);
        for (Menu menu:menus) {
            if(menu.isSelected()){
                RoleMenu roleMenu = new RoleMenu();
                roleMenu.setMenuId(menu.getId());
                roleMenu.setRoleId(roleId);
                mapper.insert(roleMenu);
            }
        }
    }