From b555874608be2e4bb40cceba60b497c04caef842 Mon Sep 17 00:00:00 2001
From: zhangqian <zhangqian@123.com>
Date: 星期二, 12 九月 2023 14:17:51 +0800
Subject: [PATCH] 三菱协议更改

---
 model/procedures.go |   84 +++++++++++++++++++++++++++++++++--------
 1 files changed, 67 insertions(+), 17 deletions(-)

diff --git a/model/procedures.go b/model/procedures.go
index 5ac3185..6a4762e 100644
--- a/model/procedures.go
+++ b/model/procedures.go
@@ -1,7 +1,7 @@
 package model
 
 import (
-	"apsClient/model/request"
+	"apsClient/model/common"
 	"apsClient/pkg/logx"
 	"apsClient/pkg/sqlitex"
 	"encoding/json"
@@ -12,11 +12,16 @@
 type (
 	Procedures struct {
 		gorm.Model     `json:"-"`
-		ID             int                      `gorm:"primarykey"`
-		OrderID        string                   `gorm:"index;type:varchar(191);not null;comment:璁㈠崟ID" json:"-"`
-		Status         ProcedureStatus          `json:"-"`
-		ProcedureData  string                   `json:"-"`                  //request.ProductProcedure  json涓�
-		ProceduresInfo request.ProductProcedure `json:"procedure" gorm:"-"` //request.ProductProcedure  瀵硅薄
+		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"`
+		StartTime      int64  `gorm:"comment:璁″垝寮�濮嬫椂闂�" json:"startTime"`
+		EndTime        int64  `gorm:"comment:璁″垝缁撴潫鏃堕棿" json:"endTime"`
+		Status         ProcedureStatus
+		ProcedureData  string                  `json:"-"`                  //common.ProductProcedure  json涓�
+		ProceduresInfo common.ProductProcedure `json:"procedure" gorm:"-"` //common.ProductProcedure  瀵硅薄
 	}
 
 	ProceduresSearch struct {
@@ -27,27 +32,31 @@
 		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 {
 	return "procedures"
 }
-func (slf *Procedures) AfterFind() {
-	var proceduresInfo request.ProductProcedure
+func (slf *Procedures) AfterFind(db *gorm.DB) error {
+	var proceduresInfo common.ProductProcedure
 	err := json.Unmarshal([]byte(slf.ProcedureData), &proceduresInfo)
 	if err != nil {
 		logx.Errorf("AfterFind Unmarshal err: %v", err.Error())
-		return
+		return err
 	}
 	slf.ProceduresInfo = proceduresInfo
+	return nil
 }
 
 func NewProceduresSearch(db *gorm.DB) *ProceduresSearch {
@@ -72,10 +81,16 @@
 	return slf
 }
 
-func (slf *ProceduresSearch) SetOrderId(orderId string) *ProceduresSearch {
-	slf.OrderID = orderId
+func (slf *ProceduresSearch) SetWorkOrderId(orderId string) *ProceduresSearch {
+	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
@@ -86,8 +101,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
 }
 
@@ -103,8 +133,16 @@
 		db = db.Order(slf.Order)
 	}
 
-	if slf.OrderID != "" {
-		db = db.Where("order_id = ?", slf.OrderID)
+	if slf.ID != 0 {
+		db = db.Where("id = ?", slf.ID)
+	}
+
+	if slf.WorkOrderID != "" {
+		db = db.Where("work_order_id = ?", slf.WorkOrderID)
+	}
+
+	if slf.DeviceID != "" {
+		db = db.Where("device_id = ?", slf.DeviceID)
 	}
 
 	if slf.Preload {
@@ -117,6 +155,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
@@ -144,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)
 	}
 

--
Gitblit v1.8.0