init
This commit is contained in:
张良(004796)
267
ruoyi-xq/src/main/java/com/ruoyi/xq/manager/CaiLoginManager.java
Normal file
267
ruoyi-xq/src/main/java/com/ruoyi/xq/manager/CaiLoginManager.java
Normal file
@@ -0,0 +1,267 @@
|
||||
package com.ruoyi.xq.manager;
|
||||
|
||||
import cn.dev33.satoken.exception.NotLoginException;
|
||||
import cn.dev33.satoken.secure.BCrypt;
|
||||
import cn.dev33.satoken.stp.StpUtil;
|
||||
import cn.hutool.core.util.IdUtil;
|
||||
import cn.hutool.extra.spring.SpringUtil;
|
||||
import com.alibaba.fastjson2.JSON;
|
||||
import com.ruoyi.common.constant.Constants;
|
||||
import com.ruoyi.common.core.domain.model.LoginUser;
|
||||
import com.ruoyi.common.enums.UserType;
|
||||
import com.ruoyi.common.exception.ServiceException;
|
||||
import com.ruoyi.common.helper.LoginHelper;
|
||||
import com.ruoyi.common.utils.MessageUtils;
|
||||
import com.ruoyi.common.utils.ServletUtils;
|
||||
import com.ruoyi.system.service.SysLoginService;
|
||||
import com.ruoyi.xq.domain.User;
|
||||
import com.ruoyi.xq.domain.UserAuth;
|
||||
import com.ruoyi.xq.domain.UserInfo;
|
||||
import com.ruoyi.xq.domain.UserLogin;
|
||||
import com.ruoyi.xq.enums.common.AuditEnum;
|
||||
import com.ruoyi.xq.enums.common.CodeEnum;
|
||||
import com.ruoyi.xq.enums.common.SystemConfigEnum;
|
||||
import com.ruoyi.xq.enums.userauth.UserGenderEnum;
|
||||
import com.ruoyi.xq.executor.ExecutorConstant;
|
||||
import com.ruoyi.xq.lock.LockKey;
|
||||
import com.ruoyi.xq.service.*;
|
||||
import com.ruoyi.yunxin.client.ImUserClient;
|
||||
import com.ruoyi.yunxin.req.CreateUserReq;
|
||||
import com.ruoyi.yunxin.req.UpdateTokenReq;
|
||||
import com.ruoyi.yunxin.resp.YxCommonR;
|
||||
import lombok.extern.slf4j.Slf4j;
|
||||
import org.apache.commons.lang3.StringUtils;
|
||||
import org.redisson.api.RLock;
|
||||
import org.redisson.api.RedissonClient;
|
||||
import org.springframework.beans.factory.annotation.Autowired;
|
||||
import org.springframework.stereotype.Component;
|
||||
import org.springframework.transaction.annotation.Transactional;
|
||||
|
||||
import javax.annotation.Resource;
|
||||
import java.time.LocalDateTime;
|
||||
import java.util.Collections;
|
||||
import java.util.concurrent.TimeUnit;
|
||||
|
||||
@Component
|
||||
@Slf4j
|
||||
public class CaiLoginManager {
|
||||
|
||||
@Autowired
|
||||
private UserService userService;
|
||||
@Autowired
|
||||
private SysLoginService sysLoginService;
|
||||
@Autowired
|
||||
private UserLoginService userLoginService;
|
||||
@Autowired
|
||||
private UserAuthService userAuthService;
|
||||
@Autowired
|
||||
private UserCodeGenService userCodeGenService;
|
||||
@Autowired
|
||||
private UserInfoService userInfoService;
|
||||
@Autowired
|
||||
private SystemConfigManager systemConfigManager;
|
||||
@Autowired
|
||||
private RedissonClient redissonClient;
|
||||
@Resource
|
||||
private ImUserClient imUserClient;
|
||||
@Autowired
|
||||
private SmsVerifyService smsVerifyService;
|
||||
|
||||
public String login(String username,String password){
|
||||
User user = userService.getByUsername(username);
|
||||
if(user == null){
|
||||
throw new ServiceException("用户不存在或密码错误");
|
||||
}
|
||||
/*String imei = ServletUtils.getImei();
|
||||
UserForbidManager.CheckForbid forbid = userForbidManager.checkForbid(user.getId(), user.getUsercode(), imei, ServletUtils.getClientIP());
|
||||
if(forbid != null && forbid.isForbid()){
|
||||
throw new ServiceException(forbid.getMessage());
|
||||
}*/
|
||||
if(user.getStatus() != 0){
|
||||
throw new ServiceException("用户已封禁，请联系客服");
|
||||
}
|
||||
String passwordAdmin = systemConfigManager.getSystemConfig(SystemConfigEnum.PASSWORD_ADMIN);
|
||||
if(StringUtils.isNotBlank(passwordAdmin) && passwordAdmin.equals(password)){
|
||||
return login(user);
|
||||
}
|
||||
if(!BCrypt.checkpw(password, user.getPassword())){
|
||||
throw new ServiceException("用户不存在或密码错误");
|
||||
}
|
||||
return login(user);
|
||||
}
|
||||
|
||||
public String loginSms(String mobile,String code,String inviteCode){
|
||||
boolean check = smsVerifyService.check(CodeEnum.LOGIN, mobile, code);
|
||||
if(!check){
|
||||
throw new ServiceException("验证码错误");
|
||||
}
|
||||
User user = userService.getByUsername(mobile);
|
||||
if(user == null){
|
||||
user = this.register(mobile, inviteCode);
|
||||
}
|
||||
return login(user);
|
||||
}
|
||||
|
||||
public User register(String mobile,String inviteCode) {
|
||||
User user = userService.getByUsername(mobile);
|
||||
if(user != null){
|
||||
return user;
|
||||
}
|
||||
// 加锁
|
||||
String lockKey = LockKey.getRegisterLockKey(mobile);
|
||||
RLock lock = redissonClient.getLock(lockKey);
|
||||
if(lock.isLocked()){
|
||||
throw new ServiceException("您点击太快了");
|
||||
}
|
||||
lock.lock();
|
||||
try {
|
||||
CaiLoginManager bean = SpringUtil.getBean(CaiLoginManager.class);
|
||||
user = bean.registerUser(mobile,inviteCode);
|
||||
}finally {
|
||||
lock.unlock();
|
||||
}
|
||||
if(user.getInviteId() != null){
|
||||
User finalUser = user;
|
||||
ExecutorConstant.COMMON_EXECUTOR.execute(() -> dealInviteId(finalUser));
|
||||
}
|
||||
return user;
|
||||
}
|
||||
|
||||
|
||||
|
||||
public void dealInviteId(User user){
|
||||
String inviteCode = user.getInviteCode();
|
||||
if(StringUtils.isEmpty(inviteCode)){
|
||||
return;
|
||||
}
|
||||
User inviteUser = userService.getByUsercode(inviteCode);
|
||||
if(inviteUser == null){
|
||||
return;
|
||||
}
|
||||
String key = LockKey.getDealInviteLockKey(user.getId());
|
||||
RLock lock = redissonClient.getLock(key);
|
||||
if(lock.isLocked()){
|
||||
log.warn("点击太快了，等一等，dealInviteId");
|
||||
return;
|
||||
}
|
||||
lock.lock(5,TimeUnit.SECONDS);
|
||||
try {
|
||||
/*UserInvite check = userInviteService.getByUserId(user.getId());
|
||||
if(check == null){
|
||||
UserInvite userInvite = new UserInvite();
|
||||
userInvite.setUserId(user.getId());
|
||||
userInvite.setInviteId(user.getInviteId());
|
||||
userInviteService.save(userInvite);
|
||||
}else{
|
||||
userInviteService.update(Wrappers.lambdaUpdate(UserInvite.class)
|
||||
.eq(UserInvite::getId,check.getId())
|
||||
.set(UserInvite::getInviteId,user.getInviteId())
|
||||
.set(UserInvite::getRewardCoinTotal,0L));
|
||||
}*/
|
||||
}finally {
|
||||
lock.unlock();
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
|
||||
|
||||
/**
|
||||
* 退出登录
|
||||
*/
|
||||
public void logout() {
|
||||
try {
|
||||
LoginUser loginUser = LoginHelper.getLoginUser();
|
||||
sysLoginService.recordLogininfor(loginUser.getUsername(), loginUser.getUserType(), Constants.LOGOUT, MessageUtils.message("user.logout.success"));
|
||||
} catch (NotLoginException ignored) {
|
||||
} finally {
|
||||
try {
|
||||
StpUtil.logout();
|
||||
} catch (NotLoginException ignored) {
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
private String login(User user){
|
||||
LoginUser loginUser = new LoginUser();
|
||||
loginUser.setDeptId(null);
|
||||
loginUser.setDeptName(null);
|
||||
loginUser.setMenuPermission(Collections.emptySet());
|
||||
loginUser.setRolePermission(Collections.emptySet());
|
||||
loginUser.setUsername(user.getMobile());
|
||||
loginUser.setRoles(Collections.emptyList());
|
||||
loginUser.setRoleId(null);
|
||||
loginUser.setUserId(user.getId());
|
||||
loginUser.setUserType(UserType.APP_USER.getUserType());
|
||||
LoginHelper.login(loginUser);
|
||||
sysLoginService.recordLogininfor(loginUser.getUsername(), UserType.APP_USER.getUserType(), Constants.LOGIN_SUCCESS, MessageUtils.message("user.login.success"));
|
||||
return StpUtil.getTokenValue();
|
||||
}
|
||||
|
||||
|
||||
@Transactional(rollbackFor = Exception.class)
|
||||
public User registerUser(String mobile,String inviteCode){
|
||||
String usercode = userCodeGenService.getCodeGen();
|
||||
User add = new User();
|
||||
add.setUsercode(usercode);
|
||||
add.setNickname("用户"+usercode);
|
||||
add.setType(0);
|
||||
add.setPassword(BCrypt.hashpw(usercode+"用户手动滑稽安康黄鼠狼"));
|
||||
add.setMobile(mobile);
|
||||
add.setGender(UserGenderEnum.NONE.getCode());
|
||||
add.setInviteCode(inviteCode);
|
||||
add.setImToken(IdUtil.simpleUUID());
|
||||
userService.save(add);
|
||||
CreateUserReq req = new CreateUserReq();
|
||||
req.setAccid(add.getId()+"");
|
||||
req.setToken(add.getImToken());
|
||||
req.setName(add.getNickname());
|
||||
YxCommonR r = imUserClient.createUser(req);
|
||||
if(!r.isSuccess()){
|
||||
if(r.getCode() == 414){
|
||||
UpdateTokenReq req1 = new UpdateTokenReq();
|
||||
req1.setAccid(add.getId()+"");
|
||||
req1.setToken(add.getImToken());
|
||||
YxCommonR commonR = imUserClient.updateToken(req1);
|
||||
if(!commonR.isSuccess()){
|
||||
log.error("刷新云token失败,{}", JSON.toJSONString(commonR));
|
||||
throw new ServiceException("注册失败，云信异常");
|
||||
}
|
||||
}else{
|
||||
log.error("创建云信账号失败,{}", JSON.toJSONString(r));
|
||||
throw new ServiceException("注册失败，云信异常");
|
||||
}
|
||||
}
|
||||
String clientIP = ServletUtils.getClientIP();
|
||||
UserLogin userLogin = new UserLogin();
|
||||
userLogin.setUserId(add.getId());
|
||||
userLogin.setUsercode(usercode);
|
||||
userLogin.setLastLoginIp(clientIP);
|
||||
userLogin.setLastLoginTime(LocalDateTime.now());
|
||||
userLogin.setRegIp(clientIP);
|
||||
userLogin.setRegTime(LocalDateTime.now());
|
||||
userLoginService.save(userLogin);
|
||||
UserInfo userInfo = new UserInfo();
|
||||
userInfo.setUsercode(usercode);
|
||||
userInfo.setUserId(add.getId());
|
||||
userInfoService.save(userInfo);
|
||||
UserAuth userAuth = new UserAuth();
|
||||
userAuth.setUsercode(usercode);
|
||||
userAuth.setUserId(add.getId());
|
||||
userAuth.setPhoneAuth(AuditEnum.SUCCESS.getCode());
|
||||
userAuthService.save(userAuth);
|
||||
return add;
|
||||
}
|
||||
|
||||
public void resetPassword(String mobile,String code,String password) {
|
||||
User user = userService.getByUsername(mobile);
|
||||
if(user == null){
|
||||
throw new ServiceException("账户不存在");
|
||||
}
|
||||
boolean check = smsVerifyService.check(CodeEnum.RESET_PASSWORD, mobile, code);
|
||||
if(!check){
|
||||
throw new ServiceException("验证码错误");
|
||||
}
|
||||
userService.resetPassword(user.getId(),password);
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user