银行创建角色时赋值银行类型

delivery-module-system/delivery-module-system-biz/src/main/java/com/unis/module/system/controller/admin/user/UserController.java

@@ -1,17 +1,22 @@
 package com.unis.module.system.controller.admin.user;
 
 import cn.hutool.core.collection.CollUtil;
+import com.baomidou.mybatisplus.core.toolkit.ObjectUtils;
+import com.baomidou.mybatisplus.core.toolkit.StringUtils;
 import com.unis.framework.apilog.core.annotation.ApiAccessLog;
 import com.unis.framework.common.enums.CommonStatusEnum;
 import com.unis.framework.common.pojo.CommonResult;
 import com.unis.framework.common.pojo.PageParam;
 import com.unis.framework.common.pojo.PageResult;
 import com.unis.framework.excel.core.util.ExcelUtils;
+import com.unis.framework.security.core.LoginUser;
+import com.unis.framework.security.core.util.SecurityFrameworkUtils;
 import com.unis.module.system.controller.admin.user.vo.user.*;
 import com.unis.module.system.convert.user.UserConvert;
 import com.unis.module.system.dal.dataobject.dept.DeptDO;
 import com.unis.module.system.dal.dataobject.user.AdminUserDO;
 import com.unis.module.system.enums.common.SexEnum;
+import com.unis.module.system.enums.permission.RoleIdEnum;
 import com.unis.module.system.service.dept.DeptService;
 import com.unis.module.system.service.permission.RoleService;
 import com.unis.module.system.service.user.AdminUserService;
@@ -31,9 +36,11 @@ import java.io.IOException;
 import java.util.*;
 
 import static com.unis.framework.apilog.core.enums.OperateTypeEnum.EXPORT;
+import static com.unis.framework.common.exception.util.ServiceExceptionUtil.exception;
 import static com.unis.framework.common.pojo.CommonResult.error;
 import static com.unis.framework.common.pojo.CommonResult.success;
 import static com.unis.framework.common.util.collection.CollectionUtils.convertList;
+import static com.unis.module.infra.enums.ErrorCodeConstants.ERROR;
 
 @Tag(name = "管理后台 - 用户")
 @RestController
@@ -67,7 +74,26 @@ public class UserController {
                 && (reqVO.getBankType() == null || reqVO.getBankType().isEmpty())) {
             return error(400, "请选择银行类型");
         }
-
+        LoginUser loginUser = SecurityFrameworkUtils.getLoginUser();
+        if (ObjectUtils.isEmpty(loginUser)) {
+            throw exception(ERROR, "不存在用户信息");
+        }
+        String roles = loginUser.getInfo().get("roleIds");
+        if (ObjectUtils.isEmpty(roles)) {
+            throw exception(ERROR, "当前用户不存在角色信息");
+        }
+        if (roles.contains("[") && roles.contains("]")) {
+            String roleIdsReal = roles.substring(1, roles.length() - 1);
+            String[] items = roleIdsReal.split(",");
+            List<String> roleIdList = Arrays.asList(items);
+            if (roleIdList.contains(RoleIdEnum.BANK_ADMIN.getRoleId())) {
+                String bankType = loginUser.getInfo().get("bankType");
+                if (StringUtils.isBlank(bankType)) {
+                    throw exception(ERROR, "当前银行用户不存在银行信息");
+                }
+                reqVO.setBankType(bankType);
+            }
+        }
         Long id = userService.createUser(reqVO);
         return success(id);
     }