From 5fe484ee4989f5b93228f951cf50ec6ab3288b61 Mon Sep 17 00:00:00 2001 From: jjxu <428192774@qq.com> Date: Thu, 13 Apr 2023 14:14:11 +0800 Subject: [PATCH 1/3] =?UTF-8?q?=E4=BF=AE=E5=A4=8D=E7=89=B9=E5=AE=9A?= =?UTF-8?q?=E6=83=85=E5=86=B5=E4=B8=8B=E4=BC=9A=E9=87=8D=E5=A4=8D=E5=88=9B?= =?UTF-8?q?=E5=BB=BA=E9=82=80=E8=AF=B7=E5=85=B3=E7=B3=BB=E7=9A=84=E9=97=AE?= =?UTF-8?q?=E9=A2=98?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- cmd/internal/controller/artistInfo_user.go | 23 ++++++++++++++-------- 1 file changed, 15 insertions(+), 8 deletions(-) diff --git a/cmd/internal/controller/artistInfo_user.go b/cmd/internal/controller/artistInfo_user.go index dcf29a3..f0ad4b3 100644 --- a/cmd/internal/controller/artistInfo_user.go +++ b/cmd/internal/controller/artistInfo_user.go @@ -192,16 +192,23 @@ func (a *ArtistInfoUserProvider) BindInviteInvitedAccount(ctx context.Context, i fmt.Println("被邀请人userId:", invitedUser.ID) 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("invited_code = ?", in.InvitedCode).Find(&data).Error; err != nil { return nil, err } - data.UserId = int32(inviteUser.ID) - 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 + if data.ID == 0 { + //数据不存在则创建 + data.UserId = int32(inviteUser.ID) + data.InvitedId = int32(invitedUser.ID) + data.InvitedCode = invitedUser.InviteCode + data.InviteCode = inviteUser.InviteCode + if err = db.DB.Debug().Create(&data).Error; err != nil { + return nil, err + } + } else if data.InviteCode == in.InviteCode { + //如果数据存更新刷新时间 + if err = db.DB.Debug().Save(&data).Error; err != nil { + return nil, err + } } // 添加到次数统计 From a05a7f9153c1063f9657b688936de31419bea95d Mon Sep 17 00:00:00 2001 From: jjxu <428192774@qq.com> Date: Thu, 13 Apr 2023 14:15:30 +0800 Subject: [PATCH 2/3] =?UTF-8?q?=E6=9B=B4=E6=96=B0=E6=B3=A8=E9=87=8A?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- cmd/internal/controller/artistInfo_user.go | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/cmd/internal/controller/artistInfo_user.go b/cmd/internal/controller/artistInfo_user.go index f0ad4b3..b9027b3 100644 --- a/cmd/internal/controller/artistInfo_user.go +++ b/cmd/internal/controller/artistInfo_user.go @@ -205,7 +205,7 @@ func (a *ArtistInfoUserProvider) BindInviteInvitedAccount(ctx context.Context, i return nil, err } } else if data.InviteCode == in.InviteCode { - //如果数据存更新刷新时间 + //如果数据存在更新刷新时间 if err = db.DB.Debug().Save(&data).Error; err != nil { return nil, err } From 7f8760a7eb0e0312cd8e5106ce761b501d19a2f4 Mon Sep 17 00:00:00 2001 From: jjxu <428192774@qq.com> Date: Thu, 13 Apr 2023 14:16:16 +0800 Subject: [PATCH 3/3] =?UTF-8?q?=E4=BF=AE=E6=94=B9=E6=B3=A8=E9=87=8A?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- cmd/internal/controller/artistInfo_user.go | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/cmd/internal/controller/artistInfo_user.go b/cmd/internal/controller/artistInfo_user.go index b9027b3..c109e2d 100644 --- a/cmd/internal/controller/artistInfo_user.go +++ b/cmd/internal/controller/artistInfo_user.go @@ -205,7 +205,7 @@ func (a *ArtistInfoUserProvider) BindInviteInvitedAccount(ctx context.Context, i return nil, err } } else if data.InviteCode == in.InviteCode { - //如果数据存在更新刷新时间 + //如果数据存在则刷新更新时间 if err = db.DB.Debug().Save(&data).Error; err != nil { return nil, err }