【8.1.1】【db】更新获取用户角色的数据范围接口的实现

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

@@ -128,4 +128,12 @@ public interface SysRoleService extends IService<SysRole>, SysRoleServiceApi {
      */
     void refreshRoleAuth(Set<Long> roleIdList, Set<Long> menuIdList, Set<Long> menuOptionIdList);
 
+    /**
+     * 获取角色对应数据范围的类型
+     *
+     * @author fengshuonan
+     * @since 2024-03-01 14:18
+     */
+    List<SysRole> getRoleDataScopeType(List<Long> roleIdList);
+
 }

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

@@ -289,6 +289,18 @@ public class SysRoleServiceImpl extends ServiceImpl<SysRoleMapper, SysRole> impl
         }
     }
 
+    @Override
+    public List<SysRole> getRoleDataScopeType(List<Long> roleIdList) {
+        if (ObjectUtil.isEmpty(roleIdList)) {
+            return new ArrayList<>();
+        }
+
+        LambdaQueryWrapper<SysRole> sysRoleLambdaQueryWrapper = new LambdaQueryWrapper<>();
+        sysRoleLambdaQueryWrapper.in(SysRole::getRoleId, roleIdList);
+        sysRoleLambdaQueryWrapper.select(SysRole::getRoleId, SysRole::getDataScopeType);
+        return this.list(sysRoleLambdaQueryWrapper);
+    }
+
     @Override
     public List<SysRole> permissionGetRoleList(SysRoleRequest sysRoleRequest) {
         LambdaQueryWrapper<SysRole> wrapper = this.createWrapper(sysRoleRequest);

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

@@ -0,0 +1,96 @@
+package cn.stylefeng.roses.kernel.sys.modular.role.service.impl;
+
+import cn.hutool.core.collection.CollectionUtil;
+import cn.hutool.core.collection.ListUtil;
+import cn.stylefeng.roses.kernel.auth.api.context.LoginContext;
+import cn.stylefeng.roses.kernel.auth.api.pojo.login.LoginUser;
+import cn.stylefeng.roses.kernel.db.mp.datascope.UserRoleDataScopeApi;
+import cn.stylefeng.roses.kernel.db.mp.datascope.config.DataScopeConfig;
+import cn.stylefeng.roses.kernel.rule.enums.permission.DataScopeTypeEnum;
+import cn.stylefeng.roses.kernel.sys.api.SysUserRoleServiceApi;
+import cn.stylefeng.roses.kernel.sys.modular.role.entity.SysRole;
+import cn.stylefeng.roses.kernel.sys.modular.role.service.SysRoleDataScopeService;
+import cn.stylefeng.roses.kernel.sys.modular.role.service.SysRoleService;
+import org.springframework.stereotype.Service;
+
+import javax.annotation.Resource;
+import java.util.ArrayList;
+import java.util.List;
+import java.util.Set;
+
+/**
+ * 用户的数据范围的获取实现
+ *
+ * @author fengshuonan
+ * @since 2024-03-01 14:06
+ */
+@Service
+public class UserRoleDataScopeImpl implements UserRoleDataScopeApi {
+
+    @Resource
+    private SysUserRoleServiceApi sysUserRoleServiceApi;
+
+    @Resource
+    private SysRoleService sysRoleService;
+
+    @Resource
+    private SysRoleDataScopeService sysRoleDataScopeService;
+
+    @Override
+    public DataScopeConfig getUserRoleDataScopeConfig() {
+
+        // 获取当前登录用户id
+        LoginUser loginUser = LoginContext.me().getLoginUserNullable();
+        if (loginUser == null) {
+            return null;
+        }
+
+        // 获取用户的角色id集合
+        List<Long> userRoleIdList = sysUserRoleServiceApi.getUserRoleIdList(loginUser.getUserId());
+
+        // 如果这个角色列表大于1，也就是不单纯是普通角色，则将普通角色去掉，以配置的角色为主
+        Long defaultRoleId = sysRoleService.getDefaultRoleId();
+        if (CollectionUtil.contains(userRoleIdList, defaultRoleId)) {
+            userRoleIdList.remove(defaultRoleId);
+        }
+
+        // 获取这些角色对应的【最高】的数据范围，取数据范围10-50最大的数字
+        Integer maxDataScope = 0;
+        Long finalRoleId = null;
+        List<SysRole> roleDataScopeType = sysRoleService.getRoleDataScopeType(userRoleIdList);
+        for (SysRole sysRole : roleDataScopeType) {
+            if (sysRole.getDataScopeType() > maxDataScope) {
+                maxDataScope = sysRole.getDataScopeType();
+                finalRoleId = sysRole.getRoleId();
+            }
+        }
+
+        DataScopeConfig dataScopeConfig = new DataScopeConfig();
+
+        // 设置数据最终的数据范围
+        dataScopeConfig.setDataScopeType(DataScopeTypeEnum.codeToEnum(maxDataScope));
+
+        // 如果数据范围是指定部门，则需要单独查下这个角色对应的部门数据有哪些
+        if (DataScopeTypeEnum.DEFINE.getCode().equals(maxDataScope)) {
+            Set<Long> roleBindOrgIdList = sysRoleDataScopeService.getRoleBindOrgIdList(ListUtil.list(false, finalRoleId));
+            dataScopeConfig.setSpecificOrgIds(new ArrayList<>(roleBindOrgIdList));
+        }
+
+        // 设置用户id
+        dataScopeConfig.setUserId(loginUser.getUserId());
+
+        // 如果数据范围是本公司及以下，则查询当前用户的公司id
+        if (DataScopeTypeEnum.COMPANY_WITH_CHILD.getCode().equals(maxDataScope)) {
+            Long currentUserCompanyId = LoginContext.me().getCurrentUserCompanyId();
+            dataScopeConfig.setUserCompanyId(currentUserCompanyId);
+        }
+
+        // 如果是本部门数据，或者本部门及以下，需要查询当前用户的部门id
+        if (DataScopeTypeEnum.DEPT.getCode().equals(maxDataScope) || DataScopeTypeEnum.DEPT_WITH_CHILD.getCode().equals(maxDataScope)) {
+            dataScopeConfig.setUserDeptId(loginUser.getCurrentOrgId());
+        }
+
+        return dataScopeConfig;
+    }
+
+}