package cn.besteam.modules.system.biz;
|
|
import cn.besteam.common.constant.CannotDeleteRoleEnum;
|
import cn.besteam.common.constant.ErrorCode;
|
import cn.besteam.common.exception.BusinessException;
|
import cn.besteam.modules.shiro.authc.util.EnvUtils;
|
import cn.besteam.modules.system.entity.*;
|
import cn.besteam.modules.system.mapper.SysRoleMapper;
|
import cn.besteam.modules.system.service.*;
|
import cn.besteam.modules.system.vo.BatchReqVO;
|
import cn.besteam.modules.system.vo.SysRoleVO;
|
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
|
import com.baomidou.mybatisplus.core.toolkit.CollectionUtils;
|
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
|
import org.apache.commons.lang3.StringUtils;
|
import org.springframework.beans.BeanUtils;
|
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.Arrays;
|
import java.util.List;
|
import java.util.Set;
|
import java.util.stream.Collectors;
|
|
/**
|
* @author ltri
|
* @date 2020/1/7 2:30 PM
|
*/
|
@Service
|
public class SysRoleBiz {
|
|
@Autowired
|
private ISysRoleService roleService;
|
|
@Autowired
|
private ISysUserRoleService userRoleService;
|
|
@Autowired
|
private SysRoleMapper roleMapper;
|
|
@Autowired
|
private ISysRolePermissionService rolePermissionService;
|
|
@Autowired
|
private ISysUserService sysUserService;
|
|
@Autowired
|
private ISysPermissionService sysPermissionService;
|
|
/**
|
* 列表查询
|
*
|
* @param keyword
|
* @param type
|
* @return
|
*/
|
public List<SysRoleVO> listRole(String keyword, Integer type) {
|
|
SysUserRole sysUserRole = userRoleService.getOne(new QueryWrapper<SysUserRole>().lambda().eq(SysUserRole::getUserId, EnvUtils.getUser().getId()));
|
if (sysUserRole == null) {
|
return new ArrayList<>();
|
}
|
SysPermission sysPermission = null;
|
if(type!=null) {
|
if (type.equals(1) || type.equals(-1)) {
|
sysPermission = sysPermissionService.getOne(new QueryWrapper<SysPermission>().lambda().eq(SysPermission::getPerms, "role:operate:list"));
|
} else {
|
sysPermission = sysPermissionService.getOne(new QueryWrapper<SysPermission>().lambda().eq(SysPermission::getPerms, "role:user:list"));
|
}
|
if (sysPermission == null) {
|
return new ArrayList<>();
|
}
|
SysRolePermission rolePermission = rolePermissionService.getOne(new QueryWrapper<SysRolePermission>().lambda().eq(SysRolePermission::getRoleId, sysUserRole.getRoleId()).eq(SysRolePermission::getPermissionId, sysPermission.getId()));
|
if (type.equals(1) || type.equals(2)) {
|
if (rolePermission == null) {
|
return new ArrayList<>();
|
}
|
}
|
}
|
List<SysRoleVO> sysRoleList = roleMapper.listRole(keyword, type);
|
//获取角色id集合
|
List<Integer> roleIds = sysRoleList.stream().map(SysRoleVO::getId).collect(Collectors.toList());
|
if (CollectionUtils.isNotEmpty(roleIds)) {
|
//获取存在角色关联的set
|
Set<Integer> set = userRoleService.list(Wrappers.<SysUserRole>lambdaQuery().in(SysUserRole::getRoleId, roleIds)).stream().map(SysUserRole::getRoleId).collect(Collectors.toSet());
|
//返回删除标识符
|
sysRoleList.forEach(sysRoleVO -> {
|
sysRoleVO.setDelFlag(set.contains(sysRoleVO.getId()) ? 0 : 1);
|
});
|
}
|
return sysRoleList;
|
}
|
|
/**
|
* 角色管理-新增/编辑
|
*
|
* @param dto
|
*/
|
@Transactional(rollbackFor = Exception.class)
|
public void saveOrUpdate(SysRole dto) {
|
|
List<Integer> roleTypeList = userRoleService.getRoleTypeList(EnvUtils.getUser().getId());
|
if (CollectionUtils.isEmpty(roleTypeList)) {
|
throw new BusinessException(ErrorCode.ERR_500, "无操作权限");
|
}
|
if (!roleTypeList.contains(dto.getType())) {
|
throw new BusinessException(ErrorCode.ERR_500, "无操作权限");
|
}
|
//新增
|
if (dto.getId() == null || dto.getId() <= 0) {
|
if (StringUtils.isNotBlank(dto.getRoleName())) {
|
int count = roleService.count(Wrappers.<SysRole>lambdaQuery().eq(SysRole::getRoleName, dto.getRoleName()));
|
if (count > 0) {
|
throw new BusinessException(ErrorCode.ERR_500, "角色名称重复，请重新输入");
|
}
|
}
|
SysRole sysRole = new SysRole();
|
BeanUtils.copyProperties(dto, sysRole);
|
roleService.save(sysRole);
|
//
|
SysRolePermission sysRolePermission = new SysRolePermission(sysRole.getId(), 6);
|
rolePermissionService.save(sysRolePermission);
|
} else {
|
SysRole sysRole = roleService.getById(dto.getId());
|
if (sysRole == null) {
|
throw new BusinessException(ErrorCode.ROLE_NOT_FOUND);
|
}
|
SysRole sysRole1 = roleService.getOne(new QueryWrapper<SysRole>().lambda().eq(SysRole::getRoleName, dto.getRoleName()));
|
if (sysRole1 != null && !sysRole.getId().equals(sysRole1.getId())) {
|
throw new BusinessException(ErrorCode.ERR_500, "角色名称重复，请重新输入");
|
}
|
if (CannotDeleteRoleEnum.SET.contains(dto.getRoleName())) {
|
throw new BusinessException(ErrorCode.ROLE_CAN_NOT_EDIT);
|
}
|
BeanUtils.copyProperties(dto, sysRole);
|
roleService.updateById(sysRole);
|
}
|
}
|
|
/**
|
* 通过id删除
|
*
|
* @param id
|
* @return
|
*/
|
@Transactional(rollbackFor = Exception.class)
|
public boolean deleteRole(String id) {
|
|
SysRole sysRole = roleService.getById(id);
|
if (sysRole == null) {
|
throw new BusinessException(ErrorCode.ROLE_NOT_FOUND);
|
}
|
if (CannotDeleteRoleEnum.SET.contains(sysRole.getRoleName())) {
|
throw new BusinessException(ErrorCode.ROLE_CAN_NOT_DELETE);
|
}
|
int count = userRoleService.count(Wrappers.<SysUserRole>lambdaQuery().eq(SysUserRole::getRoleId, id));
|
if (count > 0) {
|
throw new BusinessException(ErrorCode.ROLE_CAN_NOT_DELETE);
|
}
|
//删除角色和权限关联
|
rolePermissionService.remove(Wrappers.<SysRolePermission>lambdaQuery().eq(SysRolePermission::getRoleId, id));
|
//删除角色
|
return roleService.removeById(id);
|
}
|
|
/**
|
* 批量删除
|
*
|
* @param vo
|
* @return
|
*/
|
@Transactional(rollbackFor = Exception.class)
|
public boolean batchDelete(BatchReqVO vo) {
|
|
if (StringUtils.isEmpty(vo.getIds())) {
|
throw new BusinessException(ErrorCode.SYSTEM_REQUEST_PARAM_ERROR);
|
}
|
List<String> deleteIds = Arrays.asList(vo.getIds().split(","));
|
List<SysRole> list = roleService.list(Wrappers.<SysRole>lambdaQuery().in(SysRole::getId, deleteIds));
|
list.forEach(sysRole -> {
|
if (CannotDeleteRoleEnum.SET.contains(sysRole.getRoleName())) {
|
throw new BusinessException(ErrorCode.ROLE_CAN_NOT_DELETE);
|
}
|
});
|
deleteIds.forEach(i -> {
|
int count = userRoleService.count(Wrappers.<SysUserRole>lambdaQuery().eq(SysUserRole::getRoleId, i));
|
if (count > 0) {
|
throw new BusinessException(ErrorCode.ROLE_CAN_NOT_DELETE);
|
}
|
});
|
//删除角色和权限关联
|
rolePermissionService.remove(Wrappers.<SysRolePermission>lambdaQuery().in(SysRolePermission::getRoleId, deleteIds));
|
//删除角色
|
return roleService.remove(Wrappers.<SysRole>lambdaQuery().in(SysRole::getId, deleteIds));
|
}
|
|
public Integer countUserByRole(Integer roleId) {
|
|
Integer count = 0;
|
List<SysUserRole> list = userRoleService.list(new QueryWrapper<SysUserRole>().lambda().eq(SysUserRole::getRoleId, roleId));
|
for (SysUserRole sysUserRole : list) {
|
SysUser sysUser = sysUserService.getById(sysUserRole.getUserId());
|
if (sysUser != null && sysUser.getDelFlag().equals("0")) {
|
count++;
|
}
|
}
|
return count;
|
}
|
|
}
|
