package com.gkhy.safePlatform.account.service.impl;
|
import com.alibaba.fastjson.JSONObject;
|
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
|
import com.gkhy.safePlatform.account.model.dto.req.MenuMetaReqDTO;
|
import com.gkhy.safePlatform.account.model.dto.req.MenuModReqDTO;
|
import com.gkhy.safePlatform.account.entity.user.*;
|
import com.gkhy.safePlatform.account.enums.MenuStatusEnum;
|
|
import java.util.ArrayList;
|
import java.util.Date;
|
|
import com.gkhy.safePlatform.account.model.dto.req.MenuAddReqDTO;
|
import com.gkhy.safePlatform.account.service.MenuService;
|
import com.gkhy.safePlatform.account.service.RedisService;
|
import com.gkhy.safePlatform.account.service.baseService.*;
|
import com.gkhy.safePlatform.account.utils.RedisUtils;
|
import com.gkhy.safePlatform.account.utils.TokenUtil;
|
import com.gkhy.safePlatform.commons.co.ContextCacheUser;
|
import com.gkhy.safePlatform.commons.enums.E;
|
import com.gkhy.safePlatform.commons.enums.RedisKeyEnum;
|
import com.gkhy.safePlatform.commons.enums.ResultCodes;
|
import com.gkhy.safePlatform.commons.exception.AusinessException;
|
import com.gkhy.safePlatform.commons.exception.BusinessException;
|
import com.gkhy.safePlatform.commons.utils.StringUtils;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.stereotype.Service;
|
import org.springframework.transaction.annotation.Transactional;
|
|
import java.util.List;
|
import java.util.concurrent.TimeUnit;
|
|
@Service("menuService")
|
public class MenuServiceImpl implements MenuService {
|
|
|
@Autowired
|
private MenuInfoService menuInfoService;
|
@Autowired
|
private ProjectInfoService projectInfoService;
|
@Autowired
|
private RoleInfoService roleInfoService;
|
@Autowired
|
private ProjectMenuInfoService projectMenuInfoService;
|
@Autowired
|
private RoleProjectInfoService roleProjectInfoService;
|
@Autowired
|
private RoleMenuInfoService roleMenuInfoService;
|
@Autowired
|
private TokenUtil tokenConfig;
|
@Autowired
|
private RedisService redisService;
|
|
@Override
|
@Transactional
|
public void addMenu(MenuAddReqDTO menuAddDto, ContextCacheUser currentUser) {
|
|
MenuMetaReqDTO meta = menuAddDto.getMeta();
|
if (meta == null) {
|
throw new AusinessException(E.DATA_PARAM_NULL,"属性为空");
|
}
|
if (StringUtils.isBlank(meta.getTitle())) {
|
throw new AusinessException(E.DATA_PARAM_NULL, "菜单名称不能为空");
|
}
|
if (StringUtils.isBlank(menuAddDto.getName())) {
|
throw new AusinessException(E.DATA_PARAM_NULL, "路由名称不能为空");
|
}
|
if (menuAddDto.getPriority() == null) {
|
throw new AusinessException(E.DATA_PARAM_NULL, "优先级不能为空");
|
}
|
if (StringUtils.isBlank(menuAddDto.getPath())) {
|
throw new AusinessException(E.DATA_PARAM_NULL, "路由不能为空");
|
}
|
if (StringUtils.isBlank(menuAddDto.getComponent())) {
|
throw new AusinessException(E.DATA_PARAM_NULL, "组件不能为空");
|
}
|
MenuInfo menuInfo;
|
// name 路由判断
|
String name = menuAddDto.getName().trim();
|
menuInfo = menuInfoService.getMenuInfoByName(name);
|
if (menuInfo != null) {
|
throw new AusinessException(E.DATA_DATABASE_EXIST, "路由已存在");
|
}
|
// path 路由地址判断
|
String path = menuAddDto.getPath().trim();
|
menuInfo = menuInfoService.getMenuInfoByPath(path);
|
if (menuInfo != null) {
|
throw new AusinessException(E.DATA_DATABASE_EXIST, "路由地址已存在");
|
}
|
// component 组件判断（组件根菜单可以重复，子菜单不可以重复）
|
String component = menuAddDto.getComponent().trim();
|
menuInfo = menuInfoService.getMenuInfoByComponent(component);
|
if (menuInfo != null) {
|
throw new AusinessException(E.DATA_DATABASE_EXIST, "组件已存在");
|
}
|
|
List<String> roleCodes = meta.getRoles();
|
// 校验角色是否存在
|
List<RoleInfo> roleInfos = roleInfoService.getRoleInfoByCodes(roleCodes);
|
|
Long projectId = menuAddDto.getProjectId();
|
ProjectInfo project = projectInfoService.getProjectById(projectId);
|
if (project == null) {
|
throw new BusinessException(ResultCodes.CLIENT_PROJECT_NOT_EXIST);
|
}
|
|
// 角色和项目是否绑定
|
for (RoleInfo role : roleInfos) {
|
RoleProjectInfo roleProjectInfo = roleProjectInfoService.getRoleProjectInfo(role.getId(), project.getId());
|
if (roleProjectInfo == null) {
|
throw new BusinessException(ResultCodes.CLIENT_ROLE_HAS_NO_PROJECT);
|
}
|
}
|
// 1.菜单插入
|
MenuInfo menuEntity = new MenuInfo();
|
menuEntity.setName(name);
|
menuEntity.setRedirect(menuAddDto.getRedirect());
|
menuEntity.setComponent(component);
|
menuEntity.setPath(path);
|
menuEntity.setPriority(menuAddDto.getPriority());
|
menuEntity.setDescription(menuAddDto.getDescription());
|
menuEntity.setTitle(meta.getTitle());
|
menuEntity.setLinkEnable(meta.getIsLink());
|
menuEntity.setHideEnable(meta.getIsHide());
|
menuEntity.setKeepAliveEnable(meta.getIsKeepAlive());
|
menuEntity.setAffixEnable(meta.getIsAffix());
|
menuEntity.setIframeEnable(meta.getIsIframe());
|
menuEntity.setIcon(meta.getIcon());
|
menuEntity.setStatus(MenuStatusEnum.ENABLED.getCode());
|
|
// 判断是否有父菜单
|
if (menuAddDto.getParentId() != null) {
|
MenuInfo parentMenu = menuInfoService.getMenuById(menuAddDto.getParentId());
|
if (parentMenu == null) {
|
throw new BusinessException(ResultCodes.CLIENT_MENU_PARENT_NOT_EXIST);
|
}
|
}
|
menuEntity.setParentId(menuAddDto.getParentId());
|
menuEntity.setPublicEnable(Boolean.FALSE);
|
menuEntity.setGmtCreate(new Date());
|
menuInfoService.saveMenu(menuEntity);
|
|
// 2.项目和菜单
|
ProjectMenuInfo projectMenuInfo = new ProjectMenuInfo();
|
projectMenuInfo.setProjectId(projectId);
|
projectMenuInfo.setMenuId(menuEntity.getId());
|
projectMenuInfoService.save(projectMenuInfo);
|
|
// 3.角色和菜单
|
List<RoleMenuInfo> roleMenus = new ArrayList<>();
|
RoleMenuInfo roleMenuInfo;
|
for (RoleInfo roleInfo : roleInfos) {
|
roleMenuInfo = new RoleMenuInfo();
|
roleMenuInfo.setRoleId(roleInfo.getId());
|
roleMenuInfo.setMenuId(menuEntity.getId());
|
roleMenus.add(roleMenuInfo);
|
}
|
roleMenuInfoService.saveRoleMenus(roleMenus);
|
|
// 4.redis更新所有菜单
|
List<MenuInfoDO> allMenu = menuInfoService.getAllMenu();
|
redisService.setCacheMenuAndExpireTime(RedisKeyEnum.AUTH_MENU.getKey(), allMenu, tokenConfig.getExpiration());
|
}
|
|
|
|
@Override
|
@Transactional
|
public void modMenu(MenuModReqDTO menuModDto, ContextCacheUser currentUser) {
|
|
if (menuModDto.getId() == null) {
|
throw new BusinessException(ResultCodes.CLIENT_PARAM_NULL);
|
}
|
MenuInfo menu = menuInfoService.getMenuById(menuModDto.getId());
|
if (menu == null) {
|
throw new BusinessException(ResultCodes.CLIENT_MENU_NOT_EXIST);
|
}
|
|
MenuMetaReqDTO meta = menuModDto.getMeta();
|
if (meta == null) {
|
throw new AusinessException(E.DATA_PARAM_NULL,"属性为空");
|
}
|
if (StringUtils.isBlank(meta.getTitle())) {
|
throw new AusinessException(E.DATA_PARAM_NULL, "菜单名称不能为空");
|
}
|
if (menuModDto.getPriority() == null) {
|
throw new AusinessException(E.DATA_PARAM_NULL, "优先级不能为空");
|
}
|
if (StringUtils.isBlank(menuModDto.getName())) {
|
throw new AusinessException(E.DATA_PARAM_NULL, "路由名称不能为空");
|
}
|
if (StringUtils.isBlank(menuModDto.getPath())) {
|
throw new AusinessException(E.DATA_PARAM_NULL, "路由不能为空");
|
}
|
if (StringUtils.isBlank(menuModDto.getComponent())) {
|
throw new AusinessException(E.DATA_PARAM_NULL, "组件不能为空");
|
}
|
|
MenuInfo menuInfo;
|
// name 路由判断
|
String name = menuModDto.getName().trim();
|
menuInfo = menuInfoService.getMenuInfoByName(name);
|
if (menuInfo != null && !menuInfo.getId().equals(menu.getId())) {
|
throw new AusinessException(E.DATA_DATABASE_EXIST, "路由已存在");
|
}
|
// path 路由地址判断
|
String path = menuModDto.getPath().trim();
|
menuInfo = menuInfoService.getMenuInfoByPath(path);
|
if (menuInfo != null && !menuInfo.getId().equals(menu.getId())) {
|
throw new AusinessException(E.DATA_DATABASE_EXIST, "路由地址已存在");
|
}
|
// component 组件判断（组件根菜单可以重复，子菜单不可以重复）
|
String component = menuModDto.getComponent().trim();
|
menuInfo = menuInfoService.getMenuInfoByComponent(component);
|
if (menuInfo != null && !menuInfo.getId().equals(menu.getId())) {
|
throw new AusinessException(E.DATA_DATABASE_EXIST, "组件已存在");
|
}
|
|
|
// 判断是否有父菜单
|
if (menuModDto.getParentId() != null) {
|
MenuInfo parentMenu = menuInfoService.getMenuById(menuModDto.getParentId());
|
if (parentMenu == null) {
|
throw new BusinessException(ResultCodes.CLIENT_MENU_PARENT_NOT_EXIST);
|
}
|
}
|
List<String> roleCodes = meta.getRoles();
|
// 校验角色是否存在
|
List<RoleInfo> roleInfos = roleInfoService.getRoleInfoByCodes(roleCodes);
|
|
Long projectId = menuModDto.getProjectId();
|
ProjectInfo project = projectInfoService.getProjectById(projectId);
|
if (project == null) {
|
throw new BusinessException(ResultCodes.CLIENT_PROJECT_NOT_EXIST);
|
}
|
// 角色和项目是否绑定
|
for (RoleInfo role : roleInfos) {
|
RoleProjectInfo roleProjectInfo = roleProjectInfoService.getRoleProjectInfo(role.getId(), project.getId());
|
if (roleProjectInfo == null) {
|
throw new BusinessException(ResultCodes.CLIENT_ROLE_HAS_NO_PROJECT);
|
}
|
}
|
// 1.更新菜单
|
MenuInfo menuEntity = new MenuInfo();
|
menuEntity.setId(menu.getId());
|
menuEntity.setName(menuModDto.getName());
|
menuEntity.setRedirect(menuModDto.getRedirect());
|
menuEntity.setComponent(menuModDto.getComponent());
|
menuEntity.setPriority(menuModDto.getPriority());
|
menuEntity.setDescription(menuModDto.getDescription());
|
// meta
|
menuEntity.setTitle(meta.getTitle());
|
menuEntity.setLinkEnable(meta.getIsLink());
|
menuEntity.setHideEnable(meta.getIsHide());
|
menuEntity.setKeepAliveEnable(meta.getIsKeepAlive());
|
menuEntity.setAffixEnable(meta.getIsAffix());
|
menuEntity.setIframeEnable(meta.getIsIframe());
|
menuEntity.setIcon(meta.getIcon());
|
menuEntity.setPath(menuModDto.getPath());
|
menuEntity.setParentId(menuModDto.getParentId());
|
menuEntity.setGmtModified(new Date());
|
menuInfoService.updateMenu(menuEntity);
|
|
// 2.菜单项目
|
// 2.1解绑该菜单的所有项目
|
projectMenuInfoService.unbindProjectByMenuId(menu.getId());
|
// 2.2绑定当前菜单的所有角色
|
projectMenuInfoService.bindProjectWithMenu(menu.getId(), projectId);
|
|
// 3.角色菜单
|
// 3.1解绑菜单的角色
|
roleMenuInfoService.unbindRoleByMenuId(menu.getId());
|
// 3.2绑定菜单和角色
|
List<RoleMenuInfo> roleMenus = new ArrayList<>();
|
RoleMenuInfo roleMenuInfo;
|
for (RoleInfo roleInfo : roleInfos) {
|
roleMenuInfo = new RoleMenuInfo();
|
roleMenuInfo.setRoleId(roleInfo.getId());
|
roleMenuInfo.setMenuId(menu.getId());
|
roleMenus.add(roleMenuInfo);
|
}
|
roleMenuInfoService.saveRoleMenus(roleMenus);
|
|
// 4.redis更新所有菜单
|
List<MenuInfoDO> allMenu = menuInfoService.getAllMenu();
|
redisService.setCacheMenuAndExpireTime(RedisKeyEnum.AUTH_MENU.getKey(), allMenu, tokenConfig.getExpiration());
|
}
|
|
@Override
|
@Transactional
|
public void delMenu(Long id,ContextCacheUser currentUser) {
|
;
|
if (id == null) {
|
throw new BusinessException(ResultCodes.CLIENT_PARAM_NULL);
|
}
|
MenuInfo menu = menuInfoService.getMenuById(id);
|
if (menu == null) {
|
throw new BusinessException(ResultCodes.CLIENT_MENU_NOT_EXIST);
|
}
|
assert menu.getStatus() != null;
|
MenuStatusEnum status = MenuStatusEnum.parse(menu.getStatus());
|
if (status != MenuStatusEnum.ENABLED) {
|
throw new BusinessException(ResultCodes.CLIENT_MENU_NOT_EXIST);
|
}
|
// 判断是否有子菜单，有不能删除，否则删除
|
Long subMenuCount = menuInfoService.countByParentId(menu.getId());
|
if (subMenuCount > 0) {
|
throw new AusinessException(E.NOT_DELETE, "当前菜单存在子菜单，请先删除子菜单");
|
}
|
// 1.删除菜单
|
MenuInfo menuEntity = new MenuInfo();
|
menuEntity.setId(id);
|
menuEntity.setStatus(MenuStatusEnum.ABANDONED.getCode());
|
menuEntity.setGmtModified(new Date());
|
menuInfoService.updateById(menuEntity);
|
// 2.项目和菜单解绑
|
projectMenuInfoService.unbindProjectByMenuId(id);
|
// 3.角色和菜单解绑
|
roleMenuInfoService.unbindRoleByMenuId(id);
|
// 4.redis更新所有菜单
|
List<MenuInfoDO> allMenu = menuInfoService.getAllMenu();
|
redisService.setCacheMenuAndExpireTime(RedisKeyEnum.AUTH_MENU.getKey(), allMenu, tokenConfig.getExpiration());
|
}
|
|
|
}
|
