From dac214fa72dc2974954a4d8ee934695f548ad155 Mon Sep 17 00:00:00 2001 From: wangpengfei <274878379@qq.com> Date: 星期三, 02 八月 2023 14:14:30 +0800 Subject: [PATCH] fix --- model/salesLeads.go | 48 ++++++++++++++++++++++++++++++++++++++++++++---- 1 files changed, 44 insertions(+), 4 deletions(-) diff --git a/model/salesLeads.go b/model/salesLeads.go index 8ff718f..796e15e 100644 --- a/model/salesLeads.go +++ b/model/salesLeads.go @@ -25,7 +25,12 @@ SalesLeadsSearch struct { SalesLeads - Orm *gorm.DB + + Orm *gorm.DB + Keyword string + OrderBy string + PageNum int + PageSize int } ) @@ -41,6 +46,12 @@ func (slf *SalesLeadsSearch) build() *gorm.DB { var db = slf.Orm.Model(&SalesLeads{}) + if slf.Keyword != "" { + db = db.Where("name LIKE ?", "%"+slf.Keyword+"%") + } + if slf.Keyword != "" { + db = db.Where("name LIKE ?", "%"+slf.Keyword+"%") + } if slf.Id != 0 { db = db.Where("id = ?", slf.Id) } @@ -102,10 +113,19 @@ return } -func (slf *SalesLeadsSearch) FindAll() (result []*SalesLeads, err error) { +func (slf *SalesLeadsSearch) FindAll() ([]*SalesLeads, int64, error) { var db = slf.build() - err = db.Preload("FollowRecord").Find(&result).Error - return + var records = make([]*SalesLeads, 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) + } + + err := db.Preload("FollowRecord").Find(&records).Error + return records, total, err } func (slf *SalesLeadsSearch) SetId(id int) *SalesLeadsSearch { @@ -123,3 +143,23 @@ err = db.First(&result).Error return } + +func (slf *SalesLeadsSearch) SetKeyword(keyword string) *SalesLeadsSearch { + slf.Keyword = keyword + return slf +} + +func (slf *SalesLeadsSearch) SetPage(page, size int) *SalesLeadsSearch { + slf.PageNum, slf.PageSize = page, size + return slf +} + +func (slf *SalesLeadsSearch) SetOrder(order string) *SalesLeadsSearch { + slf.OrderBy = order + return slf +} + +func (slf *SalesLeadsSearch) UpdateMap(data map[string]interface{}) error { + var db = slf.build() + return db.Updates(data).Error +} -- Gitblit v1.8.0