【8.0】【sys】【role】权限限制绑定：绑定菜单

kernel-s-system/system-business-permission/src/main/java/cn/stylefeng/roses/kernel/sys/modular/role/service/impl/RoleBindMenuImpl.java

@@ -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. 查询菜单下是否有菜单功能，如果有菜单功能，则也直接清空掉
+        // 2.1. 查询菜单下的所有菜单功能
-        LambdaQueryWrapper<SysMenuOptions> queryWrapper = new LambdaQueryWrapper<>();
+        List<Long> menuOptions = this.getMenuOptions(menuId);
-        queryWrapper.eq(SysMenuOptions::getMenuId, menuId);
-        queryWrapper.select(SysMenuOptions::getMenuOptionId, SysMenuOptions::getOptionName);
-        List<SysMenuOptions> totalMenuOptions = sysMenuOptionsService.list(queryWrapper);
 
         // 菜单下没有菜单功能，则直接返回
-        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,
+        roleMenuOptionsLambdaQueryWrapper.in(SysRoleMenuOptions::getMenuOptionId, menuOptions);
-                totalMenuOptions.stream().map(SysMenuOptions::getMenuOptionId).collect(Collectors.toSet()));
         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());
+    }
+
 }