From cddf1d93735052590755460f4b3c8264e0d57399 Mon Sep 17 00:00:00 2001 From: wangpengfei <274878379@qq.com> Date: 星期一, 07 八月 2023 09:39:20 +0800 Subject: [PATCH] fix --- service/serviceFeeManage.go | 152 +++++++++++++++++++++++++------------------------- 1 files changed, 76 insertions(+), 76 deletions(-) diff --git a/service/serviceFeeManage.go b/service/serviceFeeManage.go index a1ab67b..54c8b8f 100644 --- a/service/serviceFeeManage.go +++ b/service/serviceFeeManage.go @@ -1,76 +1,76 @@ -package service - -import ( - "aps_crm/model" - "aps_crm/pkg/ecode" - "aps_crm/pkg/mysqlx" -) - -type FeeManageService struct{} - -func (FeeManageService) AddServiceFeeManage(serviceFeeManage *model.ServiceFeeManage) int { - - tx := mysqlx.GetDB().Begin() - - err := model.NewClientSearch(tx).Create(serviceFeeManage.Client) - if err != nil { - tx.Rollback() - return ecode.ClientExist - } - - serviceFeeManage.ClientId = serviceFeeManage.Client.Id - err = model.NewServiceFeeManageSearch(tx).Create(serviceFeeManage) - if err != nil { - tx.Rollback() - return ecode.ServiceFeeManageExist - } - - tx.Commit() - - return ecode.OK -} - -func (FeeManageService) UpdateServiceFeeManage(serviceFeeManage *model.ServiceFeeManage) int { - // check serviceFeeManage exist - _, err := model.NewServiceFeeManageSearch(nil).SetId(serviceFeeManage.Id).Find() - if err != nil { - return ecode.ServiceFeeManageNotExist - } - - tx := mysqlx.GetDB().Begin() - - err = model.NewServiceFeeManageSearch(tx).SetId(serviceFeeManage.Id).Update(serviceFeeManage) - if err != nil { - tx.Rollback() - return ecode.ServiceFeeManageSetErr - } - - // update client - err = model.NewClientSearch(tx).SetId(serviceFeeManage.ClientId).Update(serviceFeeManage.Client) - if err != nil { - tx.Rollback() - return ecode.ClientUpdateErr - } - - tx.Commit() - - return ecode.OK -} - -func (FeeManageService) GetServiceFeeManageList(page, pageSize int, keyword string) ([]*model.ServiceFeeManage, int64, int) { - // get contact list - contacts, total, err := model.NewServiceFeeManageSearch(nil).SetKeyword(keyword).SetPage(page, pageSize).FindAll() - if err != nil { - return nil, 0, ecode.ServiceFeeManageListErr - } - return contacts, total, ecode.OK -} - -func (FeeManageService) DeleteServiceFeeManage(ids []int) int { - // delete client - err := model.NewServiceFeeManageSearch(nil).SetIds(ids).Delete() - if err != nil { - return ecode.ServiceFeeManageDeleteErr - } - return ecode.OK -} +package service + +import ( + "aps_crm/model" + "aps_crm/pkg/ecode" + "aps_crm/pkg/mysqlx" +) + +type FeeManageService struct{} + +func (FeeManageService) AddServiceFeeManage(serviceFeeManage *model.ServiceFeeManage) int { + + tx := mysqlx.GetDB().Begin() + + err := model.NewClientSearch(tx).Create(serviceFeeManage.Client) + if err != nil { + tx.Rollback() + return ecode.ClientExist + } + + serviceFeeManage.ClientId = serviceFeeManage.Client.Id + err = model.NewServiceFeeManageSearch(tx).Create(serviceFeeManage) + if err != nil { + tx.Rollback() + return ecode.ServiceFeeManageExist + } + + tx.Commit() + + return ecode.OK +} + +func (FeeManageService) UpdateServiceFeeManage(serviceFeeManage *model.ServiceFeeManage) int { + // check serviceFeeManage exist + _, err := model.NewServiceFeeManageSearch(nil).SetId(serviceFeeManage.Id).Find() + if err != nil { + return ecode.ServiceFeeManageNotExist + } + + tx := mysqlx.GetDB().Begin() + + err = model.NewServiceFeeManageSearch(tx).SetId(serviceFeeManage.Id).Update(serviceFeeManage) + if err != nil { + tx.Rollback() + return ecode.ServiceFeeManageSetErr + } + + // update client + err = model.NewClientSearch(tx).SetId(serviceFeeManage.ClientId).Update(serviceFeeManage.Client) + if err != nil { + tx.Rollback() + return ecode.ClientUpdateErr + } + + tx.Commit() + + return ecode.OK +} + +func (FeeManageService) GetServiceFeeManageList(page, pageSize int, keyword string) ([]*model.ServiceFeeManage, int64, int) { + // get contact list + contacts, total, err := model.NewServiceFeeManageSearch(nil).SetKeyword(keyword).SetPage(page, pageSize).FindAll() + if err != nil { + return nil, 0, ecode.ServiceFeeManageListErr + } + return contacts, total, ecode.OK +} + +func (FeeManageService) DeleteServiceFeeManage(ids []int) int { + // delete client + err := model.NewServiceFeeManageSearch(nil).SetIds(ids).Delete() + if err != nil { + return ecode.ServiceFeeManageDeleteErr + } + return ecode.OK +} -- Gitblit v1.8.0