liujiandao
2023-10-23 b9a62531fb93aba42d13e62fd1b03d8dba112fa6
models/location_product.go
@@ -22,12 +22,13 @@
   LocationProductSearch struct {
      LocationProduct
      Order    string
      PageNum  int
      PageSize int
      Keyword  string
      Orm      *gorm.DB
      Preload  bool
      Order       string
      PageNum     int
      PageSize    int
      Keyword     string
      Orm         *gorm.DB
      Preload     bool
      LocationIds []int
   }
)
@@ -69,6 +70,21 @@
   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) SetLocationIds(locationIds []int) *LocationProductSearch {
   slf.LocationIds = locationIds
   return slf
}
func (slf *LocationProductSearch) build() *gorm.DB {
   var db = slf.Orm.Model(&LocationProduct{})
@@ -80,13 +96,31 @@
   }
   if slf.Keyword != "" {
      db = db.Where("product_name like ?", fmt.Sprintf("%%%v%%", slf.Keyword))
      //db = db.Where("Product.name like ?", fmt.Sprintf("%%%v%%", slf.Keyword))
      db.Joins("left join wms_location on wms_location_product.location_id = wms_location.id").
         Joins("left join material on wms_location_product.product_id = material.id").
         Joins("left join wms_product_category on wms_location_product.product_category_id = wms_product_category.id").
         Where("wms_location.name like ?", fmt.Sprintf("%%%v%%", slf.Keyword)).
         Or("material.name like ?", fmt.Sprintf("%%%v%%", slf.Keyword)).
         Or("wms_product_category.name like ?", fmt.Sprintf("%%%v%%", slf.Keyword))
   }
   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)
   }
   if len(slf.LocationIds) > 0 {
      db = db.Where("location_id in (?)", slf.LocationIds)
   }
   return db
}