根据仓库返回产品剩余数量

zhangqian

2024-03-14 527bc709815b9ea0f2ed14450262c9a5832c141f

 models/warehouse.go

@@ -38,6 +38,7 @@
      Keyword  string
      Orm      *gorm.DB
      Preload  bool
      Codes    []string
   }
)

@@ -79,6 +80,11 @@
   return slf
}

func (slf *WarehouseSearch) SetCodes(codes []string) *WarehouseSearch {
   slf.Codes = codes
   return slf
}

func (slf *WarehouseSearch) SetName(name string) *WarehouseSearch {
   slf.Name = name
   return slf
@@ -116,10 +122,15 @@
   if slf.Preload {
      db = db.Preload("Company")
   }

   if slf.Code != "" {
      db = db.Where("code = ?", slf.Code)
   }

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

   return db
}