From 5ec088f7b2c1281a1874ef6ba4e783a64acb8079 Mon Sep 17 00:00:00 2001
From: MrShi <1878285526@qq.com>
Date: 星期二, 12 十一月 2024 09:15:19 +0800
Subject: [PATCH] 优化

---
 server/service/src/main/java/com/doumee/service/system/impl/SystemRoleServiceImpl.java |    4 ++++
 1 files changed, 4 insertions(+), 0 deletions(-)

diff --git a/server/service/src/main/java/com/doumee/service/system/impl/SystemRoleServiceImpl.java b/server/service/src/main/java/com/doumee/service/system/impl/SystemRoleServiceImpl.java
index 201129c..7dcaaf7 100644
--- a/server/service/src/main/java/com/doumee/service/system/impl/SystemRoleServiceImpl.java
+++ b/server/service/src/main/java/com/doumee/service/system/impl/SystemRoleServiceImpl.java
@@ -100,6 +100,10 @@
 
     @Override
     public List<SystemRole> findList(SystemRole systemRole) {
+        LoginUserInfo user = (LoginUserInfo) SecurityUtils.getSubject().getPrincipal();
+        if(Objects.nonNull(user)&&(user.getType().equals(Constants.ONE)||user.getType().equals(Constants.TWO))){
+            systemRole.setCompanyId(user.getCompanyId());
+        }
         Wrapper<SystemRole> wrapper = new QueryWrapper<>(systemRole);
         return systemRoleMapper.selectList(wrapper);
     }

--
Gitblit v1.9.3