jiangshuai
2023-10-20 96844c22ef3fba86a55e0af1b51bc1009d6fa950
models/operation.go
@@ -28,13 +28,8 @@
      CompanyName       string                   `json:"companyName" gorm:"type:varchar(127);comment:公司名称-客户"`
      Comment           string                   `json:"comment" gorm:"type:text;comment:备注"`
      Details []*OperationDetails `json:"details" gorm:"foreignKey:OperationID;references:Id"`
      //Weight          decimal.Decimal          `json:"weight" gorm:"type:decimal(20,2);comment:重量(kg)"`
      //TransferWeight  decimal.Decimal          `json:"transferWeight" gorm:"type:decimal(20,2);comment:物流重量(kg)"`
      //CarrierID       int                      `json:"carrierID" gorm:"type:int;comment:承运商ID"`
      //CarrierName     string                   `json:"carrierName" gorm:"type:varchar(63);comment:承运商名称"`
      //Tracking        string                   `json:"tracking" gorm:"type:varchar(127);comment:追踪参考"`
      Details           []*OperationDetails        `json:"details" gorm:"foreignKey:OperationID;references:Id"`
      BaseOperationType constvar.BaseOperationType `json:"baseOperationType" gorm:"type:tinyint;not null;comment:基础作业类型"` //基础作业类型
   }
   OperationSearch struct {
@@ -46,6 +41,7 @@
      Orm      *gorm.DB
      Preload  bool
      Disuse   bool
      Ids      []int
   }
)
@@ -107,6 +103,15 @@
   return slf
}
func (slf *OperationSearch) SetIds(ids []int) *OperationSearch {
   slf.Ids = ids
   return slf
}
func (slf *OperationSearch) SetStatus(status constvar.OperationStatus) *OperationSearch {
   slf.Status = status
   return slf
}
func (slf *OperationSearch) build() *gorm.DB {
   var db = slf.Orm.Model(&Operation{})
@@ -118,7 +123,7 @@
   }
   if slf.Keyword != "" {
      db = db.Where("product_name like ?", fmt.Sprintf("%%%v%%", slf.Keyword))
      db = db.Where("number like ? or source_number like ?", fmt.Sprintf("%%%v%%", slf.Keyword), fmt.Sprintf("%%%v%%", slf.Keyword))
   }
   if slf.SourceNumber != "" {
@@ -134,11 +139,19 @@
   }
   if slf.Preload {
      db = db.Model(&Operation{}).Preload("Details").Preload("FromLocation").Preload("ToLocation")
      db = db.Model(&Operation{}).Preload("Details").Preload("Details.Product").Preload("FromLocation").Preload("ToLocation")
   }
   if slf.Disuse {
      db = db.Where("operation_type_id = ?", 0)
   }
   if len(slf.Ids) > 0 {
      db = db.Where("id in (?)", slf.Ids)
   }
   if slf.Status > 0 {
      db = db.Where("status = ?", slf.Status)
   }
   return db
@@ -241,7 +254,7 @@
   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 {
   if err := db.Preload("FromLocation").Preload("ToLocation").Find(&records).Error; err != nil {
      return records, total, fmt.Errorf("find records err: %v", err)
   }
@@ -257,7 +270,7 @@
   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 {
   if err := db.Preload("FromLocation").Preload("ToLocation").Find(&records).Error; err != nil {
      return records, fmt.Errorf("find records err: %v", err)
   }