From ef7d02ce56e98e8cbc5ed6ec746ea39bf0b8c9e2 Mon Sep 17 00:00:00 2001 From: zhangqian <zhangqian@123.com> Date: 星期六, 16 九月 2023 22:44:58 +0800 Subject: [PATCH] 增加实时获取生产进度接口 --- model/work_order.go | 24 ++++++++++++++++++++---- 1 files changed, 20 insertions(+), 4 deletions(-) diff --git a/model/work_order.go b/model/work_order.go index 0626042..076edef 100644 --- a/model/work_order.go +++ b/model/work_order.go @@ -23,7 +23,6 @@ StartTime int64 `gorm:"comment:璁″垝寮�濮嬫椂闂�" json:"startTime"` EndTime int64 `gorm:"comment:璁″垝缁撴潫鏃堕棿" json:"endTime"` Status OrderStatus - StatusList []OrderStatus } OrderSearch struct { @@ -35,6 +34,8 @@ Preload bool StartTimeMax int64 StartTimeMin int64 + StatusList []OrderStatus + WorkOrderIDs []string } ) @@ -47,7 +48,7 @@ ) func (slf *Order) TableName() string { - return "Order" + return "work_order" } func NewOrderSearch(db *gorm.DB) *OrderSearch { @@ -101,6 +102,11 @@ return slf } +func (slf *OrderSearch) SetWorkOrderIds(workOrderIds []string) *OrderSearch { + slf.WorkOrderIDs = workOrderIds + return slf +} + func (slf *OrderSearch) SetId(id uint) *OrderSearch { slf.ID = id return slf @@ -126,7 +132,7 @@ } if len(slf.StatusList) != 0 { - db = db.Where("status != ?", slf.StatusList) + db = db.Where("status in ?", slf.StatusList) } if slf.StartTimeMax != 0 { @@ -166,7 +172,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