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,