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

Merge branch 'xjjdev'

Browse Source
This commit is contained in:
徐俊杰 2023-03-31 11:42:58 +08:00
commit 73264cde5d
1 changed files with 16 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -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) { 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 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("被邀请人不存在")
} else {
return nil, err
}
}
}
// 查询邀请人是否存在
var inviteUser model.User
if err = db.DB.Debug().Where("invited_code = ?", in.InviteCode).First(&inviteUser).Error; err != nil {
if err == gorm.ErrRecordNotFound { if err == gorm.ErrRecordNotFound {
return nil, errors.New("邀请人不存在") return nil, errors.New("邀请人不存在")
} else { } else {
return nil, err return nil, err
} }
} }
} fmt.Println("邀请人userId:", inviteUser.ID)
fmt.Println("邀请人userId:", thisUser.ID) fmt.Println("被邀请人userId:", invitedUser.ID)
var data model.Invite var data model.Invite
// 受邀请者只能绑定一个邀请人 // 受邀请者只能绑定一个邀请人
if err = db.DB.Debug().Where("invite_code = ? AND invited_code = ?", in.InviteCode, in.InvitedCode).Find(&data).Error; err != nil { if err = db.DB.Debug().Where("invite_code = ? AND invited_code = ?", in.InviteCode, in.InvitedCode).Find(&data).Error; err != nil {
return nil, err return nil, err
} }
data.UserId = int32(thisUser.ID) data.UserId = int32(inviteUser.ID)
data.InvitedId = in.InvitedUserId data.InvitedId = in.InvitedUserId
data.InvitedCode = in.InvitedCode data.InvitedCode = in.InvitedCode
data.InviteCode = in.InviteCode data.InviteCode = in.InviteCode