From 178d6e7507ef4806ca02692c4ee24bb98dba8362 Mon Sep 17 00:00:00 2001 From: zhangqian <zhangqian@123.com> Date: 星期二, 19 三月 2024 19:19:46 +0800 Subject: [PATCH] 生丝检验部分逻辑修正 --- models/workshop_manage.go | 31 +++++++++++++++++++++++++++++++ 1 files changed, 31 insertions(+), 0 deletions(-) diff --git a/models/workshop_manage.go b/models/workshop_manage.go index 6d920cb..dc5166b 100644 --- a/models/workshop_manage.go +++ b/models/workshop_manage.go @@ -49,11 +49,29 @@ return slf } +func (slf *WorkshopManageSearch) SetWorkshopNumber(workshopNumber string) *WorkshopManageSearch { + slf.WorkshopNumber = workshopNumber + return slf +} + +func (slf *WorkshopManageSearch) SetGroupNumber(groupNumber int) *WorkshopManageSearch { + slf.GroupNumber = groupNumber + return slf +} + func (slf *WorkshopManageSearch) build() *gorm.DB { db := slf.Orm.Table(slf.TableName()) if slf.ID > 0 { db = db.Where("id = ?", slf.ID) + } + + if slf.WorkshopNumber != "" { + db = db.Where("workshop_number = ?", slf.WorkshopNumber) + } + + if slf.GroupNumber != 0 { + db = db.Where("group_number = ?", slf.GroupNumber) } return db @@ -89,6 +107,19 @@ return records, total, nil } +func (slf *WorkshopManageSearch) FindAll() ([]*WorkshopManage, error) { + var ( + records = make([]*WorkshopManage, 0) + db = slf.build() + ) + + if err := db.Find(&records).Error; err != nil { + return records, fmt.Errorf("find records err: %v", err) + } + + return records, nil +} + func (slf *WorkshopManageSearch) Save(record *WorkshopManage) error { var db = slf.build() -- Gitblit v1.8.0