perf:调整获取自己的功能点

This commit is contained in:
曾文豪
2024-06-26 09:03:36 +08:00
parent d77eb50a5b
commit 78cb8dafeb
2 changed files with 23 additions and 9 deletions

View File

@@ -4,7 +4,10 @@ import cn.hutool.core.util.StrUtil;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page; import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.tiesheng.role.pojos.dao.*; import com.tiesheng.role.pojos.dao.*;
import com.tiesheng.role.pojos.dto.*; import com.tiesheng.role.pojos.dto.GroupRxUpdateDTO;
import com.tiesheng.role.pojos.dto.MenuListDTO;
import com.tiesheng.role.pojos.dto.OwnerMenuDTO;
import com.tiesheng.role.pojos.dto.OwnerPointDTO;
import com.tiesheng.role.pojos.vo.GroupTypeDTO; import com.tiesheng.role.pojos.vo.GroupTypeDTO;
import com.tiesheng.role.pojos.vo.RoleUserPageVO; import com.tiesheng.role.pojos.vo.RoleUserPageVO;
import com.tiesheng.role.pojos.vo.ServiceMenuVO; import com.tiesheng.role.pojos.vo.ServiceMenuVO;
@@ -284,13 +287,25 @@ public class RoleController {
TokenBean tokenBean = TsTokenConfig.get(); TokenBean tokenBean = TsTokenConfig.get();
List<CoreRoleAuthority> allOwnerMenus = coreRoleService.getOwnerAuthorityLeafList(tokenBean.getId(), tokenBean.getRoleId()); List<CoreRoleAuthority> allOwnerMenus = coreRoleService.getOwnerAuthorityLeafList(tokenBean.getId(), tokenBean.getRoleId());
String parentId;
if (StrUtil.isNotEmpty(dto.getParentNo())) {
CoreRoleAuthority selected = coreRoleService.getAuthorityMapper().selectOne(new QueryWrapper<CoreRoleAuthority>()
.eq("no", dto.getParentNo())
.eq("is_deleted", 0)
.last("limit 1")
);
parentId = selected == null ? "" : selected.getId();
} else {
parentId = "";
}
List<CoreRoleAuthority> collect = allOwnerMenus.stream() List<CoreRoleAuthority> collect = allOwnerMenus.stream()
.filter(it -> Objects.equals(it.getType(), "point")) .filter(it -> Objects.equals(it.getType(), "point"))
.filter(it -> it.getService().equals(dto.getService())) .filter(it -> it.getService().equals(dto.getService()))
.filter(it -> it.getPlatform().equals(dto.getPlatform())) .filter(it -> it.getPlatform().equals(dto.getPlatform()))
.filter(it -> { .filter(it -> {
if (StrUtil.isNotEmpty(dto.getParent())) { if (StrUtil.isNotEmpty(parentId)) {
return dto.getParent().equals(it.getParent()); return parentId.equals(it.getParent());
} }
return true; return true;
}).collect(Collectors.toList()); }).collect(Collectors.toList());

View File

@@ -2,18 +2,17 @@ package com.tiesheng.role.pojos.dto;
public class OwnerPointDTO extends OwnerMenuDTO { public class OwnerPointDTO extends OwnerMenuDTO {
private String parent; private String parentNo;
/////////////////////////////////////////////////////////////////////////// ///////////////////////////////////////////////////////////////////////////
// setter\getter // setter\getter
/////////////////////////////////////////////////////////////////////////// ///////////////////////////////////////////////////////////////////////////
public String getParent() { public String getParentNo() {
return parent; return parentNo;
} }
public void setParent(String parent) { public void setParentNo(String parentNo) {
this.parent = parent; this.parentNo = parentNo;
} }
} }