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