From ef7d02ce56e98e8cbc5ed6ec746ea39bf0b8c9e2 Mon Sep 17 00:00:00 2001
From: zhangqian <zhangqian@123.com>
Date: 星期六, 16 九月 2023 22:44:58 +0800
Subject: [PATCH] 增加实时获取生产进度接口

---
 model/procedures.go |   62 +++++++++++++++++++++++++++++--
 1 files changed, 58 insertions(+), 4 deletions(-)

diff --git a/model/procedures.go b/model/procedures.go
index a1fb11b..99e45f4 100644
--- a/model/procedures.go
+++ b/model/procedures.go
@@ -15,6 +15,9 @@
 		ID             int    `gorm:"primarykey"`
 		WorkOrderID    string `gorm:"index;type:varchar(191);not null;comment:宸ュ崟ID" json:"-"`
 		OrderID        string `gorm:"index;type:varchar(191);not null;comment:璁㈠崟ID" json:"-"`
+		DeviceID       string `gorm:"index;type:varchar(191);comment:璁惧ID" json:"deviceId"`
+		ProcedureID    string `gorm:"index;type:varchar(191);comment:宸ュ簭ID" json:"procedureId"`
+		Position       int    `gorm:"type:int;comment:宸ヤ綔浣嶇疆" json:"position"` //姣忎釜璁惧鍙兘鏈夊涓満浣嶅悓鏃剁敓浜э紝鐢╬osition琛ㄧず浣嶇疆
 		StartTime      int64  `gorm:"comment:璁″垝寮�濮嬫椂闂�" json:"startTime"`
 		EndTime        int64  `gorm:"comment:璁″垝缁撴潫鏃堕棿" json:"endTime"`
 		Status         ProcedureStatus
@@ -30,14 +33,17 @@
 		Orm          *gorm.DB
 		Preload      bool
 		StartTimeMax int64
+		StatusNot    ProcedureStatus
+		ProcedureIds []string
 	}
 )
 
 type ProcedureStatus int
 
 const (
-	ProcedureStatusUnFinished ProcedureStatus = 1
-	ProcedureStatusFinished   ProcedureStatus = 2
+	ProcedureStatusWaitProcess ProcedureStatus = 1
+	ProcedureStatusProcessing  ProcedureStatus = 2
+	ProcedureStatusFinished    ProcedureStatus = 3
 )
 
 func (slf *Procedures) TableName() string {
@@ -80,6 +86,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
@@ -90,8 +102,23 @@
 	return slf
 }
 
+func (slf *ProceduresSearch) SetProcedureIds(procedureIds []string) *ProceduresSearch {
+	slf.ProcedureIds = procedureIds
+	return slf
+}
+
+func (slf *ProceduresSearch) SetDeviceId(id string) *ProceduresSearch {
+	slf.DeviceID = id
+	return slf
+}
+
 func (slf *ProceduresSearch) SetStatus(status ProcedureStatus) *ProceduresSearch {
 	slf.Status = status
+	return slf
+}
+
+func (slf *ProceduresSearch) SetStatusNot(status ProcedureStatus) *ProceduresSearch {
+	slf.StatusNot = status
 	return slf
 }
 
@@ -115,6 +142,10 @@
 		db = db.Where("work_order_id = ?", slf.WorkOrderID)
 	}
 
+	if slf.DeviceID != "" {
+		db = db.Where("device_id = ?", slf.DeviceID)
+	}
+
 	if slf.Preload {
 		db = db.Preload("InputMaterials").Preload("OutputMaterials")
 	}
@@ -125,6 +156,18 @@
 
 	if slf.Status != 0 {
 		db = db.Where("status = ?", slf.Status)
+	}
+
+	if slf.StatusNot != 0 {
+		db = db.Where("status <> ?", slf.StatusNot)
+	}
+
+	if len(slf.ProcedureIds) > 0 {
+		db = db.Where("procedure_id in ?", slf.ProcedureIds)
+	}
+
+	if slf.ProcedureID != "" {
+		db = db.Where("procedure_id = ?", slf.ProcedureID)
 	}
 
 	return db
@@ -152,10 +195,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)
 	}
 
@@ -229,6 +272,17 @@
 	return records, total, nil
 }
 
+func (slf *ProceduresSearch) Count() (int64, error) {
+	var (
+		total int64
+		db    = slf.build()
+	)
+	if err := db.Count(&total).Error; err != nil {
+		return total, fmt.Errorf("find count err: %v", err)
+	}
+	return total, nil
+}
+
 func (slf *ProceduresSearch) FindNotTotal() ([]*Procedures, error) {
 	var (
 		records = make([]*Procedures, 0)

--
Gitblit v1.8.0