package com.dsh.upms.controller;
|
|
|
import cn.mb.cloud.common.api.dto.MenuRouterTree;
|
import cn.mb.cloud.common.api.vo.MenuVO;
|
import cn.mb.cloud.common.api.vo.TreeUtil;
|
import cn.mb.cloud.common.core.constant.CommonConstants;
|
import cn.mb.cloud.common.core.constant.enums.ErrorCodeConstants;
|
import cn.mb.cloud.common.core.exception.BusinessException;
|
import cn.mb.cloud.common.core.util.ResponseData;
|
import cn.mb.cloud.common.log.annotation.SysLog;
|
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
|
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
|
import com.dsh.upms.entity.SysMenu;
|
import com.dsh.upms.entity.SysRoleMenu;
|
import com.dsh.upms.model.vo.LoginUserVo;
|
import com.dsh.upms.service.ISysMenuService;
|
import com.dsh.utils.login.LoginHelper;
|
import com.dsh.utils.login.Tree;
|
import io.swagger.annotations.Api;
|
import io.swagger.annotations.ApiImplicitParam;
|
import io.swagger.annotations.ApiImplicitParams;
|
import io.swagger.annotations.ApiOperation;
|
import org.apache.commons.lang3.StringUtils;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.web.bind.annotation.DeleteMapping;
|
import org.springframework.web.bind.annotation.GetMapping;
|
import org.springframework.web.bind.annotation.PathVariable;
|
import org.springframework.web.bind.annotation.PostMapping;
|
import org.springframework.web.bind.annotation.RequestBody;
|
import org.springframework.web.bind.annotation.RequestMapping;
|
import org.springframework.web.bind.annotation.RequestParam;
|
import org.springframework.web.bind.annotation.RestController;
|
|
import java.util.ArrayList;
|
import java.util.Comparator;
|
import java.util.List;
|
import java.util.Objects;
|
import java.util.Set;
|
import java.util.stream.Collectors;
|
|
/**
|
* <p>
|
* 权限菜单 前端控制器
|
* </p>
|
*
|
* @author null123
|
* @since 2021-04-29
|
*/
|
@Api("权限菜单")
|
@RestController
|
@RequestMapping("/sysMenu")
|
public class SysMenuController {
|
|
|
@Autowired
|
private ISysMenuService sysMenuService;
|
|
/**
|
* 分页查询
|
*
|
* @param page 分页对象
|
* @param sysMenu 菜单权限表
|
* @return
|
*/
|
@GetMapping("/page")
|
@ApiOperation(value = "分页查询", tags = {"权限菜单"}, notes = "", response = SysMenu.class)
|
@ApiImplicitParams({
|
@ApiImplicitParam(value = "每页条数", name = "size", required = true, dataType = "Integer"),
|
@ApiImplicitParam(value = "当前页数", name = "current", required = true, dataType = "Integer"),
|
@ApiImplicitParam(name = "Authorization", value = "Bearer +token", required = true, dataType = "String", paramType = "header", defaultValue = "Bearer eyJhbGciOiJIUzUxMiJ9.....")
|
})
|
public ResponseData getSysMenuPage(Page page, SysMenu sysMenu) {
|
Page pages = sysMenuService.page(page, Wrappers.query(sysMenu));
|
return ResponseData.success(pages);
|
}
|
|
|
/**
|
* 通过id查询菜单权限
|
*
|
* @param id id
|
* @return SysMenu
|
*/
|
@GetMapping("/getById")
|
@ApiOperation(value = "通过id查询菜单权限", tags = {"权限菜单"}, notes = "", response = SysMenu.class)
|
@ApiImplicitParams({
|
@ApiImplicitParam(value = "菜单ID", name = "id", required = true, dataType = "Long"),
|
@ApiImplicitParam(name = "Authorization", value = "Bearer +token", required = true, dataType = "String", paramType = "header", defaultValue = "Bearer eyJhbGciOiJIUzUxMiJ9.....")
|
})
|
public ResponseData getById(Long id) {
|
SysMenu sysMenu = sysMenuService.getById(id);
|
return ResponseData.success(sysMenu);
|
}
|
|
/**
|
* 新增菜单权限表
|
*
|
* @param sysMenu 菜单权限表
|
* @return SysMenu
|
*/
|
@SysLog("新增菜单权限表")
|
@PostMapping("/create")
|
public ResponseData save(@RequestBody SysMenu sysMenu) {
|
boolean result = sysMenuService.save(sysMenu);
|
return ResponseData.success(result);
|
}
|
|
/**
|
* 修改菜单权限表
|
*
|
* @param sysMenu 菜单权限表
|
* @return R
|
*/
|
@SysLog("修改菜单权限表")
|
@PostMapping("updateById")
|
@ApiOperation(value = "修改菜单权限表", tags = {"权限菜单"}, notes = "")
|
@ApiImplicitParams({
|
@ApiImplicitParam(name = "Authorization", value = "Bearer +token", required = true, dataType = "String", paramType = "header", defaultValue = "Bearer eyJhbGciOiJIUzUxMiJ9.....")
|
})
|
public ResponseData updateById(@RequestBody SysMenu sysMenu) {
|
boolean result = sysMenuService.updateById(sysMenu);
|
return ResponseData.success(result);
|
}
|
|
/**
|
* 通过id删除菜单权限表
|
*
|
* @param id id
|
* @return R
|
*/
|
@SysLog("删除菜单权限表")
|
@DeleteMapping("/removeById")
|
public ResponseData removeById(@PathVariable Long id) {
|
boolean result = sysMenuService.removeById(id);
|
return ResponseData.success(result);
|
}
|
|
/**
|
* 启用禁用
|
*
|
* @param id
|
* @return
|
*/
|
@GetMapping("enable")
|
@ApiOperation(value = "启用禁用", tags = {"权限菜单"}, notes = "")
|
@ApiImplicitParams({
|
@ApiImplicitParam(value = "菜单ID", name = "id", required = true, dataType = "Long"),
|
@ApiImplicitParam(name = "Authorization", value = "Bearer +token", required = true, dataType = "String", paramType = "header", defaultValue = "Bearer eyJhbGciOiJIUzUxMiJ9.....")
|
})
|
public ResponseData<Boolean> enable(@RequestParam("id") Long id) {
|
SysMenu menu = sysMenuService.getById(id);
|
if (null != menu) {
|
menu.setEnable(!menu.getEnable());
|
} else {
|
return ResponseData.fail("权限不存在");
|
}
|
return ResponseData.success(sysMenuService.updateById(menu));
|
}
|
|
/**
|
* 返回所有树形菜单集合
|
*
|
* @return 树形菜单
|
*/
|
@GetMapping("/tree")
|
@ApiOperation(value = "返回所有树形菜单集合", tags = {"权限菜单"}, notes = "")
|
@ApiImplicitParams({
|
@ApiImplicitParam(value = "菜单名称", name = "title", required = false, dataType = "String"),
|
@ApiImplicitParam(name = "Authorization", value = "Bearer +token", required = true, dataType = "String", paramType = "header", defaultValue = "Bearer eyJhbGciOiJIUzUxMiJ9.....")
|
})
|
public ResponseData getTree(@RequestParam(value = "name", required = false) String title) {
|
try {
|
List<SysMenu> menus = sysMenuService.getBaseMapper()
|
.selectList(Wrappers.<SysMenu>query().lambda()
|
.eq(SysMenu::getDelFlag, 0).like(StringUtils.isNotBlank(title), SysMenu::getName, title)
|
.orderByAsc(SysMenu::getSort));
|
List<Tree> treeList = new ArrayList<Tree>();
|
Tree rootTree = new Tree("系统菜单");
|
for (SysMenu menu : menus) {
|
Long parent = menu.getParentId();
|
if (null == parent) {
|
Tree tree = new Tree(menu.getId().toString(), menu.getName(), Tree.TREE_NODE_STATE_DEFAULT);
|
tree.setPermission(menu.getPermission());
|
tree.setType(menu.getType());
|
tree.setName(menu.getName());
|
tree.setLable(menu.getName());
|
tree.setPath(menu.getPath());
|
tree.setIcon(menu.getIcon());
|
tree.setEnable(menu.getEnable());
|
tree.setSort(menu.getSort());
|
SysMenu.initTree(menus, tree);
|
treeList.add(tree);
|
}
|
}
|
rootTree.setChildren(treeList);
|
return new ResponseData<>(treeList);
|
} catch (Exception e) {
|
e.printStackTrace();
|
return ResponseData.fail("操作失败");
|
}
|
}
|
|
/**
|
* 返回当前登录用户的树形菜单集合
|
*
|
* @return 当前用户的树形菜单
|
*/
|
@GetMapping("/user/tree")
|
@ApiOperation(value = "返回当前登录用户的树形菜单集合", tags = {"权限菜单"}, notes = "", response = MenuRouterTree.class)
|
@ApiImplicitParams({
|
@ApiImplicitParam(name = "Authorization", value = "Bearer +token", required = true, dataType = "String", paramType = "header", defaultValue = "Bearer eyJhbGciOiJIUzUxMiJ9.....")
|
})
|
public ResponseData getUserMenu() throws BusinessException {
|
// 获取当前登录用户信息
|
LoginUserVo loginUserVo = LoginHelper.getUser();
|
if (Objects.isNull(loginUserVo)) {
|
throw new BusinessException("请登录", ErrorCodeConstants.FAIL.getValue());
|
}
|
// 获取符合条件的菜单
|
Set<MenuVO> all = sysMenuService.findMenuByUserId(loginUserVo.getId());
|
List<MenuRouterTree> menuTreeList = all.stream()
|
.filter(menuVo -> CommonConstants.MENU.equals(menuVo.getType()))
|
.map(MenuRouterTree::new)
|
.sorted(Comparator.comparingInt(MenuRouterTree::getSort))
|
.collect(Collectors.toList());
|
return ResponseData.success(TreeUtil.bulidRouter(menuTreeList));
|
}
|
|
}
|
