forked from kidgrow-microservices-platform

zhaoxiaohao
2020-04-26 fc223b26f87f7a221b38084bb62d8047974bbf1c
kidgrow-business/kidgrow-usercenter/kidgrow-usercenter-server/src/main/java/com/kidgrow/usercenter/controller/SysMenuController.java
@@ -41,7 +41,7 @@
    @Autowired
    private ISysMenuService menuService;
    /**
    /**current
     * 两层循环实现建树
     *
     * @param sysMenus
@@ -199,7 +199,8 @@
        if (CollectionUtil.isEmpty(roles)) {
            return ResultBody.ok().data(Collections.emptyList());
        }
        List<SysMenu> menus = menuService.findByRoleCodes(roles.parallelStream().map(SysRole::getCode).collect(Collectors.toSet()), CommonConstant.MENU);
        Set<String> collect = roles.stream().map(e -> e.getCode()).collect(Collectors.toSet());
        List<SysMenu> menus = menuService.findByRoleCodes(collect, CommonConstant.MENU);
        return ResultBody.ok().data(treeBuilder(menus));
    }
}