fix

zhangqian

2023-12-01 8324f872ef3a4d0c978a9b1d062800c6a1701c12

 model/production_progress.go

@@ -1,276 +1,286 @@
package model

import (
   "apsClient/pkg/sqlitex"
   "fmt"
   "github.com/jinzhu/gorm"
)

type (
   ProductionProgress struct {
      gorm.Model
      ProceduresID       uint   `gorm:"index;type:varchar(191)" json:"proceduresId"` //procedures表的id
      WorkOrderID        string `gorm:"index;type:varchar(191);not null" json:"workOrderID"`
      OrderID            string `gorm:"index;type:varchar(191);not null" json:"orderID"`
      ProcedureID        string `gorm:"type:varchar(191)" json:"procedureId"`
      ProductProcedureID string `gorm:"type:varchar(191);not null" json:"productProcedureID"` //产品工序id
      DeviceID           string `gorm:"type:varchar(191);not null" json:"deviceId"`
      FinishedQuantity   int64  `gorm:"type:int;not null" json:"finishedQuantity"`
      Channel            int32  `gorm:"type:int" json:"channel"` //通道
      TotalQuantity      int64  `gorm:"type:int;not null" json:"totalQuantity"`
   }

   ProductionProgressSearch struct {
      ProductionProgress
      Order      string
      PageNum    int
      PageSize   int
      Orm        *gorm.DB
      UnFinished bool
   }
)

func (slf *ProductionProgress) TableName() string {
   return "production_progress"
}

func NewProductionProgressSearch(db *gorm.DB) *ProductionProgressSearch {
   if db == nil {
      db = sqlitex.GetDB()
   }
   return &ProductionProgressSearch{Orm: db}
}

func (slf *ProductionProgressSearch) SetOrm(tx *gorm.DB) *ProductionProgressSearch {
   slf.Orm = tx
   return slf
}

func (slf *ProductionProgressSearch) SetPage(page, size int) *ProductionProgressSearch {
   slf.PageNum, slf.PageSize = page, size
   return slf
}

func (slf *ProductionProgressSearch) SetOrder(order string) *ProductionProgressSearch {
   slf.Order = order
   return slf
}

func (slf *ProductionProgressSearch) SetWorkOrderId(orderId string) *ProductionProgressSearch {
   slf.WorkOrderID = orderId
   return slf
}

func (slf *ProductionProgressSearch) SetProcedureId(procedureId string) *ProductionProgressSearch {
   slf.ProcedureID = procedureId
   return slf
}

func (slf *ProductionProgressSearch) SetProceduresId(proceduresId uint) *ProductionProgressSearch {
   slf.ProceduresID = proceduresId
   return slf
}

func (slf *ProductionProgressSearch) SetDeviceId(id string) *ProductionProgressSearch {
   slf.DeviceID = id
   return slf
}

func (slf *ProductionProgressSearch) SetId(id uint) *ProductionProgressSearch {
   slf.ID = id
   return slf
}

func (slf *ProductionProgressSearch) SetChannel(channel int32) *ProductionProgressSearch {
   slf.Channel = channel
   return slf
}

func (slf *ProductionProgressSearch) SetUnFinished() *ProductionProgressSearch {
   slf.UnFinished = true
   return slf
}

func (slf *ProductionProgressSearch) build() *gorm.DB {
   var db = slf.Orm.Model(&ProductionProgress{})

   if slf.Order != "" {
      db = db.Order(slf.Order)
   }

   if slf.ID != 0 {
      db = db.Where("id = ?", slf.ID)
   }

   if slf.WorkOrderID != "" {
      db = db.Where("work_order_id = ?", slf.WorkOrderID)
   }

   if slf.OrderID != "" {
      db = db.Where("order_id = ?", slf.OrderID)
   }

   if slf.ProcedureID != "" {
      db = db.Where("procedure_id = ?", slf.ProcedureID)
   }

   if slf.ProceduresID != 0 {
      db = db.Where("procedures_id = ?", slf.ProceduresID)
   }

   if slf.DeviceID != "" {
      db = db.Where("device_id = ?", slf.DeviceID)
   }

   if slf.Channel != 0 {
      db = db.Where("channel = ?", slf.Channel)
   }

   if slf.UnFinished {
      db = db.Where("finished_quantity <  total_quantity")
   }

   return db
}

