This commit is contained in:
dute7liang
2024-01-02 02:01:53 +08:00
parent 59341aa956
commit 3569bad84e
7 changed files with 106 additions and 5 deletions

View File

@@ -3,6 +3,7 @@ package com.ruoyi.cai.service;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.baomidou.mybatisplus.extension.service.IService;
import com.ruoyi.cai.domain.UserInvite;
import com.ruoyi.cai.dto.admin.vo.UserInviteAdminVo;
import com.ruoyi.cai.dto.app.vo.invite.InviteHomeVo;
import com.ruoyi.cai.dto.app.vo.user.UserListVo;
import com.ruoyi.common.core.domain.PageQuery;
@@ -20,4 +21,8 @@ public interface UserInviteService extends IService<UserInvite> {
InviteHomeVo inviteHome(Long userId);
Page<UserListVo> inviteUserPage(PageQuery pageQuery, Long userId);
Page<UserInviteAdminVo> pageAdmin(PageQuery pageQuery, UserInviteAdminVo bo);
boolean updateUserInviteRate(UserInvite bo);
}

View File

@@ -3,16 +3,21 @@ package com.ruoyi.cai.service.impl;
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.domain.UserInfo;
import com.ruoyi.cai.domain.UserInvite;
import com.ruoyi.cai.dto.admin.vo.UserInviteAdminVo;
import com.ruoyi.cai.dto.app.dto.InviteCountDTO;
import com.ruoyi.cai.dto.app.vo.invite.InviteHomeVo;
import com.ruoyi.cai.dto.app.vo.user.UserListVo;
import com.ruoyi.cai.mapper.UserInviteMapper;
import com.ruoyi.cai.service.UserInfoService;
import com.ruoyi.cai.service.UserInviteService;
import com.ruoyi.cai.service.UserService;
import com.ruoyi.common.core.domain.PageQuery;
import com.ruoyi.common.exception.ServiceException;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
import java.util.ArrayList;
import java.util.List;
@@ -26,14 +31,16 @@ import java.util.List;
@Service
public class UserInviteServiceImpl extends ServiceImpl<UserInviteMapper, UserInvite> implements UserInviteService {
@Autowired
private UserService userService;
@Autowired
private UserInfoService userInfoService;
@Override
public UserInvite getByUserId(Long id) {
return this.getOne(Wrappers.lambdaQuery(UserInvite.class).eq(UserInvite::getUserId,id).last("limit 1"));
}
@Autowired
private UserService userService;
@Override
public InviteHomeVo inviteHome(Long userId) {
InviteHomeVo vo = new InviteHomeVo();
@@ -55,4 +62,38 @@ public class UserInviteServiceImpl extends ServiceImpl<UserInviteMapper, UserInv
public Page<UserListVo> inviteUserPage(PageQuery pageQuery, Long userId) {
return baseMapper.inviteUserPage(pageQuery.build(),userId);
}
@Override
public Page<UserInviteAdminVo> pageAdmin(PageQuery pageQuery, UserInviteAdminVo bo) {
return baseMapper.pageAdmin(pageQuery.build(), bo);
}
@Override
@Transactional(rollbackFor = Exception.class)
public boolean updateUserInviteRate(UserInvite bo) {
UserInvite userInvite = this.getById(bo.getId());
if(userInvite == null){
throw new ServiceException("数据不存在");
}
UserInvite update = new UserInvite();
update.setId(bo.getId());
update.setGiftIncomeRate(bo.getGiftIncomeRate());
update.setVideoIncomeRate(bo.getVideoIncomeRate());
update.setGuardIncomeRate(bo.getGuardIncomeRate());
update.setPayIncomeRate(bo.getPayIncomeRate());
boolean b = this.updateById(update);
if(b){
UserInfo userInfo = new UserInfo();
userInfo.setGiftIncomeRate(bo.getGiftIncomeRate());
userInfo.setVideoIncomeRate(bo.getVideoIncomeRate());
userInfo.setGuardIncomeRate(bo.getGuardIncomeRate());
userInfo.setPayIncomeRate(bo.getPayIncomeRate());
userInfoService.update(Wrappers.lambdaUpdate(UserInfo.class).eq(UserInfo::getUserId,userInvite.getUserId())
.set(UserInfo::getGiftIncomeRate,bo.getGiftIncomeRate())
.set(UserInfo::getVideoIncomeRate,bo.getVideoIncomeRate())
.set(UserInfo::getGuardIncomeRate,bo.getGuardIncomeRate())
.set(UserInfo::getPayIncomeRate,bo.getPayIncomeRate()));
}
return b;
}
}