仓库功能修改

liujiandao

2023-10-09 2716487d13f3521adeb25a48aa07a89f885d32c7

 models/location_product.go

@@ -69,6 +69,16 @@
   return slf
}

func (slf *LocationProductSearch) SetProductId(productId string) *LocationProductSearch {
   slf.ProductId = productId
   return slf
}

func (slf *LocationProductSearch) SetLocationId(locationId int) *LocationProductSearch {
   slf.LocationId = locationId
   return slf
}

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

@@ -85,6 +95,14 @@

   if slf.Preload {
      db = db.Model(&LocationProduct{}).Preload("Location").Preload("Area").Preload("ProductCategory").Preload("Product")
   }

   if slf.LocationId != 0 {
      db = db.Where("location_id = ?", slf.LocationId)
   }

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

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

@@ -166,7 +184,7 @@
      db      = slf.build()
   )

   if err := db.Find(&records).Error; err != nil {
   if err := db.Preload("Location").Preload("ProductCategory").Preload("Product").Find(&records).Error; err != nil {
      return records, fmt.Errorf("find records err: %v", err)
   }