// Create 单条插入
func (slf *ProductionProgressSearch) Create(record *ProductionProgress) error {
   var db = slf.build()

   if err := db.Create(record).Error; err != nil {
      return fmt.Errorf("create err: %v, record: %+v", err, record)
   }

   return nil
}

func (slf *ProductionProgressSearch) Save(record *ProductionProgress) error {
   var db = slf.build()

   if err := db.Save(record).Error; err != nil {
      return fmt.Errorf("save err: %v, record: %+v", err, record)
   }

   return nil
}

func (slf *ProductionProgressSearch) UpdateByMap(upMap map[string]interface{}) error {
   var (
      db = slf.build()
   )

   if err := db.Updates(upMap).Error; err != nil {
      return fmt.Errorf("update by map err: %v, upMap: %+v", err, upMap)
   }

   return nil
}

func (slf *ProductionProgressSearch) UpdateByQuery(query string, args []interface{}, upMap map[string]interface{}) error {
   var (
      db = slf.Orm.Table(slf.TableName()).Where(query, args...)
   )

   if err := db.Updates(upMap).Error; err != nil {
      return fmt.Errorf("update by query err: %v, query: %s, args: %+v, upMap: %+v", err, query, args, upMap)
   }

   return nil
}

func (slf *ProductionProgressSearch) Delete() error {
   var db = slf.build()

   if err := db.Unscoped().Delete(&ProductionProgress{}).Error; err != nil {
      return err
   }

   return nil
}

func (slf *ProductionProgressSearch) First() (*ProductionProgress, error) {
   var (
      record = new(ProductionProgress)
      db     = slf.build()
   )

   if err := db.First(record).Error; err != nil {
      return record, err
   }

   return record, nil
}

func (slf *ProductionProgressSearch) Find() ([]*ProductionProgress, int64, error) {
   var (
      records = make([]*ProductionProgress, 0)
      total   int64
      db      = slf.build()
   )

   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 err := db.Find(&records).Error; err != nil {
      return records, total, fmt.Errorf("find records err: %v", err)
   }

   return records, total, nil
}

func (slf *ProductionProgressSearch) FindNotTotal() ([]*ProductionProgress, error) {
   var (
      records = make([]*ProductionProgress, 0)
      db      = slf.build()
   )

   if slf.PageNum*slf.PageSize > 0 {
      db = db.Offset((slf.PageNum - 1) * slf.PageSize).Limit(slf.PageSize)
   }
   if err := db.Find(&records).Error; err != nil {
      return records, fmt.Errorf("find records err: %v", err)
   }

   return records, nil
}

// FindByQuery 指定条件查询.
func (slf *ProductionProgressSearch) FindByQuery(query string, args []interface{}) ([]*ProductionProgress, int64, error) {
   var (
      records = make([]*ProductionProgress, 0)
      total   int64
      db      = slf.Orm.Table(slf.TableName()).Where(query, args...)
   )

   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 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)
   }

   return records, total, nil
}

