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