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