From cab6eea89a0d82710369604ecb51f1cdb122e433 Mon Sep 17 00:00:00 2001 From: zhangqian <zhangqian@123.com> Date: 星期四, 19 十月 2023 17:26:34 +0800 Subject: [PATCH] fix --- model/procedures.go | 48 ++++++++++++++++++++++++++++++++---------------- 1 files changed, 32 insertions(+), 16 deletions(-) diff --git a/model/procedures.go b/model/procedures.go index cb8d660..4c93059 100644 --- a/model/procedures.go +++ b/model/procedures.go @@ -6,23 +6,23 @@ "apsClient/pkg/sqlitex" "encoding/json" "fmt" - "gorm.io/gorm" + "github.com/jinzhu/gorm" ) 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:"-"` - DeviceID string `gorm:"index;type:varchar(191);comment:璁惧ID" json:"deviceId"` - ProcedureID string `gorm:"index;type:varchar(191);comment:宸ュ簭ID" json:"procedureId"` - Channel int32 `gorm:"index;comment:閫氶亾" json:"channel"` //閫氶亾 - 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 瀵硅薄 + gorm.Model + WorkOrderID string `gorm:"index;type:varchar(191);not null" json:"-"` + OrderID string `gorm:"index;type:varchar(191);not null" json:"-"` + DeviceID string `gorm:"index;type:varchar(191)" json:"deviceId"` + ProcedureID string `gorm:"index;type:varchar(191)" json:"procedureId"` + Channel int32 `gorm:"index;" json:"channel"` //閫氶亾 + ProcessModelNumber string `gorm:"index;" json:"processModelNumber"` //宸ヨ壓妯″瀷缂栧彿 + StartTime int64 `json:"startTime"` + EndTime int64 `json:"endTime"` + Status ProcedureStatus + ProcedureData string `json:"-"` //common.ProductProcedure json涓� + ProceduresInfo common.ProductProcedure `json:"procedure" gorm:"-"` //common.ProductProcedure 瀵硅薄 } ProceduresSearch struct { @@ -104,7 +104,7 @@ return slf } -func (slf *ProceduresSearch) SetId(id int) *ProceduresSearch { +func (slf *ProceduresSearch) SetId(id uint) *ProceduresSearch { slf.ID = id return slf } @@ -179,7 +179,7 @@ } if len(slf.ProcedureIds) > 0 { - db = db.Where("procedure_id in ?", slf.ProcedureIds) + db = db.Where("procedure_id IN (?)", slf.ProcedureIds) } if slf.ProcedureID != "" { @@ -187,7 +187,7 @@ } if len(slf.Channels) > 0 { - db = db.Where("channel in ?", slf.Channels) + db = db.Where("channel IN (?)", slf.Channels) } return db @@ -215,6 +215,22 @@ return nil } +func (slf *ProceduresSearch) Upsert(record *Procedures) error { + var db = slf.build() + old, err := slf.First() + if err != gorm.ErrRecordNotFound && old.ID != 0 { + record.ID = old.ID + err = db.Save(&record).Error + } else { + err = db.Create(&record).Error + } + if err != nil { + return fmt.Errorf("save err: %v, record: %+v", err, record) + } + + return nil +} + func (slf *ProceduresSearch) Updates(record *Procedures) error { var db = slf.build() -- Gitblit v1.8.0