jiangping
2024-05-08 be0301e67768fb183df7e29f411a7e54dbf15c40
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) {