fix

zhangqian

2023-10-19 cab6eea89a0d82710369604ecb51f1cdb122e433

 model/process_model.go

@@ -1,34 +1,52 @@
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"` //工艺模型编号
      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
      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
   }
)

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 +58,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 +78,33 @@
   return slf
}

func (slf *ProcessModelSearch) SetNumbers(numbers []string) *ProcessModelSearch {
   slf.Numbers = numbers
   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 +123,28 @@
      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)
   }

   if len(slf.Numbers) != 0 {
      db = db.Where("number in ?", slf.Numbers)
   }

   return db
@@ -173,8 +228,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 +244,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 +265,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 +282,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)