前端邀请码注册功能，绑定角色

byzs-web/src/main/java/cn/iocoder/byzs/module/web/controller/admin/login/WebLoginController.java

@@ -75,8 +75,8 @@ public class WebLoginController {
     @PermitAll
     @Operation(summary = "注册租户")
     public CommonResult<Boolean> register(@RequestBody @Valid WebRegisterVO reqVO) {
-        Long register = webLoginServiceImpl.register(reqVO);
-        return success(register != null);
+        Long userId = webLoginServiceImpl.register(reqVO);
+        return success(userId != null);
     }
 
     @PostMapping("/login")

byzs-web/src/main/java/cn/iocoder/byzs/module/web/service/login/WebLoginServiceImpl.java

@@ -6,19 +6,26 @@ import cn.iocoder.byzs.module.system.controller.admin.invitecode.vo.InviteCodeSa
 import cn.iocoder.byzs.module.system.dal.dataobject.invitecode.InviteCodeDO;
 import cn.iocoder.byzs.module.system.dal.dataobject.user.AdminUserDO;
 import cn.iocoder.byzs.module.system.service.invitecode.InviteCodeService;
+import cn.iocoder.byzs.module.system.service.permission.PermissionService;
 import cn.iocoder.byzs.module.system.service.user.AdminUserServiceImpl;
 import cn.iocoder.byzs.module.web.controller.admin.login.vo.WebRegisterVO;
 import jakarta.annotation.Resource;
+import org.springframework.stereotype.Service;
 
 import java.time.LocalDateTime;
+import java.util.Arrays;
+import java.util.Set;
+import java.util.stream.Collectors;
 
 import static cn.iocoder.byzs.framework.common.exception.util.ServiceExceptionUtil.exception;
 import static cn.iocoder.byzs.module.system.enums.ErrorCodeConstants.USER_CODE_FAILURE;
-
+@Service
 public class WebLoginServiceImpl {
     @Resource
     private AdminUserServiceImpl userService;
     @Resource
+    private PermissionService permissionService;
+    @Resource
     private InviteCodeService inviteCodeService;
 
     public Long register(WebRegisterVO registerVO) {
@@ -34,6 +41,9 @@ public class WebLoginServiceImpl {
         BeanUtils.copyProperties(registerVO, registerReqVo);
         Long userId = userService.registerUser(registerReqVo);
         AdminUserDO user = userService.getUser(userId);
+        Set<Long> roleIds = Arrays.stream(inviteCode.getRoleIds().split(","))
+                .map(s ->Long.parseLong(s.trim())).collect(Collectors.toSet());
+        permissionService.assignUserRole(userId, roleIds);
 
         // 更新邀请码为已使用
         inviteCodeService.updateInviteCode(new InviteCodeSaveReqVO()
@@ -44,7 +54,7 @@ public class WebLoginServiceImpl {
                         .setExpireTime(LocalDateTime.now().plusDays(inviteCode.getValidTime()))
                         .setStatus("2"));
 
-        // 构建返回值
+        // 用户id
         return userId;
     }
 }