From 9315e1501182905ba5f36362947429ff46620944 Mon Sep 17 00:00:00 2001
From: jjxu <428192774@qq.com>
Date: Fri, 31 Mar 2023 11:42:49 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8D=E6=9F=A5=E8=AF=A2=E4=B8=8D?=
 =?UTF-8?q?=E5=88=B0=E9=82=80=E8=AF=B7=E4=BA=BAid=E7=9A=84=E9=97=AE?=
 =?UTF-8?q?=E9=A2=98?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 cmd/internal/controller/artistInfo_user.go | 22 ++++++++++++++++------
 1 file changed, 16 insertions(+), 6 deletions(-)

diff --git a/cmd/internal/controller/artistInfo_user.go b/cmd/internal/controller/artistInfo_user.go
index ea2909c..d67f3e3 100644
--- a/cmd/internal/controller/artistInfo_user.go
+++ b/cmd/internal/controller/artistInfo_user.go
@@ -168,24 +168,34 @@ func (a *ArtistInfoUserProvider) GetUserMsg(ctx context.Context, req *artistInfo
 
 // 绑定邀请人和受邀请人的账号,并加入到次数统计
 func (a *ArtistInfoUserProvider) BindInviteInvitedAccount(ctx context.Context, in *artistInfoUser.BindInviteInvitedAccountRequest) (res *artistInfoUser.BindInviteInvitedAccountRespond, err error) {
-	// 查询邀请人是否存在
-	var thisUser model.User
+	// 查询被邀请人是否存在
+	var invitedUser model.User
 	if in.InvitedUserId == 0 {
-		if err = db.DB.Debug().Where("invited_code = ?", in.InviteCode).First(&thisUser).Error; err != nil {
+		if err = db.DB.Debug().Where("invited_code = ?", in.InviteCode).First(&invitedUser).Error; err != nil {
 			if err == gorm.ErrRecordNotFound {
-				return nil, errors.New("邀请人不存在")
+				return nil, errors.New("被邀请人不存在")
 			} else {
 				return nil, err
 			}
 		}
 	}
-	fmt.Println("邀请人userId:", thisUser.ID)
+	// 查询邀请人是否存在
+	var inviteUser model.User
+	if err = db.DB.Debug().Where("invited_code = ?", in.InviteCode).First(&inviteUser).Error; err != nil {
+		if err == gorm.ErrRecordNotFound {
+			return nil, errors.New("邀请人不存在")
+		} else {
+			return nil, err
+		}
+	}
+	fmt.Println("邀请人userId:", inviteUser.ID)
+	fmt.Println("被邀请人userId:", invitedUser.ID)
 	var data model.Invite
 	// 受邀请者只能绑定一个邀请人
 	if err = db.DB.Debug().Where("invite_code = ? AND invited_code = ?", in.InviteCode, in.InvitedCode).Find(&data).Error; err != nil {
 		return nil, err
 	}
-	data.UserId = int32(thisUser.ID)
+	data.UserId = int32(inviteUser.ID)
 	data.InvitedId = in.InvitedUserId
 	data.InvitedCode = in.InvitedCode
 	data.InviteCode = in.InviteCode