|  |  |  | 
|---|
|  |  |  | import java.util.ArrayList; | 
|---|
|  |  |  | import java.util.List; | 
|---|
|  |  |  | import java.util.Objects; | 
|---|
|  |  |  | import java.util.stream.Collectors; | 
|---|
|  |  |  |  | 
|---|
|  |  |  | @Service | 
|---|
|  |  |  | public class SystemMenuBizImpl implements SystemMenuBiz { | 
|---|
|  |  |  | 
|---|
|  |  |  | List<SystemMenu> menuPool; | 
|---|
|  |  |  | if (currentMenu.getParentId() == null) { | 
|---|
|  |  |  | menuPool = systemMenuService.findRootList(); | 
|---|
|  |  |  | //2024年11月16日13:51:53 过滤同类型数据 | 
|---|
|  |  |  | menuPool = menuPool.stream().filter(i->Constants.equalsInteger(i.getType(),currentMenu.getType())).collect(Collectors.toList()); | 
|---|
|  |  |  | } else { | 
|---|
|  |  |  | SystemMenu queryDto = new SystemMenu(); | 
|---|
|  |  |  | queryDto.setParentId(currentMenu.getParentId()); | 
|---|
|  |  |  | 
|---|
|  |  |  | public List<SystemMenuListVO> findTreeByType(Integer type) { | 
|---|
|  |  |  | List<SystemMenuListVO> menus = systemMenuService.findList(type); | 
|---|
|  |  |  | List<SystemMenuListVO> rootMenus = new ArrayList<>(); | 
|---|
|  |  |  | if(type.equals(Constants.ONE)){ | 
|---|
|  |  |  | menus = menus.stream().filter(m->!m.getDisabled()).collect(Collectors.toList()); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | // 添加根菜单 | 
|---|
|  |  |  | for (SystemMenu menu : menus) { | 
|---|
|  |  |  |  | 
|---|
|  |  |  | if (menu.getParentId() == null) { | 
|---|
|  |  |  | SystemMenuListVO rootMenu = new SystemMenuListVO(); | 
|---|
|  |  |  | BeanUtils.copyProperties(menu, rootMenu, "children"); | 
|---|
|  |  |  | 
|---|
|  |  |  | List<SystemMenu> menus = new ArrayList<SystemMenu>(); | 
|---|
|  |  |  | List<SystemMenuNodeVO> rootNodes = new ArrayList<>(); | 
|---|
|  |  |  | menus = systemMenuService.findByUserId(userId,type); | 
|---|
|  |  |  | if(type.equals(Constants.ONE)){ | 
|---|
|  |  |  | if(type.equals(Constants.ONE) || type.equals(Constants.TWO)){ | 
|---|
|  |  |  | LoginUserInfo loginUserInfo = (LoginUserInfo) SecurityUtils.getSubject().getPrincipal(); | 
|---|
|  |  |  | Company company = companyService.findById(loginUserInfo.getCompanyId()); | 
|---|
|  |  |  | if(Objects.isNull(company)){ | 
|---|
|  |  |  | return rootNodes; | 
|---|
|  |  |  | } | 
|---|
|  |  |  | SystemUser systemUser = systemUserService.findById(userId); | 
|---|
|  |  |  | SystemUser dbUser = new SystemUser(); | 
|---|
|  |  |  | dbUser.setId(loginUserInfo.getId()); | 
|---|
|  |  |  | SystemUser systemUser = systemUserService.findOne(dbUser); | 
|---|
|  |  |  | if(Objects.isNull(systemUser)){ | 
|---|
|  |  |  | return rootNodes; | 
|---|
|  |  |  | } | 
|---|
|  |  |  | if(company.getUsername().equals(systemUser.getUsername())){ | 
|---|
|  |  |  | if(company.getUsername().equals(systemUser.getUsername()) | 
|---|
|  |  |  | || (Constants.equalsInteger(type,Constants.ONE) && !Constants.equalsInteger(systemUser.getCompanyId(),company.getId()))){ | 
|---|
|  |  |  | SystemMenu systemMenu = new SystemMenu(); | 
|---|
|  |  |  | systemMenu.setType(type); | 
|---|
|  |  |  | systemMenu.setDeleted(Boolean.FALSE); | 
|---|
|  |  |  | systemMenu.setDisabled(Boolean.FALSE); | 
|---|
|  |  |  | menus = systemMenuService.findList(systemMenu); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | } | 
|---|