diff --git a/cmd/internal/controller/artistInfo_user.go b/cmd/internal/controller/artistInfo_user.go index 95a1325..0994ae9 100644 --- a/cmd/internal/controller/artistInfo_user.go +++ b/cmd/internal/controller/artistInfo_user.go @@ -184,16 +184,13 @@ func (a *ArtistInfoUserProvider) BindInviteInvitedAccount(ctx context.Context, i if err = db.DB.Where("invited_id = ?", in.InvitedUserId).Find(&data).Error; err != nil { return nil, err } - if data.ID == 0 { - data = model.Invite{ - UserId: int32(thisUser.ID), - InvitedId: in.InvitedUserId, - InvitedCode: in.InvitedCode, - InviteCode: in.InviteCode, - } - if err = db.DB.Create(&data).Error; err != nil { - return nil, err - } + data.UserId = int32(thisUser.ID) + data.InvitedId = in.InvitedUserId + data.InvitedCode = in.InvitedCode + data.InviteCode = in.InviteCode + //如果数据存更新刷新时间 + if err = db.DB.Save(&data).Error; err != nil { + return nil, err } // 添加到次数统计