From 8fab255a4509a7ec5bfc7439b6e352054d677e29 Mon Sep 17 00:00:00 2001 From: zhangqian <zhangqian@123.com> Date: 星期四, 10 八月 2023 17:40:58 +0800 Subject: [PATCH] fix --- model/serviceFollowup.go | 305 ++++++++++++++++++++++++++------------------------ 1 files changed, 160 insertions(+), 145 deletions(-) diff --git a/model/serviceFollowup.go b/model/serviceFollowup.go index d2d4ada..917d2a5 100644 --- a/model/serviceFollowup.go +++ b/model/serviceFollowup.go @@ -1,145 +1,160 @@ -package model - -import ( - "aps_crm/constvar" - "aps_crm/pkg/mysqlx" - "gorm.io/gorm" -) - -type ( - ServiceFollowup struct { - Id int `json:"id" gorm:"column:id;primary_key;AUTO_INCREMENT"` - ClientId int `json:"clientId" gorm:"column:client_id;type:int;comment:瀹㈡埛id"` - Number string `json:"number" gorm:"column:number;type:varchar(255);comment:鍚堝悓缂栧彿"` - ContactId int `json:"contactId" gorm:"column:contact_id;type:int;comment:鑱旂郴浜篿d"` - ServiceId int `json:"serviceId" gorm:"column:service_id;type:int;comment:瀹㈡埛鏈嶅姟鍗昳d"` - ServiceOrder ServiceOrder `gorm:"foreignKey:ServiceId"` - CustomerServiceSheet CustomerServiceSheet `json:"customerServiceSheet" gorm:"foreignKey:ServiceId"` - MemberId int `json:"memberId" gorm:"column:member_id;type:int;comment:鏈嶅姟浜哄憳id"` - PlanId int `json:"planId" gorm:"column:plan_id;type:int;comment:鏈嶅姟璁″垝id"` - SatisfactionId int `json:"satisfactionId" gorm:"column:satisfaction_id;type:int;comment:婊℃剰搴d"` - TimelyRateId int `json:"timelyRateId" gorm:"column:timely_rate_id;type:int;comment:鍙婃椂鐜噄d"` - SolveRateId int `json:"solveRateId" gorm:"column:solve_rate_id;type:int;comment:瑙e喅鐜噄d"` - IsVisitId int `json:"isVisitId" gorm:"column:is_visit_id;type:int;comment:鏈嶅姟浜哄憳鏄惁鏉ヨ繃id"` - OldMemberId int `json:"oldMemberId" gorm:"column:old_member_id;type:int;comment:鍘熸湇鍔′汉鍛�"` - Remark string `json:"remark" gorm:"column:remark;type:text;comment:澶囨敞"` - File string `json:"file" gorm:"column:file;type:varchar(255);comment:闄勪欢"` - - gorm.Model `json:"-"` - } - - ServiceFollowupSearch struct { - ServiceFollowup - Orm *gorm.DB - KeywordType constvar.ServiceFollowupKeywordType - Keyword interface{} - OrderBy string - PageNum int - PageSize int - } -) - -func (ServiceFollowup) TableName() string { - return "service_followup" -} - -func NewServiceFollowupSearch() *ServiceFollowupSearch { - return &ServiceFollowupSearch{ - Orm: mysqlx.GetDB(), - } -} - -func (slf *ServiceFollowupSearch) build() *gorm.DB { - var db = slf.Orm.Model(&ServiceFollowup{}) - if slf.Id != 0 { - db = db.Where("id = ?", slf.Id) - } - - switch slf.KeywordType { - case constvar.ServiceFollowupKeywordFollowupNo: - db = db.Where("number = ?", slf.Keyword) - case constvar.ServiceFollowupKeywordCustomerName: - db = db.Where("client_id = ?", slf.Keyword) - case constvar.ServiceFollowupKeywordContactName: - db = db.Where("contact_id = ?", slf.Keyword) - case constvar.ServiceFollowupKeywordCustomerServiceNo: - db = db.Where("service_id = ?", slf.Keyword) - case constvar.ServiceFollowupKeywordVisitor: - db = db.Where("member_id = ?", slf.Keyword) - case constvar.ServiceFollowupKeywordSatisfactionDegree: - db = db.Where("satisfaction_id = ?", slf.Keyword) - - } - - return db -} - -func (slf *ServiceFollowupSearch) Create(record *ServiceFollowup) error { - var db = slf.build() - return db.Create(record).Error -} - -func (slf *ServiceFollowupSearch) Update(record *ServiceFollowup) error { - var db = slf.build() - return db.Updates(record).Error -} - -func (slf *ServiceFollowupSearch) Delete() error { - var db = slf.build() - return db.Delete(&ServiceFollowup{}).Error -} - -func (slf *ServiceFollowupSearch) Find() (*ServiceFollowup, error) { - var db = slf.build() - var record = &ServiceFollowup{} - err := db.First(record).Error - return record, err -} - -func (slf *ServiceFollowupSearch) FindAll() ([]*ServiceFollowup, int64, error) { - var db = slf.build() - var records = make([]*ServiceFollowup, 0) - var total int64 - if err := db.Count(&total).Error; err != nil { - return records, total, err - } - if slf.PageNum > 0 && slf.PageSize > 0 { - db = db.Limit(slf.PageSize).Offset((slf.PageNum - 1) * slf.PageSize) - } - - if slf.PageNum > 0 && slf.PageSize > 0 { - db = db.Limit(slf.PageSize).Offset((slf.PageNum - 1) * slf.PageSize) - } - - err := db.Preload("CustomerServiceSheet").Preload("ServiceOrder").Find(&records).Error - return records, total, err -} - -func (slf *ServiceFollowupSearch) SetId(id int) *ServiceFollowupSearch { - slf.Id = id - return slf -} - -func (slf *ServiceFollowupSearch) SetKeywordType(keyword constvar.ServiceFollowupKeywordType) *ServiceFollowupSearch { - slf.KeywordType = keyword - return slf -} - -func (slf *ServiceFollowupSearch) SetKeyword(keyword string) *ServiceFollowupSearch { - slf.Keyword = keyword - return slf -} - -func (slf *ServiceFollowupSearch) SetPage(page, size int) *ServiceFollowupSearch { - slf.PageNum, slf.PageSize = page, size - return slf -} - -func (slf *ServiceFollowupSearch) SetOrder(order string) *ServiceFollowupSearch { - slf.OrderBy = order - return slf -} -func (slf *ServiceFollowupSearch) SetIds(ids []int) *ServiceFollowupSearch { - slf.Orm = slf.Orm.Where("id in (?)", ids) - return slf -} +package model + +import ( + "aps_crm/constvar" + "aps_crm/pkg/mysqlx" + "gorm.io/gorm" +) + +type ( + ServiceFollowup struct { + Id int `json:"id" gorm:"column:id;primary_key;AUTO_INCREMENT"` + ClientId int `json:"clientId" gorm:"column:client_id;type:int;comment:瀹㈡埛id"` + Client Client `json:"client" gorm:"foreignKey:ClientId"` + Number string `json:"number" gorm:"column:number;type:varchar(255);comment:鍚堝悓缂栧彿"` + ContactId int `json:"contactId" gorm:"column:contact_id;type:int;comment:鑱旂郴浜篿d"` + Contact Contact `json:"contact" gorm:"foreignKey:ContactId"` + ServiceId int `json:"serviceId" gorm:"column:service_id;type:int;comment:瀹㈡埛鏈嶅姟鍗昳d"` + ServiceOrder ServiceOrder `gorm:"foreignKey:ServiceId"` + CustomerServiceSheet CustomerServiceSheet `json:"customerServiceSheet" gorm:"foreignKey:ServiceId"` + MemberId int `json:"memberId" gorm:"column:member_id;type:int;comment:鏈嶅姟浜哄憳id"` + PlanId int `json:"planId" gorm:"column:plan_id;type:int;comment:鏈嶅姟璁″垝id"` + SatisfactionId int `json:"satisfactionId" gorm:"column:satisfaction_id;type:int;comment:婊℃剰搴d"` + TimelyRateId int `json:"timelyRateId" gorm:"column:timely_rate_id;type:int;comment:鍙婃椂鐜噄d"` + SolveRateId int `json:"solveRateId" gorm:"column:solve_rate_id;type:int;comment:瑙e喅鐜噄d"` + IsVisitId int `json:"isVisitId" gorm:"column:is_visit_id;type:int;comment:鏈嶅姟浜哄憳鏄惁鏉ヨ繃id"` + OldMemberId int `json:"oldMemberId" gorm:"column:old_member_id;type:int;comment:鍘熸湇鍔′汉鍛�"` + Remark string `json:"remark" gorm:"column:remark;type:text;comment:澶囨敞"` + File string `json:"file" gorm:"column:file;type:varchar(255);comment:闄勪欢"` + + gorm.Model `json:"-"` + } + + ServiceFollowupSearch struct { + ServiceFollowup + Orm *gorm.DB + KeywordType constvar.ServiceFollowupKeywordType + Keyword interface{} + OrderBy string + PageNum int + PageSize int + Preload bool + } +) + +func (ServiceFollowup) TableName() string { + return "service_followup" +} + +func NewServiceFollowupSearch() *ServiceFollowupSearch { + return &ServiceFollowupSearch{ + Orm: mysqlx.GetDB(), + } +} + +func (slf *ServiceFollowupSearch) build() *gorm.DB { + var db = slf.Orm.Model(&ServiceFollowup{}) + if slf.Id != 0 { + db = db.Where("id = ?", slf.Id) + } + + switch slf.KeywordType { + case constvar.ServiceFollowupKeywordFollowupNo: + db = db.Where("number = ?", slf.Keyword) + case constvar.ServiceFollowupKeywordCustomerName: + db = db.Where("client_id = ?", slf.Keyword) + case constvar.ServiceFollowupKeywordContactName: + db = db.Where("contact_id = ?", slf.Keyword) + case constvar.ServiceFollowupKeywordCustomerServiceNo: + db = db.Where("service_id = ?", slf.Keyword) + case constvar.ServiceFollowupKeywordVisitor: + db = db.Where("member_id = ?", slf.Keyword) + case constvar.ServiceFollowupKeywordSatisfactionDegree: + db = db.Where("satisfaction_id = ?", slf.Keyword) + + } + + if slf.Preload { + db = db.Preload("CustomerServiceSheet"). + Preload("ServiceOrder"). + Preload("Client"). + Preload("Contact") + } + + return db +} + +func (slf *ServiceFollowupSearch) Create(record *ServiceFollowup) error { + var db = slf.build() + return db.Create(record).Error +} + +func (slf *ServiceFollowupSearch) Update(record *ServiceFollowup) error { + var db = slf.build() + return db.Updates(record).Error +} + +func (slf *ServiceFollowupSearch) Delete() error { + var db = slf.build() + return db.Delete(&ServiceFollowup{}).Error +} + +func (slf *ServiceFollowupSearch) Find() (*ServiceFollowup, error) { + var db = slf.build() + var record = &ServiceFollowup{} + err := db.First(record).Error + return record, err +} + +func (slf *ServiceFollowupSearch) FindAll() ([]*ServiceFollowup, int64, error) { + var db = slf.build() + var records = make([]*ServiceFollowup, 0) + var total int64 + if err := db.Count(&total).Error; err != nil { + return records, total, err + } + if slf.PageNum > 0 && slf.PageSize > 0 { + db = db.Limit(slf.PageSize).Offset((slf.PageNum - 1) * slf.PageSize) + } + + if slf.PageNum > 0 && slf.PageSize > 0 { + db = db.Limit(slf.PageSize).Offset((slf.PageNum - 1) * slf.PageSize) + } + + err := db.Find(&records).Error + return records, total, err +} + +func (slf *ServiceFollowupSearch) SetId(id int) *ServiceFollowupSearch { + slf.Id = id + return slf +} + +func (slf *ServiceFollowupSearch) SetPreload(preload bool) *ServiceFollowupSearch { + slf.Preload = preload + return slf +} + +func (slf *ServiceFollowupSearch) SetKeywordType(keyword constvar.ServiceFollowupKeywordType) *ServiceFollowupSearch { + slf.KeywordType = keyword + return slf +} + +func (slf *ServiceFollowupSearch) SetKeyword(keyword string) *ServiceFollowupSearch { + slf.Keyword = keyword + return slf +} + +func (slf *ServiceFollowupSearch) SetPage(page, size int) *ServiceFollowupSearch { + slf.PageNum, slf.PageSize = page, size + return slf +} + +func (slf *ServiceFollowupSearch) SetOrder(order string) *ServiceFollowupSearch { + slf.OrderBy = order + return slf +} +func (slf *ServiceFollowupSearch) SetIds(ids []int) *ServiceFollowupSearch { + slf.Orm = slf.Orm.Where("id in (?)", ids) + return slf +} -- Gitblit v1.8.0