325 lines
13 KiB
Java
325 lines
13 KiB
Java
package com.ruoyi.cai.auth;
|
||
|
||
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.baomidou.mybatisplus.core.toolkit.Wrappers;
|
||
import com.ruoyi.cai.domain.*;
|
||
import com.ruoyi.cai.dto.commom.ignoredata.InviteIgnoreData;
|
||
import com.ruoyi.cai.enums.CodeEnum;
|
||
import com.ruoyi.cai.enums.GenderEnum;
|
||
import com.ruoyi.cai.enums.IgnoreDataTypeEnum;
|
||
import com.ruoyi.cai.enums.SystemConfigEnum;
|
||
import com.ruoyi.cai.executor.ExecutorConstant;
|
||
import com.ruoyi.cai.manager.*;
|
||
import com.ruoyi.cai.service.*;
|
||
import com.ruoyi.cai.util.RandomSjUtil;
|
||
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.common.utils.ip.AddressUtils;
|
||
import com.ruoyi.system.service.SysLoginService;
|
||
import com.ruoyi.yunxin.Yunxin;
|
||
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.math.BigDecimal;
|
||
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 CodeManager codeManager;
|
||
@Autowired
|
||
private UserCodeGenService userCodeGenService;
|
||
@Autowired
|
||
private UserInfoService userInfoService;
|
||
@Autowired
|
||
private SystemConfigManager systemConfigManager;
|
||
@Autowired
|
||
private AccountService accountService;
|
||
@Autowired
|
||
private UserOnlineService userOnlineService;
|
||
@Autowired
|
||
private RedissonClient redissonClient;
|
||
@Autowired
|
||
private UserInviteService userInviteService;
|
||
@Autowired
|
||
private UnionService unionService;
|
||
@Autowired
|
||
private UnionUserService unionUserService;
|
||
@Resource
|
||
private ImUserClient imUserClient;
|
||
@Autowired
|
||
private UserCountService userCountService;
|
||
@Autowired
|
||
private UserForbidManager userForbidManager;
|
||
@Autowired
|
||
private AwardManager awardManager;
|
||
@Autowired
|
||
private SmsVerifyService smsVerifyService;
|
||
@Autowired
|
||
private IgnoreDataService ignoreDataService;
|
||
|
||
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 register(CaiRegisterUser caiUser) {
|
||
User user = userService.getByUsername(caiUser.getUsername());
|
||
if(user != null){
|
||
throw new ServiceException("手机号已存在");
|
||
}
|
||
if(!caiUser.getPassword().equals(caiUser.getPasswordCheck())){
|
||
throw new ServiceException("两次输入密码不一致，请检查");
|
||
}
|
||
boolean check = smsVerifyService.check(CodeEnum.REGISTER, caiUser.getUsername(), caiUser.getCode());
|
||
if(!check){
|
||
throw new ServiceException("验证码错误");
|
||
}
|
||
// 加锁
|
||
String lockKey = LockManager.getRegisterLockKey(caiUser.getUsername());
|
||
RLock lock = redissonClient.getLock(lockKey);
|
||
if(lock.isLocked()){
|
||
throw new ServiceException("您点击太快了");
|
||
}
|
||
lock.lock();
|
||
try {
|
||
CaiLoginManager bean = SpringUtil.getBean(CaiLoginManager.class);
|
||
user = bean.registerUser(caiUser);
|
||
}finally {
|
||
lock.unlock();
|
||
}
|
||
awardManager.giveRegisterAsync(user.getId());
|
||
if(user.getInviteId() != null){
|
||
User finalUser = user;
|
||
ExecutorConstant.COMMON_EXECUTOR.execute(() -> dealInviteId(finalUser,true));
|
||
}
|
||
return login(user);
|
||
}
|
||
|
||
|
||
|
||
public void dealInviteId(User user, boolean openIgnore){
|
||
Long inviteId = user.getInviteId();
|
||
if(inviteId == null){
|
||
return;
|
||
}
|
||
User inviteUser = userService.getById(inviteId);
|
||
if(inviteUser == null){
|
||
return;
|
||
}
|
||
String key = LockManager.getDealInviteLockKey(user.getId());
|
||
RLock lock = redissonClient.getLock(key);
|
||
if(lock.isLocked()){
|
||
log.warn("点击太快了，等一等，dealInviteId");
|
||
return;
|
||
}
|
||
lock.lock(5,TimeUnit.SECONDS);
|
||
try {
|
||
if(openIgnore){
|
||
Integer inviteBindRate = systemConfigManager.getSystemConfigOfInt(SystemConfigEnum.INVITE_BIND_RATE);
|
||
boolean sj = RandomSjUtil.rateSj(inviteBindRate);
|
||
if(!sj){
|
||
InviteIgnoreData data = new InviteIgnoreData();
|
||
data.setUserId(user.getId());
|
||
data.setInviteId(inviteId);
|
||
ignoreDataService.saveIgnoreData(IgnoreDataTypeEnum.INVITE,inviteUser.getId(),data);
|
||
return;
|
||
}
|
||
}
|
||
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));
|
||
}
|
||
// 处理工会
|
||
Union union = null; //
|
||
if(inviteUser.getIsUnion() == 1){
|
||
union = unionService.getByUserId(inviteUser.getId());
|
||
} else {
|
||
UnionUser unionUser = unionUserService.getByUserId(inviteUser.getId());
|
||
if(unionUser != null){
|
||
union = unionService.getById(unionUser);
|
||
}
|
||
}
|
||
if(union == null){
|
||
log.info("邀请人没有工会，无法入会 inviteId={}",inviteId);
|
||
return;
|
||
}
|
||
UnionUser checkUnionUser = unionUserService.getByUserId(user.getId());
|
||
if(checkUnionUser != null){
|
||
UnionUser unionUser = new UnionUser();
|
||
unionUser.setUnionId(union.getId());
|
||
unionUser.setUnionUserId(union.getUserId());
|
||
unionUser.setUserId(user.getId());
|
||
unionUser.setVideoDivide(systemConfigManager.getSystemConfigOfBigDecimal(SystemConfigEnum.DEFAULT_UNION_VIDEO_INCOME_RATE));
|
||
unionUser.setGiftDivide(systemConfigManager.getSystemConfigOfBigDecimal(SystemConfigEnum.DEFAULT_UNION_VIDEO_INCOME_RATE));
|
||
unionUser.setVipDivide(BigDecimal.ZERO);
|
||
unionUser.setEnableRate(true);
|
||
unionUserService.save(unionUser);
|
||
}
|
||
}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(CaiRegisterUser user){
|
||
String usercode = userCodeGenService.getCodeGen();
|
||
User add = new User();
|
||
add.setUsercode(usercode);
|
||
add.setNickname("用户"+usercode);
|
||
add.setType(0);
|
||
add.setPassword(BCrypt.hashpw(user.getPassword()));
|
||
add.setMobile(user.getUsername());
|
||
add.setGender(GenderEnum.NONE.getCode());
|
||
add.setAvatar(GenderEnum.NONE.getDefaultAvatar());
|
||
add.setInviteId(user.getInviteId());
|
||
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();
|
||
UserInfo userInfo = new UserInfo();
|
||
userInfo.setUserId(add.getId());
|
||
userInfo.setVideoIncomeRate(systemConfigManager.getSystemConfigOfBigDecimal(SystemConfigEnum.DEFAULT_VIDEO_INCOME_RATE));
|
||
userInfo.setGuardIncomeRate(systemConfigManager.getSystemConfigOfBigDecimal(SystemConfigEnum.DEFAULT_GUARD_INCOME_RATE));
|
||
userInfo.setGiftIncomeRate(systemConfigManager.getSystemConfigOfBigDecimal(SystemConfigEnum.DEFAULT_GIFT_INCOME_RATE));
|
||
userInfo.setPayIncomeRate(systemConfigManager.getSystemConfigOfBigDecimal(SystemConfigEnum.DEFAULT_PAY_INCOME_RATE));
|
||
userInfo.setLoginCount(1);
|
||
userInfo.setLastLoginIp(clientIP);
|
||
userInfo.setLastLoginTime(LocalDateTime.now());
|
||
userInfo.setRegIp(clientIP);
|
||
userInfo.setRegTime(LocalDateTime.now());
|
||
userInfoService.save(userInfo);
|
||
Account account = new Account();
|
||
account.setUserId(add.getId());
|
||
accountService.save(account);
|
||
UserOnline userOnline = new UserOnline();
|
||
userOnline.setUserId(add.getId());
|
||
userOnlineService.save(userOnline);
|
||
UserCount userCount = new UserCount();
|
||
userCount.setUserId(add.getId());
|
||
userCountService.save(userCount);
|
||
return add;
|
||
}
|
||
|
||
public void resetPassword(ResetPasswordReq code) {
|
||
User user = userService.getByUsername(code.getMobile());
|
||
if(user == null){
|
||
throw new ServiceException("账户不存在");
|
||
}
|
||
boolean check = smsVerifyService.check(CodeEnum.RESET_PASSWORD, code.getMobile(), code.getCode());
|
||
if(!check){
|
||
throw new ServiceException("验证码错误");
|
||
}
|
||
userService.resetPassword(user.getId(),code.getPassword());
|
||
}
|
||
}
|