From 1dd416deaf053a23a56b42a1ab18fe9f621fefb0 Mon Sep 17 00:00:00 2001
From: liujiandao <274878379@qq.com>
Date: 星期一, 13 十一月 2023 20:01:30 +0800
Subject: [PATCH] 仓库与位置修改

---
 models/location.go |   10 ++++++++++
 1 files changed, 10 insertions(+), 0 deletions(-)

diff --git a/models/location.go b/models/location.go
index 028e875..ad60d86 100644
--- a/models/location.go
+++ b/models/location.go
@@ -38,6 +38,7 @@
 		Orm        *gorm.DB
 		Preload    bool
 		JointNames []string
+		Ids        []int
 	}
 )
 
@@ -66,6 +67,11 @@
 
 func (slf *LocationSearch) SetID(ID int) *LocationSearch {
 	slf.Id = ID
+	return slf
+}
+
+func (slf *LocationSearch) SetIds(ids []int) *LocationSearch {
+	slf.Ids = ids
 	return slf
 }
 
@@ -116,6 +122,10 @@
 		db = db.Where("id = ?", slf.Id)
 	}
 
+	if len(slf.Ids) > 0 {
+		db = db.Where("id in (?)", slf.Ids)
+	}
+
 	if slf.Order != "" {
 		db = db.Order(slf.Order)
 	}

--
Gitblit v1.8.0