From 09da100729793848bc01b51f7b05ca3f1e7ec64f Mon Sep 17 00:00:00 2001 From: jiangping <jp@doumee.com> Date: 星期四, 20 六月 2024 14:05:35 +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