103 lines
3.7 KiB
Java
103 lines
3.7 KiB
Java
package com.ruoyi.system.service;
|
|
|
|
import cn.dev33.satoken.secure.BCrypt;
|
|
import com.ruoyi.component.core.constant.CacheConstants;
|
|
import com.ruoyi.component.core.constant.Constants;
|
|
import com.ruoyi.component.core.constant.GlobalConstants;
|
|
import com.ruoyi.component.log.event.LogininforEvent;
|
|
import com.ruoyi.component.redis.util.RedisUtils;
|
|
import com.ruoyi.system.domain.SysUser;
|
|
import com.ruoyi.component.core.domain.model.RegisterBody;
|
|
import com.ruoyi.component.core.enums.UserType;
|
|
import com.ruoyi.component.core.exception.user.CaptchaException;
|
|
import com.ruoyi.component.core.exception.user.CaptchaExpireException;
|
|
import com.ruoyi.component.core.exception.user.UserException;
|
|
import com.ruoyi.component.core.util.MessageUtils;
|
|
import com.ruoyi.component.core.util.ServletUtils;
|
|
import com.ruoyi.component.core.util.StringUtils;
|
|
import com.ruoyi.component.core.util.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 = GlobalConstants.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);
|
|
}
|
|
|
|
}
|