From 1df0424033208701c7d1f40c095eaf1b60dafe32 Mon Sep 17 00:00:00 2001 From: zhangqian <zhangqian@123.com> Date: 星期四, 07 九月 2023 17:26:00 +0800 Subject: [PATCH] fix --- model/procedures.go | 14 ++++++++++++-- model/work_order.go | 12 +++++++++++- 2 files changed, 23 insertions(+), 3 deletions(-) diff --git a/model/procedures.go b/model/procedures.go index 9cc75f2..6a4762e 100644 --- a/model/procedures.go +++ b/model/procedures.go @@ -85,6 +85,12 @@ slf.WorkOrderID = orderId return slf } + +func (slf *ProceduresSearch) SetProcedureId(id string) *ProceduresSearch { + slf.ProcedureID = id + return slf +} + func (slf *ProceduresSearch) SetStartTimeMax(ts int64) *ProceduresSearch { slf.StartTimeMax = ts return slf @@ -159,6 +165,10 @@ db = db.Where("procedure_id in ?", slf.ProcedureIds) } + if slf.ProcedureID != "" { + db = db.Where("procedure_id = ?", slf.ProcedureID) + } + return db } @@ -184,10 +194,10 @@ return nil } -func (slf *ProceduresSearch) Save(record *Procedures) error { +func (slf *ProceduresSearch) Updates(record *Procedures) error { var db = slf.build() - if err := db.Save(record).Error; err != nil { + if err := db.Updates(record).Error; err != nil { return fmt.Errorf("save err: %v, record: %+v", err, record) } diff --git a/model/work_order.go b/model/work_order.go index c6567b3..347d0ea 100644 --- a/model/work_order.go +++ b/model/work_order.go @@ -166,7 +166,17 @@ return nil } -func (slf *OrderSearch) Save(record *Order) error { +func (slf *OrderSearch) Updates(record *Order) error { + var db = slf.build() + + if err := db.Updates(record).Error; err != nil { + return fmt.Errorf("save err: %v, record: %+v", err, record) + } + + return nil +} + +func (slf *OrderSearch) Upsert(record *Order) error { var db = slf.build() if err := db.Save(record).Error; err != nil { -- Gitblit v1.8.0