From 9c2489c0c360c8dda36d3cbe1dba79222096b2dd Mon Sep 17 00:00:00 2001 From: wangpengfei <274878379@qq.com> Date: 星期一, 07 八月 2023 10:16:39 +0800 Subject: [PATCH] Merge branch 'master' into fly --- service/serviceFeeManage.go | 157 +++++++++++++++++++++++++++------------------------- 1 files changed, 81 insertions(+), 76 deletions(-) diff --git a/service/serviceFeeManage.go b/service/serviceFeeManage.go index 54c8b8f..49fa66c 100644 --- a/service/serviceFeeManage.go +++ b/service/serviceFeeManage.go @@ -1,76 +1,81 @@ -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/constvar" + "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) DeleteServiceFeeManage(ids []int) int { + // delete client + err := model.NewServiceFeeManageSearch(nil).SetIds(ids).Delete() + if err != nil { + return ecode.ServiceFeeManageDeleteErr + } + return ecode.OK +} + +func (FeeManageService) GetServiceFeeManageList(page, pageSize int, queryClass constvar.ServiceFeeQueryClass, keywordType constvar.ServiceFeeKeywordType, keyword string) ([]*model.ServiceFeeManage, int64, int) { + // get contact list + contacts, total, err := model.NewServiceFeeManageSearch(nil). + SetQueryClass(queryClass). + SetKeywordType(keywordType). + SetKeyword(keyword). + SetPage(page, pageSize).FindAll() + if err != nil { + return nil, 0, ecode.ServiceFeeManageListErr + } + return contacts, total, ecode.OK +} -- Gitblit v1.8.0