发货支持仓库传参

zhangqian

2024-04-07 a8914a16b23e93f6bfd12bcfd5cbe8b24cf7eb84

 models/location_product_amount.go

@@ -35,6 +35,7 @@
      //LocationProductIds []int
      LocationIds []int
      ProductIds  []string
      Ids         []int
   }

   LocationProductAmountWithOperation struct {
@@ -80,6 +81,11 @@

func (slf *LocationProductAmountSearch) SetID(id int) *LocationProductAmountSearch {
   slf.Id = id
   return slf
}

func (slf *LocationProductAmountSearch) SetIds(ids []int) *LocationProductAmountSearch {
   slf.Ids = ids
   return slf
}

@@ -151,6 +157,10 @@

   if slf.ProductId != "" {
      db = db.Where("product_id=?", slf.ProductId)
   }

   if len(slf.Ids) > 0 {
      db = db.Where("id in (?)", slf.Ids)
   }

   return db
@@ -247,3 +257,24 @@

   return record, db.First(&record)
}

func (slf *LocationProductAmountSearch) Count() (int64, error) {
   var (
      total int64
      db    = slf.build()
   )
   err := db.Count(&total).Error
   return total, err
}

func (slf *LocationProductAmountSearch) 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
}