From 8dfb8feb32bb5e4e460e23dcde42612a26fa2bcb Mon Sep 17 00:00:00 2001
From: zhangqian <zhangqian@123.com>
Date: 星期日, 27 八月 2023 01:11:28 +0800
Subject: [PATCH] fix

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

diff --git a/model/procedures.go b/model/procedures.go
index 1dfc611..48534b3 100644
--- a/model/procedures.go
+++ b/model/procedures.go
@@ -1,39 +1,59 @@
 package model
 
 import (
+	"apsClient/model/common"
+	"apsClient/pkg/logx"
 	"apsClient/pkg/sqlitex"
+	"encoding/json"
 	"fmt"
 	"gorm.io/gorm"
 )
 
 type (
 	Procedures struct {
-		gorm.Model        `json:"-"`
-		Id                int        `json:"id"`
-		ProcedureId       string     `json:"orderId"`
-		ProcedureName     string     `json:"product"`
-		DeviceId          string     `json:"deviceId"`
-		StartTime         string     `json:"startTime"`
-		EndTime           string     `json:"endTime"`
-		WorkHours         string     `json:"workHours"`
-		InputMaterialsId  int        `json:"-"`
-		InputMaterials    *Materials `json:"inputMaterials" gorm:"foreignKey:InputMaterialsId"`
-		OutputMaterialsId int        `json:"-"`
-		OutputMaterials   *Materials `json:"outputMaterials" gorm:"foreignKey:InputMaterialsId"`
+		gorm.Model     `json:"-"`
+		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:"-"`
+		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 {
 		Procedures
-		Order    string
-		PageNum  int
-		PageSize int
-		Orm      *gorm.DB
-		Preload  bool
+		Order        string
+		PageNum      int
+		PageSize     int
+		Orm          *gorm.DB
+		Preload      bool
+		StartTimeMax int64
+		StatusNot    ProcedureStatus
 	}
+)
+
+type ProcedureStatus int
+
+const (
+	ProcedureStatusWaitProcess ProcedureStatus = 1
+	ProcedureStatusProcessing  ProcedureStatus = 2
+	ProcedureStatusFinished    ProcedureStatus = 3
 )
 
 func (slf *Procedures) TableName() string {
 	return "procedures"
+}
+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 err
+	}
+	slf.ProceduresInfo = proceduresInfo
+	return nil
 }
 
 func NewProceduresSearch(db *gorm.DB) *ProceduresSearch {
@@ -58,6 +78,30 @@
 	return slf
 }
 
+func (slf *ProceduresSearch) SetWorkOrderId(orderId string) *ProceduresSearch {
+	slf.WorkOrderID = orderId
+	return slf
+}
+func (slf *ProceduresSearch) SetStartTimeMax(ts int64) *ProceduresSearch {
+	slf.StartTimeMax = ts
+	return slf
+}
+
+func (slf *ProceduresSearch) SetId(id int) *ProceduresSearch {
+	slf.ID = 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
+}
+
 func (slf *ProceduresSearch) SetPreload(preload bool) *ProceduresSearch {
 	slf.Preload = preload
 	return slf
@@ -69,10 +113,31 @@
 	if slf.Order != "" {
 		db = db.Order(slf.Order)
 	}
+
+	if slf.ID != 0 {
+		db = db.Where("id = ?", slf.ID)
+	}
+
+	if slf.WorkOrderID != "" {
+		db = db.Where("work_order_id = ?", slf.WorkOrderID)
+	}
+
 	if slf.Preload {
 		db = db.Preload("InputMaterials").Preload("OutputMaterials")
 	}
 
+	if slf.StartTimeMax != 0 {
+		db = db.Where("start_time <= ?", slf.StartTimeMax)
+	}
+
+	if slf.Status != 0 {
+		db = db.Where("status = ?", slf.Status)
+	}
+
+	if slf.StatusNot != 0 {
+		db = db.Where("status != ?", slf.StatusNot)
+	}
+
 	return db
 }
 

--
Gitblit v1.8.0