From c5a0eb549cba2cd358a2d0496c44f3a289f15d9c Mon Sep 17 00:00:00 2001
From: wangpengfei <274878379@qq.com>
Date: 星期一, 28 八月 2023 14:22:06 +0800
Subject: [PATCH] fix

---
 model/subOrder.go |   11 +++++++++--
 1 files changed, 9 insertions(+), 2 deletions(-)

diff --git a/model/subOrder.go b/model/subOrder.go
index 16d2b82..d798b86 100644
--- a/model/subOrder.go
+++ b/model/subOrder.go
@@ -34,9 +34,12 @@
 	return "sub_order"
 }
 
-func NewSubOrderSearch() *SubOrderSearch {
+func NewSubOrderSearch(db *gorm.DB) *SubOrderSearch {
+	if db == nil {
+		db = mysqlx.GetDB()
+	}
 	return &SubOrderSearch{
-		Orm: mysqlx.GetDB(),
+		Orm: db,
 	}
 }
 
@@ -132,3 +135,7 @@
 	slf.SearchMap = data
 	return slf
 }
+func (slf *SubOrderSearch) SetIds(ids []int) *SubOrderSearch {
+	slf.Orm = slf.Orm.Where("id in (?)", ids)
+	return slf
+}

--
Gitblit v1.8.0