From be19d957b591ad27a47057929c4a8694f0776a1a Mon Sep 17 00:00:00 2001
From: jiangshuai <291802688@qq.com>
Date: 星期四, 26 十月 2023 10:15:26 +0800
Subject: [PATCH] bug修改
---
models/location_product.go | 17 +++++++++++++++++
1 files changed, 17 insertions(+), 0 deletions(-)
diff --git a/models/location_product.go b/models/location_product.go
index f42e2fe..f737765 100644
--- a/models/location_product.go
+++ b/models/location_product.go
@@ -85,6 +85,11 @@
return slf
}
+func (slf *LocationProductSearch) SetAreaId(areaId int) *LocationProductSearch {
+ slf.AreaId = areaId
+ return slf
+}
+
func (slf *LocationProductSearch) build() *gorm.DB {
var db = slf.Orm.Model(&LocationProduct{})
@@ -119,6 +124,10 @@
if len(slf.LocationIds) > 0 {
db = db.Where("location_id in (?)", slf.LocationIds)
+ }
+
+ if slf.AreaId != 0 {
+ db = db.Where("area_id = ?", slf.AreaId)
}
return db
@@ -206,3 +215,11 @@
return records, nil
}
+
+func (slf *LocationProductSearch) FirstRes() (*LocationProduct, *gorm.DB) {
+ var (
+ record = new(LocationProduct)
+ db = slf.build()
+ )
+ return record, db.First(record)
+}
--
Gitblit v1.8.0