邀请新用户记录添加

彭雪彬

2025-10-22 06dc40189efbf8464fe8d629c32b95fe02a76240

 oying-system/src/main/java/com/oying/modules/system/service/impl/InvitationsServiceImpl.java

@@ -1,10 +1,12 @@
package com.oying.modules.system.service.impl;

import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.oying.modules.system.domain.Invitations;
import com.oying.modules.system.domain.User;
import com.oying.modules.system.domain.dto.InvitationsVo;
import com.oying.modules.system.mapper.UserMapper;
import com.oying.utils.FileUtil;
import com.oying.utils.SecurityUtils;
import lombok.RequiredArgsConstructor;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
@@ -95,7 +97,15 @@

    @Override
    public Object addInvitations(InvitationsVo invitationsVo) {
        // SecurityUtils.getCurrentUserId();
        // 判断是否为新用户
        if(userMapper.selectById(invitationsVo.getNewUserId()) == null) {
            throw new RuntimeException("用户不存在");
        }
        Invitations selectOne = invitationsMapper.selectOne(new QueryWrapper<Invitations>().eq("new_user_id", invitationsVo.getNewUserId()));
        if(selectOne != null) {
            throw new RuntimeException("用户已被邀请过！");
        }
        // 查询邀请人信息
        User user = userMapper.selectById(invitationsVo.getInviterId());
        Invitations invitations = new Invitations();
@@ -103,13 +113,13 @@
        invitations.setInviterPhone(user.getUsername());
        invitations.setInviterName(user.getNickName());
        invitations.setInviterOpenId(user.getOpenid());
        if(invitationsVo.getUserType().equals("QS")) {
        if(invitationsVo.getUserType().equals("QS") || invitationsVo.getUserType().equals("MJQS")) {
            invitations.setInviterOpenId(user.getRiderOpenId());
        }
        // 查询被邀请人的信息
        User newUser = userMapper.selectById(invitationsVo.getNewUserId());
        invitations.setNewUserOpenId(newUser.getOpenid());
        if(invitationsVo.getUserType().equals("QS")) {
        if(invitationsVo.getUserType().equals("QS") || invitationsVo.getUserType().equals("MJQS")) {
            invitations.setNewUserOpenId(newUser.getRiderOpenId());
        }
        invitations.setNewUserId(invitationsVo.getNewUserId());