mirror of https://gitee.com/stylefeng/roses
【8.0】【sys】【role】权限限制绑定：绑定菜单
fengshuonan
parent
b4a69e9b49
commit
6af917f7b4
|
|
@ -6,10 +6,14 @@ import cn.stylefeng.roses.kernel.sys.modular.menu.entity.SysMenuOptions;
|
|||
import cn.stylefeng.roses.kernel.sys.modular.menu.service.SysMenuOptionsService;
|
||||
import cn.stylefeng.roses.kernel.sys.modular.menu.service.SysMenuService;
|
||||
import cn.stylefeng.roses.kernel.sys.modular.role.action.RoleAssignOperateAction;
|
||||
import cn.stylefeng.roses.kernel.sys.modular.role.action.RoleBindLimitAction;
|
||||
import cn.stylefeng.roses.kernel.sys.modular.role.entity.SysRoleLimit;
|
||||
import cn.stylefeng.roses.kernel.sys.modular.role.entity.SysRoleMenu;
|
||||
import cn.stylefeng.roses.kernel.sys.modular.role.entity.SysRoleMenuOptions;
|
||||
import cn.stylefeng.roses.kernel.sys.modular.role.enums.PermissionNodeTypeEnum;
|
||||
import cn.stylefeng.roses.kernel.sys.modular.role.enums.RoleLimitTypeEnum;
|
||||
import cn.stylefeng.roses.kernel.sys.modular.role.pojo.request.RoleBindPermissionRequest;
|
||||
import cn.stylefeng.roses.kernel.sys.modular.role.service.SysRoleLimitService;
|
||||
import cn.stylefeng.roses.kernel.sys.modular.role.service.SysRoleMenuOptionsService;
|
||||
import cn.stylefeng.roses.kernel.sys.modular.role.service.SysRoleMenuService;
|
||||
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
|
||||
|
|
@ -28,7 +32,7 @@ import java.util.stream.Collectors;
|
|||
* @since 2023/9/8 15:31
|
||||
*/
|
||||
@Service
|
||||
public class RoleBindMenuImpl implements RoleAssignOperateAction {
|
||||
public class RoleBindMenuImpl implements RoleAssignOperateAction, RoleBindLimitAction {
|
||||
|
||||
@Resource
|
||||
private SysRoleMenuService sysRoleMenuService;
|
||||
|
|
@ -42,6 +46,9 @@ public class RoleBindMenuImpl implements RoleAssignOperateAction {
|
|||
@Resource
|
||||
private SysMenuService sysMenuService;
|
||||
|
||||
@Resource
|
||||
private SysRoleLimitService sysRoleLimitService;
|
||||
|
||||
@Override
|
||||
public PermissionNodeTypeEnum getNodeType() {
|
||||
return PermissionNodeTypeEnum.MENU;
|
||||
|
|
@ -59,7 +66,7 @@ public class RoleBindMenuImpl implements RoleAssignOperateAction {
|
|||
sysRoleMenuLambdaQueryWrapper.eq(SysRoleMenu::getRoleId, roleId);
|
||||
this.sysRoleMenuService.remove(sysRoleMenuLambdaQueryWrapper);
|
||||
|
||||
// 2. 如果是选中，则执行菜单和角色的绑定
|
||||
// 1.2. 如果是选中，则执行菜单和角色的绑定
|
||||
// 查询菜单对应的appId，冗余一下appId
|
||||
Map<Long, Long> menuAppIdMap = sysMenuService.getMenuAppId(ListUtil.list(false, menuId));
|
||||
Long appId = menuAppIdMap.get(menuId);
|
||||
|
|
@ -71,38 +78,108 @@ public class RoleBindMenuImpl implements RoleAssignOperateAction {
|
|||
this.sysRoleMenuService.save(sysRoleMenu);
|
||||
}
|
||||
|
||||
// 3. 查询菜单下是否有菜单功能，如果有菜单功能，则也直接清空掉
|
||||
LambdaQueryWrapper<SysMenuOptions> queryWrapper = new LambdaQueryWrapper<>();
|
||||
queryWrapper.eq(SysMenuOptions::getMenuId, menuId);
|
||||
queryWrapper.select(SysMenuOptions::getMenuOptionId, SysMenuOptions::getOptionName);
|
||||
List<SysMenuOptions> totalMenuOptions = sysMenuOptionsService.list(queryWrapper);
|
||||
// 2.1. 查询菜单下的所有菜单功能
|
||||
List<Long> menuOptions = this.getMenuOptions(menuId);
|
||||
|
||||
// 菜单下没有菜单功能，则直接返回
|
||||
if (ObjectUtil.isEmpty(totalMenuOptions)) {
|
||||
if (ObjectUtil.isEmpty(menuOptions)) {
|
||||
return;
|
||||
}
|
||||
|
||||
// 4. 如果有菜单功能，则执行先删除后添加的逻辑
|
||||
// 2.2. 如果有菜单功能，则执行先删除后添加的逻辑
|
||||
// 先删除角色和菜单功能的绑定
|
||||
LambdaQueryWrapper<SysRoleMenuOptions> roleMenuOptionsLambdaQueryWrapper = new LambdaQueryWrapper<>();
|
||||
roleMenuOptionsLambdaQueryWrapper.eq(SysRoleMenuOptions::getRoleId, roleId);
|
||||
roleMenuOptionsLambdaQueryWrapper.in(SysRoleMenuOptions::getMenuOptionId,
|
||||
totalMenuOptions.stream().map(SysMenuOptions::getMenuOptionId).collect(Collectors.toSet()));
|
||||
roleMenuOptionsLambdaQueryWrapper.in(SysRoleMenuOptions::getMenuOptionId, menuOptions);
|
||||
sysRoleMenuOptionsService.remove(roleMenuOptionsLambdaQueryWrapper);
|
||||
|
||||
// 5. 如果是选中状态，则从新进行这些角色和功能的绑定
|
||||
// 2.3. 如果是选中状态，则从新进行这些角色和功能的绑定
|
||||
if (roleBindPermissionRequest.getChecked()) {
|
||||
ArrayList<SysRoleMenuOptions> sysRoleMenuOptions = new ArrayList<>();
|
||||
for (SysMenuOptions totalMenuOption : totalMenuOptions) {
|
||||
for (Long menuOptionId : menuOptions) {
|
||||
SysRoleMenuOptions roleMenuOptions = new SysRoleMenuOptions();
|
||||
roleMenuOptions.setRoleId(roleId);
|
||||
roleMenuOptions.setAppId(appId);
|
||||
roleMenuOptions.setMenuId(menuId);
|
||||
roleMenuOptions.setMenuOptionId(totalMenuOption.getMenuOptionId());
|
||||
roleMenuOptions.setMenuOptionId(menuOptionId);
|
||||
sysRoleMenuOptions.add(roleMenuOptions);
|
||||
}
|
||||
this.sysRoleMenuOptionsService.saveBatch(sysRoleMenuOptions);
|
||||
}
|
||||
}
|
||||
|
||||
@Override
|
||||
public PermissionNodeTypeEnum getRoleBindLimitNodeType() {
|
||||
return this.getNodeType();
|
||||
}
|
||||
|
||||
@Override
|
||||
public void doRoleBindLimitAction(RoleBindPermissionRequest roleBindPermissionRequest) {
|
||||
|
||||
Long roleId = roleBindPermissionRequest.getRoleId();
|
||||
Long menuId = roleBindPermissionRequest.getNodeId();
|
||||
|
||||
List<SysRoleLimit> sysRoleLimitTotal = new ArrayList<>();
|
||||
|
||||
// 1. 先取消绑定，角色对菜单的限制
|
||||
LambdaQueryWrapper<SysRoleLimit> wrapper = new LambdaQueryWrapper<>();
|
||||
wrapper.eq(SysRoleLimit::getRoleId, roleId);
|
||||
wrapper.eq(SysRoleLimit::getBusinessId, menuId);
|
||||
this.sysRoleLimitService.remove(wrapper);
|
||||
|
||||
// 2. 如果是选中，则执行角色绑定菜单限制
|
||||
if (roleBindPermissionRequest.getChecked()) {
|
||||
SysRoleLimit sysRoleLimit = new SysRoleLimit();
|
||||
sysRoleLimit.setRoleId(roleId);
|
||||
sysRoleLimit.setLimitType(RoleLimitTypeEnum.MENU.getCode());
|
||||
sysRoleLimit.setBusinessId(menuId);
|
||||
sysRoleLimitTotal.add(sysRoleLimit);
|
||||
}
|
||||
|
||||
// 2.1. 查询菜单下的所有菜单功能
|
||||
List<Long> menuOptionsIds = this.getMenuOptions(menuId);
|
||||
|
||||
// 菜单下没有菜单功能，则直接返回
|
||||
if (ObjectUtil.isEmpty(menuOptionsIds)) {
|
||||
this.sysRoleLimitService.saveBatch(sysRoleLimitTotal);
|
||||
return;
|
||||
}
|
||||
|
||||
// 2.2. 如果有菜单功能，则执行先删除后添加的逻辑
|
||||
// 先删除角色和菜单功能的绑定
|
||||
LambdaQueryWrapper<SysRoleLimit> optionWrapper = new LambdaQueryWrapper<>();
|
||||
optionWrapper.eq(SysRoleLimit::getRoleId, roleId);
|
||||
optionWrapper.in(SysRoleLimit::getBusinessId, menuOptionsIds);
|
||||
this.sysRoleLimitService.remove(optionWrapper);
|
||||
|
||||
// 2.3. 如果是选中，则创建角色对菜单功能的绑定限制
|
||||
if (roleBindPermissionRequest.getChecked()) {
|
||||
for (Long menuOptionId : menuOptionsIds) {
|
||||
SysRoleLimit sysRoleLimit = new SysRoleLimit();
|
||||
sysRoleLimit.setRoleId(roleId);
|
||||
sysRoleLimit.setLimitType(RoleLimitTypeEnum.MENU_OPTIONS.getCode());
|
||||
sysRoleLimit.setBusinessId(menuOptionId);
|
||||
sysRoleLimitTotal.add(sysRoleLimit);
|
||||
}
|
||||
}
|
||||
this.sysRoleLimitService.saveBatch(sysRoleLimitTotal);
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取菜单下的所有菜单功能
|
||||
*
|
||||
* @author fengshuonan
|
||||
* @since 2023/9/8 16:02
|
||||
*/
|
||||
private List<Long> getMenuOptions(Long menuId) {
|
||||
LambdaQueryWrapper<SysMenuOptions> queryWrapper = new LambdaQueryWrapper<>();
|
||||
queryWrapper.eq(SysMenuOptions::getMenuId, menuId);
|
||||
queryWrapper.select(SysMenuOptions::getMenuOptionId);
|
||||
List<SysMenuOptions> list = sysMenuOptionsService.list(queryWrapper);
|
||||
if (ObjectUtil.isEmpty(list)) {
|
||||
return new ArrayList<>();
|
||||
}
|
||||
return list.stream().map(SysMenuOptions::getMenuOptionId).collect(Collectors.toList());
|
||||
}
|
||||
|
||||
}
|
||||
|
|
|
|||
Loading…
Reference in New Issue