package com.zt.modules.sys.service;
|
|
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
|
import com.zt.common.constant.Constant;
|
import com.zt.common.exception.ErrorCode;
|
import com.zt.common.exception.RenException;
|
import com.zt.common.service.BaseService;
|
import com.zt.common.utils.CacheUtils;
|
import com.zt.common.utils.TreeUtils;
|
import com.zt.core.context.User;
|
import com.zt.core.context.UserContext;
|
import com.zt.modules.sys.dao.SysMenuDao;
|
import com.zt.modules.sys.enums.MenuType;
|
import com.zt.modules.sys.model.SysMenu;
|
import com.zt.modules.sys.model.SysRole;
|
import com.zt.modules.sys.model.SysRoleMenu;
|
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.List;
|
import java.util.stream.Collectors;
|
|
/**
|
* 菜单管理
|
*
|
* @author hehz
|
*/
|
@Service
|
public class SysMenuService extends BaseService<SysMenuDao, SysMenu> {
|
@Autowired
|
private SysRoleMenuService sysRoleMenuService;
|
@Autowired
|
private SysRoleService sysRoleService;
|
|
public List<SysMenu> getAllList() {
|
List<SysMenu> menuList = baseDao.selectList(new QueryWrapper<SysMenu>().eq(Constant.TableColumn.IS_DELETE,
|
Constant.Bool.NO).orderByAsc(Constant.TableColumn.SORT));
|
return menuList;
|
}
|
|
public List<SysMenu> getAllTree() {
|
List<SysMenu> menuList = baseDao.selectList(new QueryWrapper<SysMenu>().eq(Constant.TableColumn.IS_DELETE,
|
Constant.Bool.NO).orderByAsc(Constant.TableColumn.SORT));
|
return TreeUtils.build(menuList);
|
}
|
|
public List<SysMenu> getAllTreeWithoutButtons() {
|
List<SysMenu> menuList = baseDao.selectList(new QueryWrapper<SysMenu>()
|
.eq(Constant.TableColumn.IS_DELETE, Constant.Bool.NO).lambda()
|
.ne(SysMenu::getType, MenuType.BUTTON.getValue()).orderByAsc(SysMenu::getSort));
|
return TreeUtils.build(menuList);
|
}
|
|
/**
|
* 获取所有系统菜单
|
*
|
* @return
|
*/
|
public List<SysMenu> getAllSystemTree() {
|
return getAllTree().stream().filter(menu -> menu.getType() == MenuType.SYSTEM.getValue())
|
.collect(Collectors.toList());
|
}
|
|
public List<SysMenu> getSystems(Integer type) {
|
User user = UserContext.getUser();
|
List<SysMenu> list = baseDao.getSystems(type, user.getId());
|
return list;
|
}
|
|
|
/**
|
* 当前用户菜单导航，不包括按钮
|
*
|
* @return
|
*/
|
public List<SysMenu> getCurrentUserMenuNav() {
|
User user = UserContext.getUser();
|
// 超级管理员，拥有最高权限
|
if (user.isSuperAdmin() && user.isSuperAdmin()) {
|
return getAllTreeWithoutButtons().stream().filter(menu -> menu.getType() == MenuType.PLATFORM.getValue())
|
.collect(Collectors.toList());
|
}
|
List<SysMenu> list;
|
if ("wxxt".equals(user.getSystemMarker())) {
|
list = baseDao.getUserMenuListComm("comm_zts");
|
}else{
|
list = baseDao.getUserMenuList(user.getTenantId(), user.getId());
|
String roleName = user.getRoleName();
|
//if (list.size() == 0 && ("smj".equals(user.getSystemMarker())
|
if ((roleName == null || "".endsWith(roleName)) && ("smj".equals(user.getSystemMarker())
|
|| "djxl".equals(user.getSystemMarker()))
|
) {
|
list = baseDao.getUserMenuListComm("comm_sy");
|
}
|
}
|
|
List<SysMenu> list_except = baseDao.getUserMenuListExcept(user.getTenantId(), user.getId());
|
for (SysMenu item : list_except) {
|
List<SysMenu> list_except2 = list.stream().filter(menu -> menu.getId().equals(item.getId()))
|
.collect(Collectors.toList());
|
for (SysMenu item2 : list_except2) {
|
item2.setShowMenu(0);
|
}
|
}
|
List<SysMenu> list2 = TreeUtils.build(list.stream().filter(menu -> !menu.getType().equals(MenuType.BUTTON.getValue()))
|
.collect(Collectors.toList()));
|
|
List<SysMenu> listOut = new ArrayList<>();
|
for (SysMenu item : list2) {
|
if ("smj".equals(user.getSystemMarker()) && (1283299421357899778L == item.getId() || 1582665097052971009L == item.getId())) {
|
//选择寿命件系统隐藏等级修理系统，维修系统， 1283299421357899778,1582665097052971009L
|
item.setShowMenu(0);
|
//continue;
|
}
|
if ("djxl".equals(user.getSystemMarker()) && (1471651415041593346L == item.getId() || 1582665097052971009L == item.getId())) {
|
//选择等级维修系统隐藏寿命件系统，维修系统， 1471651415041593346,1582665097052971009L
|
item.setShowMenu(0);
|
//continue;
|
}
|
/*
|
if ("wxxt".equals(user.getSystemMarker()) && (1283299421357899778L == item.getId() || 1471651415041593346L == item.getId())) {
|
//选择寿命件系统隐藏寿命件系统，等级修理系统， 1283299421357899778,1471651415041593346L
|
item.setShowMenu(0);
|
//continue;
|
}*/
|
listOut.add(item);
|
}
|
return listOut;
|
}
|
|
/**
|
* 当前用户可以看到的菜单列表，包括按钮
|
*
|
* @return
|
*/
|
public List<SysMenu> getCurrentUserMenu() {
|
User user = UserContext.getUser();
|
// 超级管理员，拥有最高权限
|
if (user.isPlatform() && user.isSuperAdmin()) {
|
return getAllTree().stream().filter(menu -> menu.getType() == MenuType.PLATFORM.getValue())
|
.collect(Collectors.toList());
|
}
|
|
List<SysMenu> list = baseDao.getUserMenuList(user.getTenantId(), null);
|
return TreeUtils.build(list);
|
}
|
|
@Override
|
@Transactional(rollbackFor = Exception.class)
|
public void insert(SysMenu entity) {
|
super.insert(entity);
|
CacheUtils.removeAll(Constant.Cache.TOKEN);// 清理用户权限缓存
|
/* SysRoleMenu sysRoleMenu = new SysRoleMenu();
|
sysRoleMenu.setMenuId(entity.getId());
|
sysRoleMenu.setRoleId(1271368584383188993L);
|
sysRoleMenu.setDelete(true);
|
sysRoleMenuService.insert(sysRoleMenu);
|
|
List<SysRole> allList = sysRoleService.getAllList();
|
for(SysRole sysRole:allList){
|
sysRoleMenu = new SysRoleMenu();
|
sysRoleMenu.setMenuId(entity.getId());
|
sysRoleMenu.setRoleId(sysRole.getId());
|
sysRoleMenu.setDelete(true);
|
sysRoleMenuService.insert(sysRoleMenu);
|
}*/
|
}
|
|
@Transactional(rollbackFor = Exception.class)
|
public void update(SysMenu menu) {
|
SysMenu tmp = baseDao.selectById(menu.getId());
|
if (tmp == null) {
|
throw new RenException("查询菜单信息失败");
|
}
|
List<SysMenu> all = baseDao.selectList(new QueryWrapper<SysMenu>().lambda().ne(SysMenu::getType,
|
MenuType.BUTTON.getValue()));
|
TreeUtils.updateValidate(all, menu, tmp);
|
|
// 更新菜单
|
super.update(menu);
|
CacheUtils.removeAll(Constant.Cache.TOKEN);// 清理用户权限缓存
|
}
|
|
@Transactional(rollbackFor = Exception.class)
|
public void delete(Long id) {
|
// 判断是否有子菜单或按钮
|
if (baseDao.selectCount(new QueryWrapper<SysMenu>().eq(Constant.TableColumn.IS_DELETE, Constant.Bool.NO)
|
.lambda().eq(SysMenu::getPid, id)) > 0) {
|
throw new RenException(ErrorCode.SUB_MENU_EXIST);
|
}
|
// 删除菜单
|
deleteLogic(id);
|
|
// 删除角色菜单关系
|
sysRoleMenuService.deleteByMenuId(id);
|
|
CacheUtils.removeAll(Constant.Cache.TOKEN);// 清理用户权限缓存
|
}
|
}
|
