From 9ec9a4987cb5fe4a3bd39d61029add5c51d3ab92 Mon Sep 17 00:00:00 2001 From: liujiandao <274878379@qq.com> Date: 星期六, 02 十二月 2023 11:00:26 +0800 Subject: [PATCH] 添加来源 --- service/client.go | 14 +++++++++++--- 1 files changed, 11 insertions(+), 3 deletions(-) diff --git a/service/client.go b/service/client.go index e889606..20cd752 100644 --- a/service/client.go +++ b/service/client.go @@ -44,7 +44,7 @@ return ecode.OK } -func (ClientService) UpdateClient(client *model.Client) int { +func (ClientService) UpdateClient(client *model.Client, conId int) int { // check client exist errCode := CheckClientExist(client.Id) if errCode != ecode.OK { @@ -67,13 +67,21 @@ for _, contact := range client.Contacts { contact.ClientId = client.Id - if contact.Id == 0 { - err := model.NewContactSearch(tx).Create(&contact) + + // check isFirst + errCode = setFirstContact(tx, &contact) + if errCode != ecode.OK { + return errCode + } + + if conId == 0 { + err = model.NewContactSearch(tx).Create(&contact) if err != nil { tx.Rollback() return ecode.ClientUpdateErr } } else { + contact.Id = conId err = model.NewContactSearch(tx).SetId(contact.Id).Update(&contact) if err != nil { tx.Rollback() -- Gitblit v1.8.0