|
|
@@ -11,10 +11,8 @@ import org.springblade.core.secure.utils.AuthUtil;
|
|
|
import org.springblade.core.tool.constant.BladeConstant;
|
|
|
import org.springblade.core.tool.constant.RoleConstant;
|
|
|
import org.springblade.core.tool.node.ForestNodeMerger;
|
|
|
-import org.springblade.core.tool.utils.CollectionUtil;
|
|
|
-import org.springblade.core.tool.utils.Func;
|
|
|
-import org.springblade.core.tool.utils.ObjectUtil;
|
|
|
-import org.springblade.core.tool.utils.StringUtil;
|
|
|
+import org.springblade.core.tool.node.TreeNode;
|
|
|
+import org.springblade.core.tool.utils.*;
|
|
|
import org.springblade.system.entity.Role;
|
|
|
import org.springblade.system.entity.RoleMenu;
|
|
|
import org.springblade.system.entity.RoleScope;
|
|
|
@@ -60,11 +58,37 @@ public class RoleServiceImpl extends ServiceImpl<RoleMapper, Role> implements IR
|
|
|
@Override
|
|
|
public List<RoleVO> tree(String tenantId) {
|
|
|
if (AuthUtil.getUserRole().contains("dealer")) {
|
|
|
- Role role = baseMapper.selectOne(new LambdaQueryWrapper<Role>()
|
|
|
+ List<RoleVO> list = new ArrayList<>();
|
|
|
+ Role dealerRole = baseMapper.selectOne(new LambdaQueryWrapper<Role>()
|
|
|
.eq(Role::getTenantId,tenantId)
|
|
|
.eq(Role::getIsDeleted,0)
|
|
|
.eq(Role::getRoleAlias,"dealer"));
|
|
|
- return ForestNodeMerger.merge(baseMapper.treeDealer(tenantId, role.getId()));
|
|
|
+ List<RoleVO> roleVOList = baseMapper.treeDealer(tenantId, dealerRole.getId());
|
|
|
+ RoleVO dealerRoleVO = new RoleVO();
|
|
|
+ dealerRoleVO.setId(dealerRole.getId());
|
|
|
+ dealerRoleVO.setRoleName(dealerRole.getRoleName());
|
|
|
+ dealerRoleVO.setParentId(dealerRole.getParentId());
|
|
|
+ dealerRoleVO.setRoleAlias(dealerRole.getRoleAlias());
|
|
|
+ dealerRoleVO.setSort(dealerRole.getSort());
|
|
|
+ dealerRoleVO.setChildren(roleVOList);
|
|
|
+ dealerRoleVO.setValue(dealerRole.getId());
|
|
|
+ dealerRoleVO.setTitle(dealerRole.getRoleName());
|
|
|
+ list.add(dealerRoleVO);
|
|
|
+ Role storeRole = baseMapper.selectOne(new LambdaQueryWrapper<Role>()
|
|
|
+ .eq(Role::getTenantId,tenantId)
|
|
|
+ .eq(Role::getIsDeleted,0)
|
|
|
+ .eq(Role::getRoleAlias,"store"));
|
|
|
+ RoleVO storeRoleVO = new RoleVO();
|
|
|
+ storeRoleVO.setId(storeRole.getId());
|
|
|
+ storeRoleVO.setRoleName(storeRole.getRoleName());
|
|
|
+ storeRoleVO.setParentId(storeRole.getParentId());
|
|
|
+ storeRoleVO.setRoleAlias(storeRole.getRoleAlias());
|
|
|
+ storeRoleVO.setSort(storeRole.getSort());
|
|
|
+ storeRoleVO.setChildren(new ArrayList<>());
|
|
|
+ storeRoleVO.setValue(storeRole.getId());
|
|
|
+ storeRoleVO.setTitle(storeRole.getRoleName());
|
|
|
+ list.add(storeRoleVO);
|
|
|
+ return ForestNodeMerger.merge(list);
|
|
|
}else{
|
|
|
String userRole = AuthUtil.getUserRole();
|
|
|
String excludeRole = null;
|