From b4400a06b5f801bc7a85320680d0b6c45c547ff0 Mon Sep 17 00:00:00 2001 From: wangpengfei <274878379@qq.com> Date: 星期五, 28 七月 2023 10:21:33 +0800 Subject: [PATCH] fix --- model/salesReturn.go | 41 ++++++++++++++++++++++++++++++++++++----- 1 files changed, 36 insertions(+), 5 deletions(-) diff --git a/model/salesReturn.go b/model/salesReturn.go index 074cbe1..94e1232 100644 --- a/model/salesReturn.go +++ b/model/salesReturn.go @@ -21,7 +21,12 @@ SalesReturnSearch struct { SalesReturn - Orm *gorm.DB + + Orm *gorm.DB + Keyword string + OrderBy string + PageNum int + PageSize int } ) @@ -37,6 +42,9 @@ func (slf *SalesReturnSearch) build() *gorm.DB { var db = slf.Orm.Model(&SalesReturn{}) + if slf.Keyword != "" { + db = db.Where("name LIKE ?", "%"+slf.Keyword+"%") + } if slf.Id != 0 { db = db.Where("id = ?", slf.Id) } @@ -66,14 +74,37 @@ return record, err } -func (slf *SalesReturnSearch) FindAll() ([]*SalesReturn, error) { +func (slf *SalesReturnSearch) FindAll() ([]*SalesReturn, int64, error) { var db = slf.build() - var record = make([]*SalesReturn, 0) - err := db.Preload("Products").Find(&record).Error - return record, err + var records = make([]*SalesReturn, 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("Products").Find(&records).Error + return records, total, err } func (slf *SalesReturnSearch) SetId(id int) *SalesReturnSearch { slf.Id = id return slf } + +func (slf *SalesReturnSearch) SetKeyword(keyword string) *SalesReturnSearch { + slf.Keyword = keyword + return slf +} + +func (slf *SalesReturnSearch) SetPage(page, size int) *SalesReturnSearch { + slf.PageNum, slf.PageSize = page, size + return slf +} + +func (slf *SalesReturnSearch) SetOrder(order string) *SalesReturnSearch { + slf.OrderBy = order + return slf +} \ No newline at end of file -- Gitblit v1.8.0