From b68185aed1a86536b01fdfbca6b2cca7bd71a50f Mon Sep 17 00:00:00 2001 From: zhangqian <zhangqian@123.com> Date: 星期六, 23 三月 2024 17:12:18 +0800 Subject: [PATCH] 增加token验证和数据权限过滤 --- models/outsourcing_order.go | 9 +++++++++ 1 files changed, 9 insertions(+), 0 deletions(-) diff --git a/models/outsourcing_order.go b/models/outsourcing_order.go index 8ab108c..11e19cd 100644 --- a/models/outsourcing_order.go +++ b/models/outsourcing_order.go @@ -65,6 +65,11 @@ return slf } +func (slf *OutsourcingOrderSearch) SetEnterpriseID(id uint) *OutsourcingOrderSearch { + slf.EnterpriseID = id + return slf +} + func (slf *OutsourcingOrderSearch) SetIDs(ids []uint) *OutsourcingOrderSearch { slf.IDs = ids return slf @@ -96,6 +101,10 @@ db = db.Where("id = ?", slf.ID) } + if slf.EnterpriseID != 0 { + db = db.Where("enterprise_id = ?", slf.EnterpriseID) + } + if len(slf.IDs) != 0 { db = db.Where("id in ?", slf.IDs) } -- Gitblit v1.8.0