mirror of
https://github.com/dromara/RuoYi-Vue-Plus.git
synced 2025-09-24 07:19:46 +08:00
102 lines
3.6 KiB
Java
102 lines
3.6 KiB
Java
package com.ruoyi.system.service;
|
|
|
|
import cn.dev33.satoken.secure.BCrypt;
|
|
import com.ruoyi.common.constant.CacheConstants;
|
|
import com.ruoyi.common.constant.Constants;
|
|
import com.ruoyi.common.core.domain.event.LogininforEvent;
|
|
import com.ruoyi.common.core.domain.entity.SysUser;
|
|
import com.ruoyi.common.core.domain.model.RegisterBody;
|
|
import com.ruoyi.common.enums.UserType;
|
|
import com.ruoyi.common.exception.user.CaptchaException;
|
|
import com.ruoyi.common.exception.user.CaptchaExpireException;
|
|
import com.ruoyi.common.exception.user.UserException;
|
|
import com.ruoyi.common.utils.MessageUtils;
|
|
import com.ruoyi.common.utils.ServletUtils;
|
|
import com.ruoyi.common.utils.StringUtils;
|
|
import com.ruoyi.common.utils.redis.RedisUtils;
|
|
import com.ruoyi.common.utils.spring.SpringUtils;
|
|
import lombok.RequiredArgsConstructor;
|
|
import org.springframework.stereotype.Service;
|
|
|
|
/**
|
|
* 注册校验方法
|
|
*
|
|
* @author Lion Li
|
|
*/
|
|
@RequiredArgsConstructor
|
|
@Service
|
|
public class SysRegisterService {
|
|
|
|
private final ISysUserService userService;
|
|
private final ISysConfigService configService;
|
|
|
|
/**
|
|
* 注册
|
|
*/
|
|
public void register(RegisterBody registerBody) {
|
|
String username = registerBody.getUsername();
|
|
String password = registerBody.getPassword();
|
|
// 校验用户类型是否存在
|
|
String userType = UserType.getUserType(registerBody.getUserType()).getUserType();
|
|
|
|
boolean captchaEnabled = configService.selectCaptchaEnabled();
|
|
// 验证码开关
|
|
if (captchaEnabled) {
|
|
validateCaptcha(username, registerBody.getCode(), registerBody.getUuid());
|
|
}
|
|
SysUser sysUser = new SysUser();
|
|
sysUser.setUserName(username);
|
|
sysUser.setNickName(username);
|
|
sysUser.setPassword(BCrypt.hashpw(password));
|
|
sysUser.setUserType(userType);
|
|
|
|
if (!userService.checkUserNameUnique(sysUser)) {
|
|
throw new UserException("user.register.save.error", username);
|
|
}
|
|
boolean regFlag = userService.registerUser(sysUser);
|
|
if (!regFlag) {
|
|
throw new UserException("user.register.error");
|
|
}
|
|
recordLogininfor(username, Constants.REGISTER, MessageUtils.message("user.register.success"));
|
|
}
|
|
|
|
/**
|
|
* 校验验证码
|
|
*
|
|
* @param username 用户名
|
|
* @param code 验证码
|
|
* @param uuid 唯一标识
|
|
*/
|
|
public void validateCaptcha(String username, String code, String uuid) {
|
|
String verifyKey = CacheConstants.CAPTCHA_CODE_KEY + StringUtils.defaultString(uuid, "");
|
|
String captcha = RedisUtils.getCacheObject(verifyKey);
|
|
RedisUtils.deleteObject(verifyKey);
|
|
if (captcha == null) {
|
|
recordLogininfor(username, Constants.REGISTER, MessageUtils.message("user.jcaptcha.expire"));
|
|
throw new CaptchaExpireException();
|
|
}
|
|
if (!code.equalsIgnoreCase(captcha)) {
|
|
recordLogininfor(username, Constants.REGISTER, MessageUtils.message("user.jcaptcha.error"));
|
|
throw new CaptchaException();
|
|
}
|
|
}
|
|
|
|
/**
|
|
* 记录登录信息
|
|
*
|
|
* @param username 用户名
|
|
* @param status 状态
|
|
* @param message 消息内容
|
|
* @return
|
|
*/
|
|
private void recordLogininfor(String username, String status, String message) {
|
|
LogininforEvent logininforEvent = new LogininforEvent();
|
|
logininforEvent.setUsername(username);
|
|
logininforEvent.setStatus(status);
|
|
logininforEvent.setMessage(message);
|
|
logininforEvent.setRequest(ServletUtils.getRequest());
|
|
SpringUtils.context().publishEvent(logininforEvent);
|
|
}
|
|
|
|
}
|