// FindByQueryNotTotal 指定条件查询&不查询总条数.
func (slf *ProductionProgressSearch) FindByQueryNotTotal(query string, args []interface{}) ([]*ProductionProgress, error) {
   var (
      records = make([]*ProductionProgress, 0)
      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 err := db.Find(&records).Error; err != nil {
      return records, fmt.Errorf("find by query records err: %v, query: %s, args: %+v", err, query, args)
   }

   return records, nil
}
//
//import (
//   "apsClient/pkg/sqlitex"
//   "fmt"
//   "github.com/jinzhu/gorm"
//)
//
//type (
//   ProductionProgress struct {
//      gorm.Model
//      ProceduresID       uint   `gorm:"index;type:varchar(191)" json:"proceduresId"` //procedures表的id
//      WorkOrderID        string `gorm:"index;type:varchar(191);not null" json:"workOrderID"`
//      OrderID            string `gorm:"index;type:varchar(191);not null" json:"orderID"`
//      ProcedureID        string `gorm:"type:varchar(191)" json:"procedureId"`
//      ProductProcedureID string `gorm:"type:varchar(191);" json:"productProcedureID"` //产品工序id
//      DeviceID           string `gorm:"type:varchar(191);not null" json:"deviceId"`
//      FinishedQuantity   int64  `gorm:"type:int;not null" json:"finishedQuantity"`
//      Channel            int32  `gorm:"type:int" json:"channel"` //通道
//      TotalQuantity      int64  `gorm:"type:int;not null" json:"totalQuantity"`
//   }
//
//   ProductionProgressSearch struct {
//      ProductionProgress
//      Order      string
//      PageNum    int
//      PageSize   int
//      Orm        *gorm.DB
//      UnFinished bool
//   }
//)
//
//func (slf *ProductionProgress) TableName() string {
//   return "production_progress"
//}
//
//func NewProductionProgressSearch(db *gorm.DB) *ProductionProgressSearch {
//   if db == nil {
//      db = sqlitex.GetDB()
//   }
//   return &ProductionProgressSearch{Orm: db}
//}
//
//func (slf *ProductionProgressSearch) SetOrm(tx *gorm.DB) *ProductionProgressSearch {
//   slf.Orm = tx
//   return slf
//}
//
//func (slf *ProductionProgressSearch) SetPage(page, size int) *ProductionProgressSearch {
//   slf.PageNum, slf.PageSize = page, size
//   return slf
//}
//
//func (slf *ProductionProgressSearch) SetOrder(order string) *ProductionProgressSearch {
//   slf.Order = order
//   return slf
//}
//
//func (slf *ProductionProgressSearch) SetWorkOrderId(orderId string) *ProductionProgressSearch {
//   slf.WorkOrderID = orderId
//   return slf
//}
//
//func (slf *ProductionProgressSearch) SetProcedureId(procedureId string) *ProductionProgressSearch {
//   slf.ProcedureID = procedureId
//   return slf
//}
//
//func (slf *ProductionProgressSearch) SetProceduresId(proceduresId uint) *ProductionProgressSearch {
//   slf.ProceduresID = proceduresId
//   return slf
//}
//
//func (slf *ProductionProgressSearch) SetDeviceId(id string) *ProductionProgressSearch {
//   slf.DeviceID = id
//   return slf
//}
//
//func (slf *ProductionProgressSearch) SetProductProcedureId(productProcedureId string) *ProductionProgressSearch {
//   slf.ProductProcedureID = productProcedureId
//   return slf
//}
//
//func (slf *ProductionProgressSearch) SetId(id uint) *ProductionProgressSearch {
//   slf.ID = id
//   return slf
//}
//
//func (slf *ProductionProgressSearch) SetChannel(channel int32) *ProductionProgressSearch {
//   slf.Channel = channel
//   return slf
//}
//
//func (slf *ProductionProgressSearch) SetUnFinished() *ProductionProgressSearch {
//   slf.UnFinished = true
//   return slf
//}
//
//func (slf *ProductionProgressSearch) build() *gorm.DB {
//   var db = slf.Orm.Model(&ProductionProgress{})
//
//   if slf.Order != "" {
//      db = db.Order(slf.Order)
//   }
//
//   if slf.ID != 0 {
//      db = db.Where("id = ?", slf.ID)
//   }
//
//   if slf.WorkOrderID != "" {
//      db = db.Where("work_order_id = ?", slf.WorkOrderID)
//   }
//
//   if slf.OrderID != "" {
//      db = db.Where("order_id = ?", slf.OrderID)
//   }
//
//   if slf.ProcedureID != "" {
//      db = db.Where("procedure_id = ?", slf.ProcedureID)
//   }
//
//   if slf.ProceduresID != 0 {
//      db = db.Where("procedures_id = ?", slf.ProceduresID)
//   }
//
//   if slf.DeviceID != "" {
//      db = db.Where("device_id = ?", slf.DeviceID)
//   }
//
//   if slf.Channel != 0 {
//      db = db.Where("channel = ?", slf.Channel)
//   }
//
//   if slf.UnFinished {
//      db = db.Where("finished_quantity <  total_quantity")
//   }
//
//   if slf.ProductProcedureID != "" {
//      db = db.Where("product_procedure_id = ?", slf.ProductProcedureID)
//   }
//
//   return db
//}
//
//// Create 单条插入
//func (slf *ProductionProgressSearch) Create(record *ProductionProgress) error {
//   var db = slf.build()
//
//   if err := db.Create(record).Error; err != nil {
//      return fmt.Errorf("create err: %v, record: %+v", err, record)
//   }
//
//   return nil
//}
//
//func (slf *ProductionProgressSearch) Save(record *ProductionProgress) error {
//   var db = slf.build()
//
//   if err := db.Save(record).Error; err != nil {
//      return fmt.Errorf("save err: %v, record: %+v", err, record)
//   }
//
//   return nil
//}
//
//func (slf *ProductionProgressSearch) UpdateByMap(upMap map[string]interface{}) error {
//   var (
//      db = slf.build()
//   )
//
//   if err := db.Updates(upMap).Error; err != nil {
//      return fmt.Errorf("update by map err: %v, upMap: %+v", err, upMap)
//   }
//
//   return nil
//}
//
//func (slf *ProductionProgressSearch) UpdateByQuery(query string, args []interface{}, upMap map[string]interface{}) error {
//   var (
//      db = slf.Orm.Table(slf.TableName()).Where(query, args...)
//   )
//
//   if err := db.Updates(upMap).Error; err != nil {
//      return fmt.Errorf("update by query err: %v, query: %s, args: %+v, upMap: %+v", err, query, args, upMap)
//   }
//
//   return nil
//}
//
//func (slf *ProductionProgressSearch) Delete() error {
//   var db = slf.build()
//
//   if err := db.Unscoped().Delete(&ProductionProgress{}).Error; err != nil {
//      return err
//   }
//
//   return nil
//}
//
//func (slf *ProductionProgressSearch) First() (*ProductionProgress, error) {
//   var (
//      record = new(ProductionProgress)
//      db     = slf.build()
//   )
//
//   if err := db.First(record).Error; err != nil {
//      return record, err
//   }
//
//   return record, nil
//}
//
//func (slf *ProductionProgressSearch) Find() ([]*ProductionProgress, int64, error) {
//   var (
//      records = make([]*ProductionProgress, 0)
//      total   int64
//      db      = slf.build()
//   )
//
//   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 err := db.Find(&records).Error; err != nil {
//      return records, total, fmt.Errorf("find records err: %v", err)
//   }
//
//   return records, total, nil
//}
//
//func (slf *ProductionProgressSearch) FindNotTotal() ([]*ProductionProgress, error) {
//   var (
//      records = make([]*ProductionProgress, 0)
//      db      = slf.build()
//   )
//
//   if slf.PageNum*slf.PageSize > 0 {
//      db = db.Offset((slf.PageNum - 1) * slf.PageSize).Limit(slf.PageSize)
//   }
//   if err := db.Find(&records).Error; err != nil {
//      return records, fmt.Errorf("find records err: %v", err)
//   }
//
//   return records, nil
//}
//
//// FindByQuery 指定条件查询.
//func (slf *ProductionProgressSearch) FindByQuery(query string, args []interface{}) ([]*ProductionProgress, int64, error) {
//   var (
//      records = make([]*ProductionProgress, 0)
//      total   int64
//      db      = slf.Orm.Table(slf.TableName()).Where(query, args...)
//   )
//
//   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 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)
//   }
//
//   return records, total, nil
//}
//
//// FindByQueryNotTotal 指定条件查询&不查询总条数.
//func (slf *ProductionProgressSearch) FindByQueryNotTotal(query string, args []interface{}) ([]*ProductionProgress, error) {
//   var (
//      records = make([]*ProductionProgress, 0)
//      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 err := db.Find(&records).Error; err != nil {
//      return records, fmt.Errorf("find by query records err: %v, query: %s, args: %+v", err, query, args)
//   }
//
//   return records, nil
//}