zhangqian
2023-10-09 dad1ae79fe78e770d88f41713b4c68fad562d9ef
model/process_model.go
@@ -1,34 +1,51 @@
package model
import (
   "apsClient/pkg/logx"
   "apsClient/pkg/sqlitex"
   "encoding/json"
   "fmt"
   "gorm.io/gorm"
)
type (
   // ProcessModel 工艺流程参数
   // ProcessModel 工艺参数
   ProcessModel struct {
      gorm.Model `json:"-"`
      Number     string                 `gorm:"index;column:number;type:varchar(255);not null;default '';comment:工艺模型编号" json:"number"` //工艺模型编号
      Product    string                 `gorm:"column:product;type:varchar(255);not null;default '';comment:产品名称" json:"product"`       //产品名称
      Procedure  string                 `gorm:"column:procedure;type:varchar(255);not null;default '';comment:工序" json:"procedure"`     //工序
      Params     string                 `gorm:"type:text;comment:工艺参数键值对json串"`
      ParamsMap  map[string]interface{} `json:"paramsMap" gorm:"-"`
      DeviceId   string                 `json:"deviceId" gorm:"-"` //用于过滤获取nsq消息
      gorm.Model   `json:"-"`
      Number       string                 `gorm:"index;column:number;type:varchar(255);not null;default '';comment:工艺模型编号" json:"number"` //工艺模型编号
      Product      string                 `gorm:"column:product;type:varchar(255);not null;default '';comment:产品名称" json:"product"`       //产品名称
      Procedure    string                 `gorm:"column:procedure;type:varchar(255);not null;default '';comment:工序" json:"procedure"`     //工序
      Params       string                 `gorm:"type:text;comment:工艺参数键值对json串"`
      ParamsMap    map[string]interface{} `json:"paramsMap" gorm:"-"`
      DeviceId     string                 `json:"deviceId" gorm:"-"`                    //用于过滤获取nsq消息
      IsNew        bool                   `json:"-" gorm:"column:is_new;comment:是否最新的"` //是否最新的
      IsUpdate     bool                   `json:"isUpdate" gorm:"-"`                    //前端用
      NewParamsMap map[string]interface{} `json:"newParamsMap" gorm:"-"`
      NewNumber    string                 `json:"newNumber" gorm:"-"`
   }
   ProcessModelSearch struct {
      ProcessModel
      Order    string
      PageNum  int
      PageSize int
      Orm      *gorm.DB
      Order         string
      Limit         int
      Offset        int
      Orm           *gorm.DB
      Procedures    []string
      CurrentNumber string
   }
)
func (slf *ProcessModel) TableName() string {
   return "process_model"
}
func (slf *ProcessModel) AfterFind(db *gorm.DB) error {
   err := json.Unmarshal([]byte(slf.Params), &slf.ParamsMap)
   if err != nil {
      logx.Errorf("process model json.Unmarshal:%v", err)
      return err
   }
   return nil
}
func NewProcessModelSearch() *ProcessModelSearch {
@@ -40,8 +57,8 @@
   return slf
}
func (slf *ProcessModelSearch) SetPage(page, size int) *ProcessModelSearch {
   slf.PageNum, slf.PageSize = page, size
func (slf *ProcessModelSearch) SetPage(offset, limit int) *ProcessModelSearch {
   slf.Offset, slf.Limit = offset, limit
   return slf
}
@@ -60,12 +77,28 @@
   return slf
}
func (slf *ProcessModelSearch) SetCurrentNumber(number string) *ProcessModelSearch {
   slf.CurrentNumber = number
   return slf
}
func (slf *ProcessModelSearch) SetProduct(product string) *ProcessModelSearch {
   slf.Product = product
   return slf
}
func (slf *ProcessModelSearch) SetProcedure(procedure string) *ProcessModelSearch {
   slf.Procedure = procedure
   return slf
}
func (slf *ProcessModelSearch) SetProcedures(procedures []string) *ProcessModelSearch {
   slf.Procedures = procedures
   return slf
}
func (slf *ProcessModelSearch) SetIsNew(isNew bool) *ProcessModelSearch {
   slf.IsNew = isNew
   return slf
}
@@ -84,12 +117,24 @@
      db = db.Where("`procedure` = ?", slf.Procedure)
   }
   if len(slf.Procedures) != 0 {
      db = db.Where("`procedure` in ?", slf.Procedures)
   }
   if len(slf.Number) != 0 {
      db = db.Where("number = ?", slf.Number)
   }
   if slf.Order != "" {
      db = db.Order(slf.Order)
   }
   if slf.IsNew {
      db = db.Where("is_new = ?", 1)
   }
   if len(slf.CurrentNumber) != 0 {
      db = db.Where("number != ?", slf.CurrentNumber)
   }
   return db
@@ -173,8 +218,8 @@
   if err := db.Count(&total).Error; err != nil {
      return records, total, fmt.Errorf("find count err: %v", err)
   }
   if slf.PageNum*slf.PageSize > 0 {
      db = db.Offset((slf.PageNum - 1) * slf.PageSize).Limit(slf.PageSize)
   if slf.Offset > 0 || slf.Limit > 0 {
      db = db.Offset(slf.Offset).Limit(slf.Limit)
   }
   if err := db.Find(&records).Error; err != nil {
      return records, total, fmt.Errorf("find records err: %v", err)
@@ -189,8 +234,8 @@
      db      = slf.build()
   )
   if slf.PageNum*slf.PageSize > 0 {
      db = db.Offset((slf.PageNum - 1) * slf.PageSize).Limit(slf.PageSize)
   if slf.Offset > 0 || slf.Limit > 0 {
      db = db.Offset(slf.Offset).Limit(slf.Limit)
   }
   if err := db.Find(&records).Error; err != nil {
      return records, fmt.Errorf("find records err: %v", err)
@@ -210,8 +255,8 @@
   if err := db.Count(&total).Error; err != nil {
      return records, total, fmt.Errorf("find by query count err: %v", err)
   }
   if slf.PageNum*slf.PageSize > 0 {
      db = db.Offset((slf.PageNum - 1) * slf.PageSize).Limit(slf.PageSize)
   if slf.Offset > 0 || slf.Limit > 0 {
      db = db.Offset(slf.Offset).Limit(slf.Limit)
   }
   if err := db.Find(&records).Error; err != nil {
      return records, total, fmt.Errorf("find by query records err: %v, query: %s, args: %+v", err, query, args)
@@ -227,8 +272,8 @@
      db      = slf.Orm.Table(slf.TableName()).Where(query, args...)
   )
   if slf.PageNum*slf.PageSize > 0 {
      db = db.Offset((slf.PageNum - 1) * slf.PageSize).Limit(slf.PageSize)
   if slf.Offset > 0 || slf.Limit > 0 {
      db = db.Offset(slf.Offset).Limit(slf.Limit)
   }
   if err := db.Find(&records).Error; err != nil {
      return records, fmt.Errorf("find by query records err: %v, query: %s, args: %+v", err, query, args)