io.github.wslxm.springbootplus2.manage.sys.service.impl.SysMenuServiceImpl Maven / Gradle / Ivy
The newest version!
package io.github.wslxm.springbootplus2.manage.sys.service.impl;
import cn.hutool.core.util.ObjectUtil;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import io.github.wslxm.springbootplus2.utils.JwtUtil;
import io.github.wslxm.springbootplus2.core.base.model.BasePage;
import io.github.wslxm.springbootplus2.core.base.service.impl.BaseServiceImpl;
import io.github.wslxm.springbootplus2.core.config.error.ErrorException;
import io.github.wslxm.springbootplus2.core.enums.BaseDic;
import io.github.wslxm.springbootplus2.core.result.ResultType;
import io.github.wslxm.springbootplus2.core.utils.XjBeanUtil;
import io.github.wslxm.springbootplus2.manage.sys.mapper.SysMenuMapper;
import io.github.wslxm.springbootplus2.manage.sys.mapper.SysRoleMenuMapper;
import io.github.wslxm.springbootplus2.manage.sys.model.dto.SysMenuDTO;
import io.github.wslxm.springbootplus2.manage.sys.model.entity.SysMenu;
import io.github.wslxm.springbootplus2.manage.sys.model.entity.SysRole;
import io.github.wslxm.springbootplus2.manage.sys.model.entity.SysRoleMenu;
import io.github.wslxm.springbootplus2.manage.sys.model.query.SysMenuQuery;
import io.github.wslxm.springbootplus2.manage.sys.model.vo.SysMenuVO;
import io.github.wslxm.springbootplus2.manage.sys.service.SysMenuService;
import io.github.wslxm.springbootplus2.manage.sys.service.SysRoleMenuService;
import io.github.wslxm.springbootplus2.manage.sys.service.SysRoleService;
import org.apache.commons.lang3.StringUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
import java.util.ArrayList;
import java.util.LinkedList;
import java.util.List;
import java.util.stream.Collectors;
/**
* @author wangsong
*/
@Service
public class SysMenuServiceImpl extends BaseServiceImpl implements SysMenuService {
@Autowired
private SysRoleMenuMapper roleMenuMapper;
@Autowired
private SysRoleService roleService;
@Autowired
private SysRoleMenuService roleMenuService;
@Override
public List tree(SysMenuQuery query) {
boolean isNextAll = ObjectUtil.defaultIfNull(query.getIsNextAll(), true);
String pId = query.getPid();
String roleId = query.getRoleId();
Integer root = query.getRoot();
Integer disable = query.getDisable();
// 判断是否只查询当前登录人存在的菜单
String loginUserId = ObjectUtil.defaultIfNull(query.getIsLoginUser(), () -> JwtUtil.getJwtUser(request).getUserId(), null);
// 1、查询菜单
List menuVOList = baseMapper.list(loginUserId, disable);
// 2、获取角色拥有的菜单id(没有角色id或没有 角色对应的菜单数据,创建空roleMenuIdList对象)
List userRoleMenus = roleId != null ? roleMenuMapper.findRoleId(roleId) : new ArrayList<>();
List roleMenuIdList = !userRoleMenus.isEmpty() ? userRoleMenus.stream().map(SysRoleMenu::getMenuId).collect(Collectors.toList()) : new ArrayList<>();
// 4、根据级别过滤(1-目录 2-菜单 3-页面)
if (root != null) {
menuVOList = menuVOList.stream().filter(p -> (p.getRoot() <= root)).collect(Collectors.toList());
}
// 增加path字段数据(vue路由)
for (SysMenuVO menuVo : menuVOList) {
if (StringUtils.isNotBlank(menuVo.getUrl())) {
String[] urls = menuVo.getUrl().split("/");
StringBuilder urlsStr = new StringBuilder();
for (int i = 1; i < urls.length; i++) {
urlsStr.append(urls[i]);
}
menuVo.setPathx(urlsStr.toString());
}
}
// 5、判断是否只查询指定pid 数据
List pMenuListVO = new ArrayList<>();
if (StringUtils.isNotBlank(pId)) {
// 5.1、找到父级code数据(只有一条)
for (SysMenuVO p : menuVOList) {
if (p.getId().equals(pId)) {
pMenuListVO.add(p);
break;
}
}
} else {
// 5.2、如果没有, 设置为父级字典数据为顶级pid=0的数据 (可能多条)
for (SysMenuVO p : menuVOList) {
if (p.getRoot().equals(BaseDic.MenuRoot.V1.getValue())) {
pMenuListVO.add(p);
}
}
}
// 没有找顶级id数据 或 没有菜单数据,返回空
if (pMenuListVO.isEmpty()) {
return pMenuListVO;
}
// 6、递归获取 子级数据, pDictListVO 为tree数据(tree), diceIds为指定code层级下所有字典id收集(返回 list 需要)
List menuIds = new ArrayList<>();
// 返回 tree (递归处理)
for (SysMenuVO pMenuVo : pMenuListVO) {
menuIds.add(pMenuVo.getId());
this.setChecked(pMenuVo, roleMenuIdList);
if (isNextAll) {
this.nextLowerIdNodeTreeChecked(menuVOList, pMenuVo, roleMenuIdList, menuIds, null);
} else {
this.nextLowerIdNodeTreeChecked(menuVOList, pMenuVo, roleMenuIdList, menuIds, 1);
}
}
// 7、返回
return pMenuListVO;
}
@Override
public BasePage findPage(SysMenuQuery query) {
LambdaQueryWrapper queryWrapper = new LambdaQueryWrapper()
.orderByAsc(SysMenu::getSort)
.orderByDesc(SysMenu::getCreateTime);
queryWrapper.eq(StringUtils.isNotBlank(query.getPid()), SysMenu::getPid, query.getPid());
queryWrapper.likeRight(StringUtils.isNotBlank(query.getName()), SysMenu::getName, query.getName());
queryWrapper.eq(query.getRoot() != null, SysMenu::getRoot, query.getRoot());
queryWrapper.eq(query.getDisable() != null, SysMenu::getDisable, query.getDisable());
Page page = this.page(new Page<>(query.getCurrent(), query.getSize()), queryWrapper);
return XjBeanUtil.pageVo(page, SysMenuVO.class);
}
@Override
public SysMenuVO findId(String id) {
return XjBeanUtil.convert(this.getById(id), SysMenuVO.class);
}
@Override
@Transactional(rollbackFor = Exception.class)
public String insert(SysMenuDTO dto) {
SysMenu adminMenu = dto.convert(SysMenu.class);
adminMenu.setCreateUser(JwtUtil.getJwtUser(request).getUserId());
this.save(adminMenu);
// 添加菜单给超管默认分配该菜单
SysRole sysRole = roleService.findSysRole();
roleMenuService.addRoleMenu(sysRole.getId(), adminMenu.getId());
return adminMenu.getId();
}
@Override
public boolean upd(String id, SysMenuDTO dto) {
// 修改当前数据
SysMenu entity = dto.convert(SysMenu.class);
entity.setId(id);
return this.updateById(entity);
}
@Override
public boolean updPid(String id, String pid) {
// 把顶级菜单 变更到其他目录/菜单下
SysMenu menu = this.getById(id);
menu.setPid(pid);
if (menu.getRoot().equals(BaseDic.MenuRoot.V1.getValue())) {
menu.setRoot(BaseDic.MenuRoot.V2.getValue());
}
return this.updateById(menu);
}
@Override
@Transactional(rollbackFor = Exception.class)
public List del(String id) {
// 查询
List list = this.list();
List menuVos = XjBeanUtil.listVo(list, SysMenuVO.class);
//
List menuIds = new ArrayList<>();
SysMenuVO pMenuVo = new SysMenuVO();
pMenuVo.setId(id);
this.nextLowerIdNodeTreeChecked(menuVos, pMenuVo, new ArrayList<>(), menuIds, null);
// 删除菜单
menuIds.add(id);
this.removeByIds(menuIds);
// 删除角色菜单关联数据
roleMenuService.delBatchByMenuIds(menuIds);
return menuIds;
}
@Override
public List findTree() {
List userRoleMenus = roleMenuMapper.findByUserIdAndDisableFetchMenu(JwtUtil.getJwtUser(request).getUserId(), BaseDic.Whether.V0.getValue());
if (userRoleMenus == null || userRoleMenus.isEmpty()) {
throw new ErrorException(ResultType.USER_NO_MENU);
}
List roleMenuIdList = userRoleMenus.stream().map(SysRoleMenu::getMenuId).collect(Collectors.toList());
List adminMenuList = this.list(new LambdaQueryWrapper()
.orderByAsc(SysMenu::getSort)
.eq(SysMenu::getDisable, 0)
.in(SysMenu::getId, roleMenuIdList)
);
if (adminMenuList == null || adminMenuList.isEmpty()) {
throw new ErrorException(ResultType.USER_NO_MENU);
}
List menuVos = XjBeanUtil.listVoStream(adminMenuList, SysMenuVO.class);
// return
List menuList = new LinkedList<>();
menuVos.forEach(fatherMenuVo -> {
if (fatherMenuVo.getRoot() == 1) {
this.nextLowerNode(menuVos, fatherMenuVo, roleMenuIdList);
menuList.add(fatherMenuVo);
}
});
return menuList;
}
/**
* 菜单数遍历子节点(递归遍历)，不添加没有权限的数据 Tree
*
* @param menuVos 所有节点
* @param fatherMenuVo 当前菜单节点，遍历后得到的数据的父节点
* @param roleMenuList 当前用户存在的菜单权限ID
* @return void
* @date 2019/11/13 15:20
*/
private void nextLowerNode(List menuVos, SysMenuVO fatherMenuVo, List roleMenuList) {
menuVos.forEach(menuVo -> {
if (menuVo.getPid().equals(fatherMenuVo.getId()) && roleMenuList.contains(menuVo.getId())) {
if (fatherMenuVo.getMenus() == null) {
ArrayList nextMenuVos = new ArrayList<>();
nextMenuVos.add(menuVo);
fatherMenuVo.setMenus(nextMenuVos);
} else {
fatherMenuVo.getMenus().add(menuVo);
}
// 递归
this.nextLowerNode(menuVos, menuVo, roleMenuList);
}
});
}
/**
* 获取指定父节点下的子节点(递归遍历) 权限 isChecked = true||false Tree
*
* @param menuVos 所有节点
* @param pMenuVo 当前节点
* @param roleMenuIdList 选中角色权限
* @param ids 收集指定code下所有字典数据 id
* @param recursiveHierarchy 递归层级 1-获取下一级 2-获取下2级 ....以此类推, null不限层级直到没有下级
* @return void
* @date 2019/11/13 15:20
*/
private void nextLowerIdNodeTreeChecked(List menuVos, SysMenuVO pMenuVo, List roleMenuIdList, List ids, Integer recursiveHierarchy) {
// 递归层级控制
if (recursiveHierarchy != null) {
if (recursiveHierarchy == 0) {
return;
}
recursiveHierarchy--;
}
for (SysMenuVO zMenuVo : menuVos) {
if (zMenuVo.getPid().equals(pMenuVo.getId())) {
if (pMenuVo.getMenus() == null) {
ArrayList adminMenuVos = new ArrayList<>();
adminMenuVos.add(zMenuVo);
pMenuVo.setMenus(adminMenuVos);
} else {
pMenuVo.getMenus().add(zMenuVo);
}
// 获取ids
ids.add(zMenuVo.getId());
this.setChecked(zMenuVo, roleMenuIdList);
this.nextLowerIdNodeTreeChecked(menuVos, zMenuVo, roleMenuIdList, ids, recursiveHierarchy);
}
}
}
/**
* 判断并设置为选中状态
*/
private void setChecked(SysMenuVO menu, List roleMenuIdList) {
//权限判断设置选中状态
if (roleMenuIdList.contains(menu.getId())) {
menu.setIsChecked(true);
} else {
menu.setIsChecked(false);
}
}
}