基于SSM框架的健身俱乐部管理系统设计源码

package com.ssm.healthclub.service.impl; import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; import com.baomidou.mybatisplus.core.metadata.IPage; import com.baomidou.mybatisplus.extension.plugins.pagination.Page; import com.ssm.healthclub.pojo.Menu; import com.ssm.healthclub.mapper.MenuMapper; import com.ssm.healthclub.pojo.RoleMenu; import com.ssm.healthclub.pojo.Subject; import com.ssm.healthclub.service.IMenuService; import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; import com.ssm.healthclub.service.IRoleMenuService; import com.ssm.healthclub.utils.R; import com.ssm.healthclub.utils.ResultCode; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.stereotype.Service; import org.springframework.transaction.annotation.Transactional; import java.util.*; /** * <p> * 服务实现类 * </p> * * @author gapCheung * @since 2023-05-29 */ @Service public class MenuServiceImpl extends ServiceImpl<MenuMapper, Menu> implements IMenuService { @Autowired private MenuMapper menuMapper; @Autowired private IRoleMenuService roleMenuService; /** * 分页查询菜单 * * @param title * @param pageNumber * @param pageSize * @return */ @Override public R queryPage(String title, Integer pageNumber, Integer pageSize) { Map<String, Object> map = new HashMap<>(); QueryWrapper<Menu> queryWrapper = new QueryWrapper<>(); queryWrapper.like(title != null && !"".equals(title), "title", title); queryWrapper.eq("isDel", 0).eq("type", 0); Page page = new Page(pageNumber, pageSize); IPage<Subject> ipage = menuMapper.selectPage(page, queryWrapper); map.put("total", ipage.getTotal()); map.put("rows", ipage.getRecords()); return R.success(ResultCode.SUCCESS, map); } /** * 通过用户id查询菜单列表 * * @param userId * @return */ @Override public List<Menu> listMenusByUserId(Integer userId) { return menuMapper.listMenusByUserId(userId); } /** * 分页查询权限 * * @param title * @param pageNumber * @param pageSize * @return */ @Override public R queryPagePermission(String title, Integer pageNumber, Integer pageSize) { Map<String, Object> map = new HashMap<>(); QueryWrapper<Menu> queryWrapper = new QueryWrapper<>(); queryWrapper.like(title != null && !"".equals(title), "title", title); queryWrapper.eq("isDel", 0).eq("type", 1); Page page = new Page(pageNumber, pageSize); IPage<Subject> ipage = menuMapper.selectPage(page, queryWrapper); map.put("total", ipage.getTotal()); map.put("rows", ipage.getRecords()); return R.success(ResultCode.SUCCESS, map); } /** * 更新角色的菜单和权限 * * @param ids * @param roleId * @return */ @Override @Transactional public R updMenuPermission(String ids, String roleId) { try { // 1.更新tb_menu中的remark字段 List<String> menuIdsList = new ArrayList<>(); List<Menu> menuList = menuMapper.selectList(new QueryWrapper<Menu>().eq("isDel", 0)); for (Menu menu : menuList) { menuIdsList.add(String.valueOf(menu.getId())); } // 传过来的id数组查询数据库是否已存在该权限,如果存在不改,不存在修改 String[] roleIds = ids.split(","); List<String> roleIdsList = new ArrayList<>(); for (String id : roleIds) { roleIdsList.add(id); Menu menu = menuMapper.selectById(id); if (menu.getRemark().contains(roleId)) { } else { // 赋予权限 menu.setRemark(menu.getRemark() + "," + roleId); menuMapper.update(menu, new QueryWrapper<Menu>().eq("type", 1).eq("id", menu.getId())); } } // 未传过来的id数组说明不赋予权限,如果之前有,则取消权限 List<String> different = getDifferent(menuIdsList, roleIdsList); for (String diffId : different) { Menu menu = menuMapper.selectById(diffId); if (menu.getRemark().contains(roleId)) { // 删除权限 menu.setRemark(menu.getRemark().replace(("," + roleId), "")); menuMapper.update(menu, new QueryWrapper<Menu>().eq("type", 1).eq("id", menu.getId())); } } // 2.更新角色菜单中间表 // 先删除原有表中的数据 roleMenuService.remove(new QueryWrapper<RoleMenu>().eq("roleId", roleId)); // 重新新增数据 List<RoleMenu> roleMenuList = new ArrayList<>(); for (String menuId : roleIds) { RoleMenu roleMenu = new RoleMenu(null, Integer.valueOf(roleId), Integer.valueOf(menuId)); roleMenuList.add(roleMenu); } roleMenuService.saveBatch(roleMenuList); return R.success(ResultCode.SUCCESS); } catch (Exception e) { e.printStackTrace(); return R.success(ResultCode.FAIL); } } /** * 通过角色id查询对应的菜单和权限 * * @param roleId * @return */ @Override public R showMenuPermission(String roleId) { QueryWrapper<Menu> menuQuery = new QueryWrapper<Menu>().eq("type", 0).eq("isDel", 0).eq("parentId", 0); QueryWrapper<Menu> menuChildQuery = new QueryWrapper<Menu>().eq("type", 0).eq("isDel", 0).ne("parentId", 0); QueryWrapper<Menu> perQuery = new QueryWrapper<Menu>().eq("type", 1).eq("isDel", 0); List<Menu> menuList = menuMapper.selectList(menuQuery); List<Menu> menuChildList = menuMapper.selectList(menuChildQuery); List<Menu> permissionList = menuMapper.selectList(perQuery); for (Menu menu : menuList) { List<Menu> arr = new ArrayList<>(); for (String roleIds : menu.getRemark().split(",")) { if (roleIds.equals(roleId)) { menu.setChecked(true); } } for (Menu childMenu : menuChildList) { List<Menu> ss = new ArrayList<>(); for (String roleIds : childMenu.getRemark().split(",")) { if (roleIds.equals(roleId)) { childMenu.setChecked(true); } } if (childMenu.getParentId() == menu.getId()) { arr.add(childMenu); menu.setChildren(arr); for (Menu permission : permissionList) { if (permission.getParentId() == childMenu.getId()) { ss.add(permission); childMenu.setChildren(ss); } for (String roleIds : permission.getRemark().split(",")) { if (roleIds.equals(roleId)) { permission.setChecked(true); } } } } } } return R.success(ResultCode.SUCCESS, menuList); } /** * 从两个集合中取出不同的值 * * @param list1 * @param list2 * @return */ private static List<String> getDifferent(List<String> list1, List<String> list2) { List<String> diff = new ArrayList<>(); Map<String, Integer> map = new HashMap<>(list1.size() + list2.size()); List<String> maxList = list1; List<String> minList = list2; if (list2.size() > list1.size()) { maxList = list2; minList = list1; }