work/book-api
4
1
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

优化代码

This commit is contained in:
阿沐 2022-08-30 22:58:57 +08:00
parent 0f09fce119
commit 550c1b4861
1 changed files with 67 additions and 67 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

134
fast-boot-system/src/main/java/net/maku/system/service/impl/SysMenuServiceImpl.java
View File

@ -1,7 +1,7 @@
package net.maku.system.service.impl; package net.maku.system.service.impl;
import cn.hutool.core.util.StrUtil; import cn.hutool.core.util.StrUtil;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import lombok.AllArgsConstructor; import lombok.AllArgsConstructor;
import net.maku.framework.common.constant.Constant; import net.maku.framework.common.constant.Constant;
import net.maku.framework.common.exception.FastException; import net.maku.framework.common.exception.FastException;
@ -31,87 +31,87 @@ import java.util.Set;
@Service @Service
@AllArgsConstructor @AllArgsConstructor
public class SysMenuServiceImpl extends BaseServiceImpl<SysMenuDao, SysMenuEntity> implements SysMenuService { public class SysMenuServiceImpl extends BaseServiceImpl<SysMenuDao, SysMenuEntity> implements SysMenuService {
private final SysRoleMenuService sysRoleMenuService; private final SysRoleMenuService sysRoleMenuService;
@Override @Override
@Transactional(rollbackFor = Exception.class) @Transactional(rollbackFor = Exception.class)
public void save(SysMenuVO vo) { public void save(SysMenuVO vo) {
SysMenuEntity entity = SysMenuConvert.INSTANCE.convert(vo); SysMenuEntity entity = SysMenuConvert.INSTANCE.convert(vo);
// 保存菜单 // 保存菜单
baseMapper.insert(entity); baseMapper.insert(entity);
} }
@Override @Override
@Transactional(rollbackFor = Exception.class) @Transactional(rollbackFor = Exception.class)
public void update(SysMenuVO vo) { public void update(SysMenuVO vo) {
SysMenuEntity entity = SysMenuConvert.INSTANCE.convert(vo); SysMenuEntity entity = SysMenuConvert.INSTANCE.convert(vo);
// 上级菜单不能为自己 // 上级菜单不能为自己
if(entity.getId().equals(entity.getPid())){ if (entity.getId().equals(entity.getPid())) {
throw new FastException("上级菜单不能为自己"); throw new FastException("上级菜单不能为自己");
} }
// 更新菜单 // 更新菜单
updateById(entity); updateById(entity);
} }
@Override @Override
@Transactional(rollbackFor = Exception.class) @Transactional(rollbackFor = Exception.class)
public void delete(Long id) { public void delete(Long id) {
// 删除菜单 // 删除菜单
removeById(id); removeById(id);
// 删除角色菜单关系 // 删除角色菜单关系
sysRoleMenuService.deleteByMenuId(id); sysRoleMenuService.deleteByMenuId(id);
} }
@Override @Override
public List<SysMenuVO> getMenuList(Integer type) { public List<SysMenuVO> getMenuList(Integer type) {
List<SysMenuEntity> menuList = baseMapper.getMenuList(type); List<SysMenuEntity> menuList = baseMapper.getMenuList(type);
return TreeUtils.build(SysMenuConvert.INSTANCE.convertList(menuList), Constant.ROOT); return TreeUtils.build(SysMenuConvert.INSTANCE.convertList(menuList), Constant.ROOT);
} }
@Override @Override
public List<SysMenuVO> getUserMenuList(UserDetail user, Integer type) { public List<SysMenuVO> getUserMenuList(UserDetail user, Integer type) {
List<SysMenuEntity> menuList; List<SysMenuEntity> menuList;
// 系统管理员，拥有最高权限 // 系统管理员，拥有最高权限
if(user.getSuperAdmin().equals(SuperAdminEnum.YES.getValue())){ if (user.getSuperAdmin().equals(SuperAdminEnum.YES.getValue())) {
menuList = baseMapper.getMenuList(type); menuList = baseMapper.getMenuList(type);
}else { } else {
menuList = baseMapper.getUserMenuList(user.getId(), type); menuList = baseMapper.getUserMenuList(user.getId(), type);
} }
return TreeUtils.build(SysMenuConvert.INSTANCE.convertList(menuList)); return TreeUtils.build(SysMenuConvert.INSTANCE.convertList(menuList));
} }
@Override @Override
public Long getSubMenuCount(Long pid) { public Long getSubMenuCount(Long pid) {
return count(new QueryWrapper<SysMenuEntity>().eq("pid", pid)); return count(new LambdaQueryWrapper<SysMenuEntity>().eq(SysMenuEntity::getPid, pid));
} }
@Override @Override
public Set<String> getUserAuthority(UserDetail user) { public Set<String> getUserAuthority(UserDetail user) {
// 系统管理员，拥有最高权限 // 系统管理员，拥有最高权限
List<String> authorityList; List<String> authorityList;
if(user.getSuperAdmin().equals(SuperAdminEnum.YES.getValue())) { if (user.getSuperAdmin().equals(SuperAdminEnum.YES.getValue())) {
authorityList = baseMapper.getAuthorityList(); authorityList = baseMapper.getAuthorityList();
}else{ } else {
authorityList = baseMapper.getUserAuthorityList(user.getId()); authorityList = baseMapper.getUserAuthorityList(user.getId());
} }
// 用户权限列表 // 用户权限列表
Set<String> permsSet = new HashSet<>(); Set<String> permsSet = new HashSet<>();
for(String authority : authorityList){ for (String authority : authorityList) {
if(StrUtil.isBlank(authority)){ if (StrUtil.isBlank(authority)) {
continue; continue;
} }
permsSet.addAll(Arrays.asList(authority.trim().split(","))); permsSet.addAll(Arrays.asList(authority.trim().split(",")));
} }
return permsSet; return permsSet;
} }
} }