From 740b271a0f1871288cbd8dc07f4c95d1488c6e71 Mon Sep 17 00:00:00 2001
From: liujiandao <274878379@qq.com>
Date: 星期四, 26 十月 2023 17:56:37 +0800
Subject: [PATCH] opa添加

---
 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