package cn.stylefeng.guns.sys.modular.system.service; import cn.hutool.core.bean.BeanUtil; import cn.hutool.core.util.StrUtil; import cn.stylefeng.guns.base.pojo.node.MenuNode; import cn.stylefeng.guns.base.pojo.node.ZTreeNode; import cn.stylefeng.guns.base.pojo.page.LayuiPageFactory; import cn.stylefeng.guns.sys.core.constant.factory.ConstantFactory; import cn.stylefeng.guns.sys.core.constant.state.MenuStatus; import cn.stylefeng.guns.sys.core.exception.enums.BizExceptionEnum; import cn.stylefeng.guns.sys.core.listener.ConfigListener; import cn.stylefeng.guns.base.i18n.context.UserTranslationContext; import cn.stylefeng.guns.base.i18n.enums.TranslationEnum; import cn.stylefeng.guns.sys.modular.system.entity.Menu; import cn.stylefeng.guns.sys.modular.system.mapper.MenuMapper; import cn.stylefeng.guns.sys.modular.system.model.MenuDto; import cn.stylefeng.roses.core.util.ToolUtil; import cn.stylefeng.roses.kernel.model.exception.RequestEmptyException; import cn.stylefeng.roses.kernel.model.exception.ServiceException; import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; import com.baomidou.mybatisplus.extension.plugins.pagination.Page; import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; import org.springframework.stereotype.Service; import org.springframework.transaction.annotation.Transactional; import javax.annotation.Resource; import java.util.ArrayList; import java.util.List; import java.util.Map; /** *

* 菜单表 服务实现类 *

* * @author stylefeng * @since 2018-12-07 */ @Service public class MenuService extends ServiceImpl { @Resource private MenuMapper menuMapper; /** * 添加菜单 * * @author fengshuonan * @Date 2018/12/23 5:59 PM */ @Transactional public void addMenu(MenuDto menuDto) { if (ToolUtil.isOneEmpty(menuDto, menuDto.getCode(), menuDto.getName(), menuDto.getPid(), menuDto.getMenuFlag(), menuDto.getUrl(), menuDto.getSystemType())) { throw new RequestEmptyException(); } //判断是否已经存在该编号 String existedMenuName = ConstantFactory.me().getMenuNameByCode(menuDto.getCode()); if (ToolUtil.isNotEmpty(existedMenuName)) { throw new ServiceException(BizExceptionEnum.EXISTED_THE_MENU); } //组装属性,设置父级菜单编号 Menu resultMenu = this.menuSetPcode(menuDto); resultMenu.setStatus(MenuStatus.ENABLE.getCode()); this.save(resultMenu); } /** * 更新菜单 * * @author fengshuonan * @Date 2019/2/27 4:09 PM */ @Transactional(rollbackFor = Exception.class) public void updateMenu(MenuDto menuDto) { //如果菜单为空 if (menuDto == null || ToolUtil.isOneEmpty(menuDto.getMenuId(), menuDto.getCode())) { throw new RequestEmptyException(); } //获取旧的菜单 Long id = menuDto.getMenuId(); Menu menu = this.getById(id); if (menu == null) { throw new RequestEmptyException(); } //设置父级菜单编号 Menu resultMenu = this.menuSetPcode(menuDto); //查找该节点的子集合,并修改相应的pcodes和level(因为修改菜单后,层级可能变化了) updateSubMenuLevels(menu, resultMenu); this.updateById(resultMenu); } /** * 更新所有子菜单的结构 * * @param oldMenu 原来的菜单 * @param newMenu 新菜单 * @author fengshuonan * @Date 2019/2/27 4:25 PM */ @Transactional(rollbackFor = Exception.class) public void updateSubMenuLevels(Menu oldMenu, Menu newMenu) { List menus = menuMapper.getMenusLikePcodes(oldMenu.getCode()); for (Menu menu : menus) { //更新pcode if (oldMenu.getCode().equals(menu.getPcode())) { menu.setPcode(newMenu.getCode()); } //更新pcodes String oldPcodesPrefix = oldMenu.getPcodes() + "[" + oldMenu.getCode() + "],"; String oldPcodesSuffix = menu.getPcodes().substring(oldPcodesPrefix.length()); String menuPcodes = newMenu.getPcodes() + "[" + newMenu.getCode() + "]," + oldPcodesSuffix; menu.setPcodes(menuPcodes); //更新levels int level = StrUtil.count(menuPcodes, "["); menu.setLevels(level); //更新systemType menu.setSystemType(newMenu.getSystemType()); this.updateById(menu); } } /** * 删除菜单 * * @author stylefeng * @Date 2017/5/5 22:20 */ @Transactional public void delMenu(Long menuId) { //删除菜单 this.menuMapper.deleteById(menuId); //删除关联的relation this.menuMapper.deleteRelationByMenu(menuId); } /** * 删除菜单包含所有子菜单 * * @author stylefeng * @Date 2017/6/13 22:02 */ @Transactional public void delMenuContainSubMenus(Long menuId) { Menu menu = menuMapper.selectById(menuId); //删除当前菜单 delMenu(menuId); //删除所有子菜单 List menus = menuMapper.getMenusLikePcodes(menu.getCode()); for (Menu temp : menus) { delMenu(temp.getMenuId()); } } /** * 根据条件查询菜单 * * @return * @date 2017年2月12日 下午9:14:34 */ public Page> selectMenus(String condition, String level, Long menuId) { //获取menuId的code String code = ""; if (menuId != null && menuId != 0L) { Menu menu = this.getById(menuId); code = menu.getCode(); } Page page = LayuiPageFactory.defaultPage(); return this.baseMapper.selectMenus(page, condition, level, menuId, code); } /** * 根据条件查询菜单 * * @return * @date 2017年2月12日 下午9:14:34 */ public List getMenuIdsByRoleId(Long roleId) { return this.baseMapper.getMenuIdsByRoleId(roleId); } /** * 获取菜单列表树 * * @return * @date 2017年2月19日 下午1:33:51 */ public List menuTreeList() { return this.baseMapper.menuTreeList(); } /** * 获取菜单列表树 * * @return * @date 2017年2月19日 下午1:33:51 */ public List menuTreeListByMenuIds(List menuIds) { return this.baseMapper.menuTreeListByMenuIds(menuIds); } /** * 删除menu关联的relation * * @param menuId * @return * @date 2017年2月19日 下午4:10:59 */ public int deleteRelationByMenu(Long menuId) { return this.baseMapper.deleteRelationByMenu(menuId); } /** * 获取资源url通过角色id * * @param roleId * @return * @date 2017年2月19日 下午7:12:38 */ public List getResUrlsByRoleId(Long roleId) { return this.baseMapper.getResUrlsByRoleId(roleId); } /** * 根据角色获取菜单 * * @param roleIds * @return * @date 2017年2月19日 下午10:35:40 */ public List getMenusByRoleIds(List roleIds) { List menus = this.baseMapper.getMenusByRoleIds(roleIds); //给所有的菜单url加上ctxPath for (MenuNode menuItem : menus) { menuItem.setUrl(ConfigListener.getConf().get("contextPath") + menuItem.getUrl()); //如果是当前用户选择语言是中文,则返回菜单名字,而不是字典名字 if (UserTranslationContext.getUserCurrentTrans().equals(TranslationEnum.CHINESE)) { continue; } else { //将菜单进行多语言翻译 String transName = UserTranslationContext.get("MENU_" + menuItem.getCode().toUpperCase()); menuItem.setName(transName); } } return menus; } /** * 根据code查询菜单 * * @author fengshuonan * @Date 2018/12/20 21:54 */ public Menu selectByCode(String code) { Menu menu = new Menu(); menu.setCode(code); QueryWrapper queryWrapper = new QueryWrapper<>(menu); return this.baseMapper.selectOne(queryWrapper); } /** * 根据请求的父级菜单编号设置pcode和层级 * * @author fengshuonan * @Date 2018/12/23 5:54 PM */ public Menu menuSetPcode(MenuDto menuParam) { Menu resultMenu = new Menu(); BeanUtil.copyProperties(menuParam, resultMenu); if (ToolUtil.isEmpty(menuParam.getPid()) || menuParam.getPid().equals(0L)) { resultMenu.setPcode("0"); resultMenu.setPcodes("[0],"); resultMenu.setLevels(1); } else { Long pid = menuParam.getPid(); Menu pMenu = this.getById(pid); Integer pLevels = pMenu.getLevels(); resultMenu.setPcode(pMenu.getCode()); //如果编号和父编号一致会导致无限递归 if (menuParam.getCode().equals(menuParam.getPcode())) { throw new ServiceException(BizExceptionEnum.MENU_PCODE_COINCIDENCE); } resultMenu.setLevels(pLevels + 1); resultMenu.setPcodes(pMenu.getPcodes() + "[" + pMenu.getCode() + "],"); } return resultMenu; } /** * 获取菜单树形列表 * * @author fengshuonan * @Date 2019/2/23 22:02 */ public List> selectMenuTree(String condition, String level) { List> maps = this.baseMapper.selectMenuTree(condition, level); if (maps == null) { maps = new ArrayList<>(); } //修复菜单查询bug,带条件的暂时先父级置为0 if (ToolUtil.isNotEmpty(condition) || ToolUtil.isNotEmpty(level)) { if (maps.size() > 0) { //将pcode置为root for (Map menu : maps) { menu.put("pcode", "0"); } } } return maps; } }