From 244b7b5fb973698648f9ff1fc3fbd93e7271e284 Mon Sep 17 00:00:00 2001 From: zhangqian <zhangqian@123.com> Date: 星期六, 28 十月 2023 16:45:26 +0800 Subject: [PATCH] 获取员工列表只展示下属员工 --- model/user.go | 14 ++++++++++++++ 1 files changed, 14 insertions(+), 0 deletions(-) diff --git a/model/user.go b/model/user.go index 93e8c3d..42f71dd 100644 --- a/model/user.go +++ b/model/user.go @@ -38,6 +38,7 @@ PageNum int PageSize int Orm *gorm.DB + IDs []int } ) @@ -77,6 +78,11 @@ return slf } +func (slf *UserSearch) SetUserIds(userIds []int) *UserSearch { + slf.IDs = userIds + return slf +} + func (slf *UserSearch) SetUserName(username string) *UserSearch { slf.Username = username return slf @@ -106,6 +112,14 @@ db = db.Order(slf.Order) } + if slf.ID != 0 { + db = db.Where("id = ?", slf.ID) + } + + if len(slf.IDs) > 0 { + db = db.Where("id in ?", slf.IDs) + } + return db } -- Gitblit v1.8.0