zhangqian
2023-12-08 8e941bd782f574597f19c980982afade4dc11748
model/process_model.go
@@ -1,36 +1,54 @@
package model
import (
   "apsClient/pkg/logx"
   "apsClient/pkg/sqlitex"
   "encoding/json"
   "fmt"
   "gorm.io/gorm"
   "github.com/jinzhu/gorm"
)
type (
   // ProcessModel 工艺流程参数
   // ProcessModel 工艺参数
   ProcessModel struct {
      gorm.Model `json:"-"`
      Number     string                 `gorm:"index;column:number;type:varchar(255);not null;default '';comment:工艺模型编号" json:"number"` //工艺模型编号
      OrderId    string                 `gorm:"column:order_id;type:varchar(255);not null;default '';comment:订单id" json:"orderId"`      //订单id
      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"`     //工序
      WorkOrder  string                 `gorm:"column:work_order;type:varchar(255);not null;default '';comment:工单" json:"workOrder"`    //工单
      Device     string                 `gorm:"column:device;type:varchar(255);not null;default '';comment:设备" json:"device"`           //设备
      Params     string                 `gorm:"type:text;comment:工艺参数键值对json串"`
      ParamsMap  map[string]interface{} `json:"paramsMap" gorm:"-"`
      CommonModel
      Number       string `gorm:"index;column:number;type:varchar(255);not null;default ''" json:"number"` //工艺模型编号
      Product      string `gorm:"column:product;type:varchar(255);not null;default ''" json:"product"`     //产品名称
      Procedure    string `gorm:"column:procedure;type:varchar(255);not null;default ''" json:"procedure"` //工序
      Params       string
      ParamsMap    map[string]interface{} `json:"paramsMap" gorm:"-"`
      DeviceId     string                 `json:"deviceId" gorm:"-"`      //用于过滤获取nsq消息
      IsNew        bool                   `json:"-" gorm:"column:is_new"` //是否最新的
      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
      Numbers       []string
      ProductNot    string
      ProcedureNot  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 {
@@ -42,8 +60,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
}
@@ -62,24 +80,38 @@
   return slf
}
func (slf *ProcessModelSearch) SetWorkOrder(workOrder string) *ProcessModelSearch {
   slf.WorkOrder = workOrder
func (slf *ProcessModelSearch) SetNumbers(numbers []string) *ProcessModelSearch {
   slf.Numbers = numbers
   return slf
}
func (slf *ProcessModelSearch) SetOrderId(orderId string) *ProcessModelSearch {
   slf.OrderId = orderId
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) SetDevice(device string) *ProcessModelSearch {
   slf.Device = device
func (slf *ProcessModelSearch) SetProductAndProcedureNot(product string, procedure string) *ProcessModelSearch {
   slf.ProductNot, slf.ProcedureNot = product, 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
}
@@ -90,14 +122,6 @@
      db = db.Where("id = ?", slf.ID)
   }
   if len(slf.WorkOrder) != 0 {
      db = db.Where("work_order = ?", slf.WorkOrder)
   }
   if len(slf.OrderId) != 0 {
      db = db.Where("order_id = ?", slf.OrderId)
   }
   if len(slf.Product) != 0 {
      db = db.Where("product = ?", slf.Product)
   }
@@ -106,8 +130,12 @@
      db = db.Where("`procedure` = ?", slf.Procedure)
   }
   if len(slf.Device) != 0 {
      db = db.Where("device = ?", slf.Device)
   if len(slf.ProductNot) != 0 && len(slf.ProcedureNot) != 0 {
      db = db.Not("product = ? and procedure = ? ", slf.ProductNot, slf.ProcedureNot)
   }
   if len(slf.Procedures) != 0 {
      db = db.Where("`procedure` IN (?)", slf.Procedures)
   }
   if len(slf.Number) != 0 {
@@ -116,6 +144,18 @@
   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)
   }
   if len(slf.Numbers) != 0 {
      db = db.Where("number in (?)", slf.Numbers)
   }
   return db
@@ -199,8 +239,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)
@@ -215,8 +255,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)
@@ -236,8 +276,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)
@@ -253,8 +293,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)