From c738530fb3890f50b788375ec5a81b8f03242cc6 Mon Sep 17 00:00:00 2001 From: qirong <2032486488@qq.com> Date: 星期三, 22 十一月 2023 16:11:21 +0800 Subject: [PATCH] 修改 --- ruoyi-admin/src/main/java/org/dromara/web/service/SysRegisterService.java | 12 +++++++++++- 1 files changed, 11 insertions(+), 1 deletions(-) diff --git a/ruoyi-admin/src/main/java/org/dromara/web/service/SysRegisterService.java b/ruoyi-admin/src/main/java/org/dromara/web/service/SysRegisterService.java index 4d4b072..c5aaea3 100644 --- a/ruoyi-admin/src/main/java/org/dromara/web/service/SysRegisterService.java +++ b/ruoyi-admin/src/main/java/org/dromara/web/service/SysRegisterService.java @@ -19,8 +19,12 @@ import org.dromara.common.tenant.helper.TenantHelper; import org.dromara.common.web.config.properties.CaptchaProperties; import org.dromara.system.domain.SysUser; +import org.dromara.system.domain.SysUserRole; import org.dromara.system.domain.bo.SysUserBo; +import org.dromara.system.domain.vo.SysUserVo; +import org.dromara.system.enums.RoleType; import org.dromara.system.mapper.SysUserMapper; +import org.dromara.system.mapper.SysUserRoleMapper; import org.dromara.system.service.ISysUserService; import lombok.RequiredArgsConstructor; import org.springframework.stereotype.Service; @@ -37,6 +41,7 @@ private final ISysUserService userService; private final SysUserMapper userMapper; private final CaptchaProperties captchaProperties; + private final SysUserRoleMapper userRoleMapper; /** * 娉ㄥ唽 @@ -50,7 +55,7 @@ boolean captchaEnabled = captchaProperties.getEnable(); // 楠岃瘉鐮佸紑鍏� - if (captchaEnabled) { + if (!captchaEnabled) { validateCaptcha(tenantId, username, registerBody.getCode(), registerBody.getUuid()); } SysUserBo sysUser = new SysUserBo(); @@ -70,6 +75,11 @@ if (!regFlag) { throw new UserException("user.register.error"); } + SysUserVo sysUserVo = userMapper.selectUserByUserName(username); + SysUserRole userRole = new SysUserRole(); + userRole.setUserId(sysUserVo.getUserId()); + userRole.setRoleId(Long.valueOf(RoleType.INTERNET.getType())); + userRoleMapper.insert(userRole); recordLogininfor(tenantId, username, Constants.REGISTER, MessageUtils.message("user.register.success")); } -- Gitblit v1.8.0