学生体育馆预约系统vue2+springboot+mybatis+mysql

package com.management.gymnasium.controller; import com.management.gymnasium.pojo.Menu; import com.management.gymnasium.pojo.RoleMenu; import com.management.gymnasium.pojo.User; import com.management.gymnasium.repository.MenuRepository; import com.management.gymnasium.repository.RoleMenuRepository; import com.management.gymnasium.utils.LayuiResult; import com.management.gymnasium.utils.Pager; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.web.bind.annotation.PathVariable; import org.springframework.web.bind.annotation.RequestBody; import org.springframework.web.bind.annotation.RequestMapping; import org.springframework.web.bind.annotation.RestController; import javax.servlet.http.HttpServletRequest; import java.util.ArrayList; import java.util.HashMap; import java.util.List; import java.util.Map; @RestController @RequestMapping("/menu") public class MenuController { @Autowired private RoleMenuRepository roleMenuRepository; @Autowired private MenuRepository menuRepository; @RequestMapping("/getByRoleId") public Object getByRole(HttpServletRequest request){ User user = (User) request.getSession().getAttribute("user"); user = new User(); user.setRoleId(1); return LayuiResult.LayuiData(0,"success",getMenuByRole(user.getRoleId())); } @RequestMapping("/getByUserRoleMenu/{id}") public Object getByUserRoleMenu(@PathVariable Integer id){ return LayuiResult.LayuiData(0,"success",getMenuByRole(id)); } @RequestMapping("/get") public Object get(){ List<Menu> menus = menuRepository.findAllByParentId(0); List<Object> list = new ArrayList<>(); for (Menu menu : menus) { Map<String,Object> map = new HashMap<>(); map.put("id",menu.getId()); map.put("menuName",menu.getMenuName()); map.put("menuUrl",menu.getMenuUrl()); List<Menu> menus1 = menuRepository.findAllByParentId(menu.getId()); map.put("child",menus1); list.add(map); } return LayuiResult.LayuiData(0,"请求成功",list); } @RequestMapping("/get/list") public Object getList(Integer page,Integer limit){ List<Menu> menus = menuRepository.findAll(); Pager<Menu> pager = new Pager<>(); pager.setCurentPageIndex(page); pager.setCountPerpage(limit); pager.setBigList(menus); return LayuiResult.LayuiData(0,"请求成功",pager.getSmallList(),pager.getRecordCount()); } @RequestMapping("/get/parent") public Object getParentMenu(){ List<Menu> menuList = menuRepository.findAllByParentId(0); return LayuiResult.LayuiData(0,"success",menuList); } @RequestMapping("/get/child/{id}") public Object getChildMenuByParentId(@PathVariable Integer id){ List<Menu> menuList = menuRepository.findAllByParentId(id); return LayuiResult.LayuiData(0,"success",menuList); } @RequestMapping("/update") public Object update(@RequestBody Menu menu){ menuRepository.save(menu); return LayuiResult.LayuiData(0,"菜单信息修改成功",null); } @RequestMapping("/delete/{id}") public Object delete(@PathVariable Integer id){ menuRepository.deleteById(id); return LayuiResult.LayuiData(0,"菜单信息已删除",null); } @RequestMapping("/add") public Object add(@RequestBody Menu menu){ menuRepository.save(menu); return LayuiResult.LayuiData(0,"新菜单添加成功",null); } /** * * @param id 角色id * @return */ public Object getMenuByRole(Integer id){ List<RoleMenu> roleMenus1 = roleMenuRepository.findAllByRoleId(id); List<Menu> menuParent = new ArrayList<>(); roleMenus1.forEach(roleMenu -> { Menu menu = menuRepository.findById(roleMenu.getMenuId()).get(); if(menu.getParentId() == 0){ menuParent.add(menu); } }); List<Map<String, Object>> listResult = new ArrayList<>(); menuParent.forEach(menu -> { Map<String,Object> map = new HashMap<>(); map.put("title",menu.getMenuName()); map.put("menuName",menu.getMenuName()); map.put("menuUrl",menu.getMenuUrl()); map.put("id",menu.getId()); List<Menu> menuChild = menuRepository.findAllByParentId(menu.getId()); List<Integer> list = new ArrayList<>(); menuChild.forEach(menuInt -> { list.add(menuInt.getId()); }); List<RoleMenu> roleMenus = roleMenuRepository.findAllByRoleIdAndMenuIdIn(id,list); List<Map<String,Object>> listChildMap = new ArrayList<>(); roleMenus.forEach(childRoleMenu -> { Map<String,Object> childMap = new HashMap<>(); Menu menuChildObj = menuRepository.findById(childRoleMenu.getMenuId()).get(); childMap.put("title",menuChildObj.getMenuName()); childMap.put("menuName",menuChildObj.getMenuName()); childMap.put("menuUrl",menuChildObj.getMenuUrl()); childMap.put("id",menuChildObj.getId()); listChildMap.add(childMap); }); map.put("children",listChildMap); listResult.add(map); }); return listResult; } }