jjxu/fonchain-artistinfo
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

优化邀请关系处理逻辑

Browse Source
This commit is contained in:
徐俊杰 2023-03-31 11:45:06 +08:00
parent 9315e15011
commit 143fde30e0
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
cmd/internal/controller/artistInfo_user.go
View File

@ -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,