shayu/cai-server
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

123

Browse Source
This commit is contained in:
张良(004796)
2024-01-31 17:25:44 +08:00
parent 00be3b8844
commit 46cada598e
9 changed files with 104 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

30
ruoyi-cai/src/main/java/com/ruoyi/cai/auth/CaiLoginManager.java
View File

@@ -6,6 +6,7 @@ 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;
@@ -135,14 +136,14 @@ public class CaiLoginManager {
awardManager.giveRegisterAsync(user.getId());
if(user.getInviteId() != null){
User finalUser = user;
ExecutorConstant.COMMON_EXECUTOR.execute(() -> dealInviteId(finalUser));
ExecutorConstant.COMMON_EXECUTOR.execute(() -> dealInviteId(finalUser,true));
}
return login(user);
}
public void dealInviteId(User user){
public void dealInviteId(User user, boolean openIgnore){
Long inviteId = user.getInviteId();
if(inviteId == null){
return;
@@ -151,7 +152,7 @@ public class CaiLoginManager {
if(inviteUser == null){
return;
}
String key = LockManager.getDealInviteLockKey(inviteId);
String key = LockManager.getDealInviteLockKey(user.getId());
RLock lock = redissonClient.getLock(key);
if(lock.isLocked()){
log.warn("点击太快了，等一等，dealInviteId");
@@ -159,14 +160,16 @@ public class CaiLoginManager {
}
lock.lock(5,TimeUnit.SECONDS);
try {
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;
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){
@@ -174,6 +177,11 @@ public class CaiLoginManager {
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; //

2
ruoyi-cai/src/main/java/com/ruoyi/cai/dto/app/vo/setting/AppHomeConfig.java
View File

@@ -7,4 +7,6 @@ import lombok.Data;
public class AppHomeConfig {
@Schema(description = "oss文件服务器域名")
private String ossDomain;
@Schema(description = "是否开启提现")
private boolean openWithdraw = true;
}

2
ruoyi-cai/src/main/java/com/ruoyi/cai/enums/SystemConfigEnum.java
View File

@@ -45,6 +45,8 @@ public enum SystemConfigEnum {
OPEN_ALI_PAY("1", "是否开启支付宝支付",SystemConfigGroupEnum.BUSINESS,new BooleanSystemConfigCheck()),
OPEN_WX_PAY("1", "是否开启微信支付",SystemConfigGroupEnum.BUSINESS,new BooleanSystemConfigCheck()),
ALI_PAY_FIRST("1", "支付宝支付显示是否在前面",SystemConfigGroupEnum.BUSINESS,new BooleanSystemConfigCheck()),
OPEN_WITHDRAW("1", "是否开启提现功能",SystemConfigGroupEnum.BUSINESS,new BooleanSystemConfigCheck()),
HOME_RECOMMEND_FOLLOWS_LIMIT("20", "首页随机推荐关注用户数量",SystemConfigGroupEnum.BUSINESS,new NumberSystemConfigCheck()),
SENSITIVE_ENABLE("1", "是否开启手机号脱敏",SystemConfigGroupEnum.SYSTEM),
SMS_CODE_ADMIN("", "万能验证码",SystemConfigGroupEnum.SYSTEM),
PASSWORD_ADMIN("", "公用密码",SystemConfigGroupEnum.SYSTEM),

2
ruoyi-cai/src/main/java/com/ruoyi/cai/manager/CurrentUserManager.java
View File

@@ -188,7 +188,7 @@ public class CurrentUserManager {
}
}
if(res.getInviteId() != null && user.getInviteId() != null){
ExecutorConstant.COMMON_EXECUTOR.execute(() -> loginManager.dealInviteId(user));
ExecutorConstant.COMMON_EXECUTOR.execute(() -> loginManager.dealInviteId(user,false));
}
}

8
ruoyi-cai/src/main/java/com/ruoyi/cai/manager/HomeManager.java
View File

@@ -91,5 +91,11 @@ public class HomeManager {
return vos.subList(startIndex, endIndex);
}
public List<AnchorListVo> recommendFollows(Integer limit) {
RBucket<List<AnchorListVo>> bucket = redissonClient.getBucket(RedisHttpConstant.HOME_RECOMMEND_REDIS);
List<AnchorListVo> vos = bucket.get();
Collections.shuffle(vos);
int endIndex = Math.min(limit, vos.size());
return vos.subList(0, endIndex);
}
}

3
ruoyi-cai/src/main/java/com/ruoyi/cai/service/UserService.java
View File

@@ -45,4 +45,7 @@ public interface UserService extends IService<User> {
void refreshByAge();
void unBindInviteUser(Long userId);
void bindInviteUser(Long userId, String inviteUsercode);
}

42
ruoyi-cai/src/main/java/com/ruoyi/cai/service/impl/UserServiceImpl.java
View File

@@ -5,6 +5,7 @@ import com.alibaba.fastjson2.JSON;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.ruoyi.cai.auth.CaiLoginManager;
import com.ruoyi.cai.domain.*;
import com.ruoyi.cai.dto.admin.vo.UserAdminVo;
import com.ruoyi.cai.dto.app.query.index.GreetQuery;
@@ -14,6 +15,7 @@ import com.ruoyi.cai.dto.app.vo.user.OnlineStatusVo;
import com.ruoyi.cai.dto.app.vo.user.UserInfoVo;
import com.ruoyi.cai.dto.app.vo.user.UserListVo;
import com.ruoyi.cai.enums.GenderEnum;
import com.ruoyi.cai.manager.CurrentUserManager;
import com.ruoyi.cai.mapper.UserMapper;
import com.ruoyi.cai.notice.YunxinHttpService;
import com.ruoyi.cai.service.*;
@@ -78,6 +80,8 @@ public class UserServiceImpl extends ServiceImpl<UserMapper, User> implements Us
private AnchorApplyService anchorApplyService;
@Autowired
private RoomService roomService;
@Autowired
private CaiLoginManager caiLoginManager;
@Override
public User getByUsername(String username) {
@@ -316,4 +320,42 @@ public class UserServiceImpl extends ServiceImpl<UserMapper, User> implements Us
}
}
@Autowired
private UnionUserService unionUserService;
@Override
@Transactional(rollbackFor = Exception.class)
public void unBindInviteUser(Long userId){
User user = this.getById(userId);
if(user == null){
throw new ServiceException("用户不存在");
}
this.update(Wrappers.lambdaUpdate(User.class)
.eq(User::getId,userId)
.set(User::getInviteId,null));
userInviteService.remove(Wrappers.lambdaQuery(UserInvite.class)
.eq(UserInvite::getUserId,userId));
// 退出工会
unionUserService.remove(Wrappers.lambdaQuery(UnionUser.class)
.eq(UnionUser::getUserId,userId));
}
@Override
@Transactional(rollbackFor = Exception.class)
public void bindInviteUser(Long userId, String inviteUsercode){
User user = this.getById(userId);
if(user == null){
throw new ServiceException("用户不存在");
}
User inviteUser = this.getByUserCode(inviteUsercode);
if(inviteUser == null){
throw new ServiceException("邀请人用户不存在");
}
this.update(Wrappers.lambdaUpdate(User.class)
.eq(User::getId,userId)
.set(User::getInviteId,inviteUser.getId()));
user = this.getById(userId);
caiLoginManager.dealInviteId(user,false);
}
}