From 143fde30e06720417527da4217f95910c9ea32c3 Mon Sep 17 00:00:00 2001 From: jjxu <428192774@qq.com> Date: Fri, 31 Mar 2023 11:45:06 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BC=98=E5=8C=96=E9=82=80=E8=AF=B7=E5=85=B3?= =?UTF-8?q?=E7=B3=BB=E5=A4=84=E7=90=86=E9=80=BB=E8=BE=91?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- cmd/internal/controller/artistInfo_user.go | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/cmd/internal/controller/artistInfo_user.go b/cmd/internal/controller/artistInfo_user.go index d67f3e3..f859308 100644 --- a/cmd/internal/controller/artistInfo_user.go +++ b/cmd/internal/controller/artistInfo_user.go @@ -171,7 +171,7 @@ func (a *ArtistInfoUserProvider) BindInviteInvitedAccount(ctx context.Context, i // 查询被邀请人是否存在 var invitedUser model.User if in.InvitedUserId == 0 { - if err = db.DB.Debug().Where("invited_code = ?", in.InviteCode).First(&invitedUser).Error; err != nil { + if err = db.DB.Debug().Where("invited_code = ? OR id = ?", in.InviteCode, in.InvitedUserId).First(&invitedUser).Error; err != nil { if err == gorm.ErrRecordNotFound { return nil, errors.New("被邀请人不存在") } else { @@ -196,9 +196,9 @@ func (a *ArtistInfoUserProvider) BindInviteInvitedAccount(ctx context.Context, i return nil, err } data.UserId = int32(inviteUser.ID) - data.InvitedId = in.InvitedUserId - data.InvitedCode = in.InvitedCode - data.InviteCode = in.InviteCode + data.InvitedId = int32(invitedUser.ID) + data.InvitedCode = invitedUser.InviteCode + data.InviteCode = inviteUser.InviteCode //如果数据存更新刷新时间 if err = db.DB.Debug().Save(&data).Error; err != nil { return nil, err @@ -211,7 +211,7 @@ func (a *ArtistInfoUserProvider) BindInviteInvitedAccount(ctx context.Context, i } if countData.ID == 0 { countData = model.UserInvited{ - UserId: int32(thisUser.ID), + UserId: int32(inviteUser.ID), InvitedUserId: in.InvitedUserId, //受邀请人的画家宝用户id Count: 1, InvitedCode: in.InvitedCode,