From 2f32021ab211cd901e36c4cbc5875f0f072385bb Mon Sep 17 00:00:00 2001
From: liujiandao <274878379@qq.com>
Date: 星期二, 14 十一月 2023 14:00:39 +0800
Subject: [PATCH] 自动获取物料编码

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

diff --git a/models/location.go b/models/location.go
index 15830ea..ad60d86 100644
--- a/models/location.go
+++ b/models/location.go
@@ -26,6 +26,7 @@
 		RecentlyCount        string                        `json:"recentlyCount" gorm:"type:varchar(255);comment:鏈�杩戠洏鐐�"`   //鏈�杩戠洏鐐�
 		NextCount            string                        `json:"nextCount" gorm:"type:varchar(255);comment:涓嬫鐩樼偣"`       //涓嬫鐩樼偣
 		JointName            string                        `json:"jointName" gorm:"type:varchar(255);comment:鎷兼帴鍚嶇О"`       //鎷兼帴鍚嶇О
+		Children             []*Location                   `json:"children" gorm:"-"`
 	}
 
 	LocationSearch struct {
@@ -37,6 +38,7 @@
 		Orm        *gorm.DB
 		Preload    bool
 		JointNames []string
+		Ids        []int
 	}
 )
 
@@ -65,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
 }
 
@@ -113,6 +120,10 @@
 
 	if slf.Id != 0 {
 		db = db.Where("id = ?", slf.Id)
+	}
+
+	if len(slf.Ids) > 0 {
+		db = db.Where("id in (?)", slf.Ids)
 	}
 
 	if slf.Order != "" {
@@ -310,7 +321,7 @@
 		records = make([]*Location, 0)
 		db      = slf.build()
 	)
-	err := db.Find(&records)
+	err := db.Find(&records).Error
 	if err != nil {
 		fmt.Println(err)
 		return records, fmt.Errorf("func FindAll err: %v", err)

--
Gitblit v1.8.0