From 7438f17c0b3a9f79ee5890c554bedc64c362859b Mon Sep 17 00:00:00 2001
From: zhangqian <zhangqian@123.com>
Date: 星期五, 13 十月 2023 14:04:59 +0800
Subject: [PATCH] 销售计划简单数据权限

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