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 | 42 ++++++++++++++++++++++++++++++------------ 1 files changed, 30 insertions(+), 12 deletions(-) diff --git a/model/procedures.go b/model/procedures.go index 4915825..48534b3 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,12 +12,14 @@ type ( Procedures struct { 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:"-"` - 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:"-"` + 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 { @@ -28,27 +30,30 @@ Orm *gorm.DB Preload bool StartTimeMax int64 + StatusNot ProcedureStatus } ) 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 { @@ -92,6 +97,11 @@ 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 @@ -102,6 +112,10 @@ if slf.Order != "" { db = db.Order(slf.Order) + } + + if slf.ID != 0 { + db = db.Where("id = ?", slf.ID) } if slf.WorkOrderID != "" { @@ -120,6 +134,10 @@ db = db.Where("status = ?", slf.Status) } + if slf.StatusNot != 0 { + db = db.Where("status != ?", slf.StatusNot) + } + return db } -- Gitblit v1.8.0