|
|
@@ -1,14 +1,10 @@ |
|
|
|
package com.ningdatech.pmapi.sys.manage; |
|
|
|
|
|
|
|
import cn.hutool.core.bean.BeanUtil; |
|
|
|
import com.baomidou.mybatisplus.core.toolkit.Wrappers; |
|
|
|
import com.ningdatech.basic.util.CollUtils; |
|
|
|
import com.ningdatech.pmapi.common.util.TreeUtil; |
|
|
|
import com.ningdatech.pmapi.sys.model.entity.Menu; |
|
|
|
import com.ningdatech.pmapi.sys.model.entity.RoleMenu; |
|
|
|
import com.ningdatech.pmapi.sys.model.vo.MenuRoleVO; |
|
|
|
import com.ningdatech.pmapi.sys.service.IRoleMenuService; |
|
|
|
import com.ningdatech.pmapi.user.util.LoginUserUtil; |
|
|
|
import lombok.RequiredArgsConstructor; |
|
|
|
import org.springframework.stereotype.Component; |
|
|
|
|
|
|
@@ -26,11 +22,7 @@ import java.util.List; |
|
|
|
@RequiredArgsConstructor |
|
|
|
public class MenuManage { |
|
|
|
|
|
|
|
private final IRoleMenuService roleMenuService; |
|
|
|
|
|
|
|
public List<MenuRoleVO> buildUserMenu(List<Menu> list) { |
|
|
|
List<RoleMenu> roleMenus = roleMenuService.list(Wrappers.lambdaQuery(RoleMenu.class) |
|
|
|
.in(RoleMenu::getRoleId, LoginUserUtil.getRoleIdList())); |
|
|
|
List<MenuRoleVO> menuRoles = CollUtils.convert(list, w -> BeanUtil.copyProperties(w, MenuRoleVO.class)); |
|
|
|
return TreeUtil.buildUserTree(menuRoles); |
|
|
|
} |
|
|
|