From b6fde711bf33cede4cb84f85146c32b0c0c61aa5 Mon Sep 17 00:00:00 2001
From: MrShi <1878285526@qq.com>
Date: 星期一, 04 十一月 2024 09:06:28 +0800
Subject: [PATCH] 优化需求
---
server/service/src/main/java/com/doumee/service/system/impl/SystemRoleServiceImpl.java | 11 ++++++++++-
1 files changed, 10 insertions(+), 1 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 94318e0..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
@@ -5,6 +5,7 @@
import com.doumee.core.model.LoginUserInfo;
import com.doumee.core.model.PageData;
import com.doumee.core.model.PageWrap;
+import com.doumee.core.utils.Constants;
import com.doumee.core.utils.Utils;
import com.doumee.dao.system.SystemRoleMapper;
import com.doumee.dao.system.dto.QuerySystemRoleDTO;
@@ -43,6 +44,9 @@
@Override
public Integer create(SystemRole systemRole) {
+ LoginUserInfo principal = (LoginUserInfo) SecurityUtils.getSubject().getPrincipal();
+ systemRole.setCompanyId(principal.getCompanyId());
+ systemRole.setType(principal.getType());
systemRoleMapper.insert(systemRole);
return systemRole.getId();
}
@@ -96,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);
}
@@ -105,7 +113,8 @@
LoginUserInfo principal = (LoginUserInfo) SecurityUtils.getSubject().getPrincipal();
Utils.MP.blankToNull(pageWrap.getModel());
QuerySystemRoleDTO model = pageWrap.getModel();
- model.setType(Objects.nonNull(model.getType()) ? model.getType() : principal.getType());
+ model.setType(Constants.formatIntegerNum(principal.getType()));
+ model.setCompanyId(principal.getCompanyId());
PageHelper.startPage(pageWrap.getPage(), pageWrap.getCapacity());
List<SystemRoleListVO> roleList = systemRoleMapper.selectManageList(model, pageWrap.getOrderByClause());
for (SystemRoleListVO role : roleList) {
--
Gitblit v1.9.3