package cn.besteam.modules.system.controller;
|
|
import cn.besteam.common.api.vo.Result;
|
import cn.besteam.common.constant.CommonConstant;
|
import cn.besteam.common.constant.ErrorCode;
|
import cn.besteam.common.exception.BusinessException;
|
import cn.besteam.common.util.oConvertUtils;
|
import cn.besteam.modules.system.biz.SysRoleBiz;
|
import cn.besteam.modules.system.dto.SysRoleLogoDTO;
|
import cn.besteam.modules.system.entity.SysPermission;
|
import cn.besteam.modules.system.entity.SysRole;
|
import cn.besteam.modules.system.service.ISysPermissionService;
|
import cn.besteam.modules.system.service.ISysRoleService;
|
import cn.besteam.modules.system.vo.BatchReqVO;
|
import cn.besteam.modules.system.vo.TreeModel;
|
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
|
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
|
import io.swagger.annotations.Api;
|
import io.swagger.annotations.ApiOperation;
|
import lombok.extern.slf4j.Slf4j;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.transaction.annotation.Transactional;
|
import org.springframework.validation.annotation.Validated;
|
import org.springframework.web.bind.annotation.*;
|
|
import javax.servlet.http.HttpServletRequest;
|
import java.util.ArrayList;
|
import java.util.HashMap;
|
import java.util.List;
|
import java.util.Map;
|
|
/**
|
* @Title: Controller
|
* @Description: 角色
|
* @author： besteam
|
* @date： 2020-04-17
|
* @version： V1.0
|
*/
|
@RestController
|
@RequestMapping("/system/sysRole")
|
@Slf4j
|
@Api(description = "角色")
|
public class SysRoleController {
|
|
@Autowired
|
private ISysRoleService sysRoleService;
|
|
@Autowired
|
private ISysPermissionService sysPermissionService;
|
|
@Autowired
|
private SysRoleBiz roleBiz;
|
|
/**
|
* 列表查询
|
*
|
* @param keyword
|
* @param type
|
* @return
|
*/
|
@RequestMapping(value = "/list", method = RequestMethod.GET)
|
@ApiOperation(value = "角色管理-列表模糊查询", notes = "角色管理-列表模糊查询")
|
public Result<?> queryPageList(@RequestParam(name = "keyword", required = false) String keyword, @RequestParam(name = "type",required = false) Integer type) {
|
|
return Result.ok(roleBiz.listRole(keyword, type));
|
}
|
|
/**
|
* 添加
|
*
|
* @param role
|
* @return
|
*/
|
@RequestMapping(value = "/add", method = RequestMethod.POST)
|
@ApiOperation(value = "角色管理-新增", notes = "角色管理-新增")
|
@Transactional(rollbackFor = Exception.class)
|
public Result<?> add(@RequestBody @Validated SysRole role) {
|
|
roleBiz.saveOrUpdate(role);
|
return Result.ok("添加成功！");
|
}
|
|
/**
|
* 编辑
|
*
|
* @param role
|
* @return
|
*/
|
@RequestMapping(value = "/edit", method = RequestMethod.PUT)
|
@ApiOperation(value = "角色管理-编辑", notes = "角色管理-编辑")
|
public Result<?> edit(@RequestBody SysRole role) {
|
|
roleBiz.saveOrUpdate(role);
|
return Result.ok("编辑成功！");
|
}
|
|
/**
|
* 通过id删除
|
*
|
* @param id
|
* @return
|
*/
|
@RequestMapping(value = "/delete", method = RequestMethod.DELETE)
|
@ApiOperation(value = "角色管理-通过id删除", notes = "角色管理-通过id删除")
|
public Result<?> delete(@RequestParam(name = "id") String id) {
|
|
return Result.ok(roleBiz.deleteRole(id));
|
}
|
|
/**
|
* 批量删除
|
*
|
* @param vo
|
* @return
|
*/
|
@RequestMapping(value = "/deleteBatch", method = RequestMethod.POST)
|
@ApiOperation(value = "角色管理-批量删除", notes = "角色管理-批量删除")
|
public Result<?> deleteBatch(@Validated @RequestBody BatchReqVO vo) {
|
|
return Result.ok(roleBiz.batchDelete(vo));
|
}
|
|
/**
|
* 通过id查询
|
*
|
* @param id
|
* @return
|
*/
|
@RequestMapping(value = "/queryById", method = RequestMethod.GET)
|
@ApiOperation(value = "角色管理-通过id查询", notes = "角色管理-通过id查询")
|
public Result<SysRole> queryById(@RequestParam(name = "id") String id) {
|
|
Result<SysRole> result = new Result<>();
|
SysRole sysrole = sysRoleService.getById(id);
|
if (sysrole == null) {
|
result.error500("未找到对应实体");
|
} else {
|
result.setResult(sysrole);
|
result.setSuccess(true);
|
}
|
return result;
|
}
|
|
@RequestMapping(value = "/queryall", method = RequestMethod.GET)
|
public Result<?> queryall(@RequestParam(name = "type") Integer type) {
|
|
List<SysRole> roleList = sysRoleService.list(Wrappers.<SysRole>lambdaQuery().eq(SysRole::getType, type));
|
for (SysRole sysRole : roleList) {
|
int count = roleBiz.countUserByRole(sysRole.getId());
|
sysRole.setRoleName(sysRole.getRoleName() + " (" + count + ")");
|
}
|
return Result.ok(roleList);
|
}
|
|
@ApiOperation(value = "用户角色授权功能，查询菜单权限树")
|
@RequestMapping(value = "/queryTreeList", method = RequestMethod.GET)
|
public Result<Map<String, Object>> queryTreeList(HttpServletRequest request) {
|
|
Result<Map<String, Object>> result = new Result<>();
|
//全部权限ids
|
List<Integer> ids = new ArrayList<>();
|
try {
|
LambdaQueryWrapper<SysPermission> query = new LambdaQueryWrapper<SysPermission>();
|
query.eq(SysPermission::getDelFlag, CommonConstant.DEL_FLAG_0);
|
query.orderByAsc(SysPermission::getSortNo);
|
List<SysPermission> list = sysPermissionService.list(query);
|
for (SysPermission sysPer : list) {
|
ids.add(sysPer.getId());
|
}
|
List<TreeModel> treeList = new ArrayList<>();
|
getTreeModelList(treeList, list, null);
|
Map<String, Object> resMap = new HashMap<String, Object>();
|
resMap.put("treeList", treeList); //全部树节点数据
|
resMap.put("ids", ids);//全部树ids
|
result.setResult(resMap);
|
result.setSuccess(true);
|
} catch (Exception e) {
|
log.error(e.getMessage(), e);
|
}
|
return result;
|
}
|
|
private void getTreeModelList(List<TreeModel> treeList, List<SysPermission> metaList, TreeModel temp) {
|
|
for (SysPermission permission : metaList) {
|
Integer tempPid = permission.getParentId();
|
TreeModel tree = new TreeModel(permission.getId(), tempPid, permission.getName(), permission.getRuleFlag(), permission.getIsLeaf());
|
if (temp == null && oConvertUtils.isEmpty(tempPid)) {
|
treeList.add(tree);
|
if (!tree.getIsLeaf()) {
|
getTreeModelList(treeList, metaList, tree);
|
}
|
} else if (temp != null && tempPid != null && tempPid.equals(temp.getKey())) {
|
temp.getChildren().add(tree);
|
if (!tree.getIsLeaf()) {
|
getTreeModelList(treeList, metaList, tree);
|
}
|
}
|
}
|
}
|
|
@PostMapping("/uploadLogo")
|
@ApiOperation(value = "配置logo")
|
public Result<Boolean> uploadLogo(@RequestBody SysRoleLogoDTO dto) {
|
|
SysRole sysRole = sysRoleService.getById(dto.getId());
|
if (sysRole == null) {
|
throw new BusinessException(ErrorCode.ROLE_NOT_FOUND);
|
}
|
sysRole.setLogo(dto.getLogo());
|
return Result.ok(sysRoleService.updateById(sysRole));
|
}
|
|
}
|
