From 81b6e4fb5859e273ba54d04e65739733eb7efeee Mon Sep 17 00:00:00 2001 From: wangpengfei <274878379@qq.com> Date: 星期五, 21 七月 2023 15:55:53 +0800 Subject: [PATCH] fix --- service/contact.go | 22 +++++++++++----------- 1 files changed, 11 insertions(+), 11 deletions(-) diff --git a/service/contact.go b/service/contact.go index 59bf587..c62bc8a 100644 --- a/service/contact.go +++ b/service/contact.go @@ -3,6 +3,7 @@ import ( "aps_crm/model" "aps_crm/pkg/ecode" + "aps_crm/pkg/mysqlx" "gorm.io/gorm" ) @@ -15,15 +16,14 @@ return code } - tx := model.NewContactSearch().Orm.Begin() - + tx := mysqlx.GetDB().Begin() // check isFirst errCode := setFirstContact(tx, contact) if errCode != ecode.OK { return errCode } - err := model.NewContactSearch().Create(contact) + err := model.NewContactSearch(tx).Create(contact) if err != nil { tx.Rollback() return ecode.ContactExist @@ -35,13 +35,13 @@ func (ContactService) DeleteContact(id int) int { // check contact exist - _, err := model.NewContactSearch().SetId(id).First() + _, err := model.NewContactSearch(nil).SetId(id).First() if err != nil { return ecode.ContactNotExist } // delete contact - err = model.NewContactSearch().SetId(id).Delete() + err = model.NewContactSearch(nil).SetId(id).Delete() if err != nil { return ecode.ContactDeleteErr } @@ -50,7 +50,7 @@ func (ContactService) GetContactList() ([]*model.ContactDetail, int) { // get contact list - contacts, err := model.NewContactSearch().FindAll() + contacts, err := model.NewContactSearch(nil).FindAll() if err != nil { return nil, ecode.ContactListErr } @@ -59,7 +59,7 @@ func (ContactService) UpdateContact(contact *model.Contact) int { // check contact exist - _, err := model.NewContactSearch().SetId(contact.Id).First() + _, err := model.NewContactSearch(nil).SetId(contact.Id).First() if err != nil { return ecode.ContactNotExist } @@ -69,14 +69,14 @@ return code } - tx := model.NewContactSearch().Orm.Begin() + tx := mysqlx.GetDB().Begin() code = setFirstContact(tx, contact) if code != ecode.OK { return code } // update contact - err = model.NewContactSearch().SetId(contact.Id).Update(contact) + err = model.NewContactSearch(tx).SetId(contact.Id).Update(contact) if err != nil { tx.Rollback() return ecode.ContactUpdateErr @@ -100,7 +100,7 @@ func setFirstContact(tx *gorm.DB, contact *model.Contact) int { if contact.IsFirst && contact.ClientId != 0 { - err := model.NewContactSearch().SetClientId(contact.ClientId).UpdateByMap(map[string]interface{}{ + err := model.NewContactSearch(tx).SetClientId(contact.ClientId).UpdateByMap(map[string]interface{}{ "is_first": false, }) if err != nil { @@ -114,7 +114,7 @@ // CheckContactExist check contact exist func CheckContactExist(id int) int { - tmp, err := model.NewContactSearch().SetId(id).First() + tmp, err := model.NewContactSearch(nil).SetId(id).First() if err != nil { return ecode.ContactNotExist } -- Gitblit v1.8.0