优化代码

RuoYi

2025-03-01 d91d0f51bb5eff7bed395dcfb388996fefbbd3c3

 src/main/java/com/ruoyi/project/system/service/impl/SysMenuServiceImpl.java

@@ -10,6 +10,7 @@
import java.util.stream.Collectors;

import org.springframework.beans.factory.annotation.Autowired;

import org.springframework.stereotype.Service;

import com.ruoyi.common.constant.Constants;

import com.ruoyi.common.constant.UserConstants;

import com.ruoyi.common.utils.SecurityUtils;

import com.ruoyi.common.utils.StringUtils;

@@ -36,7 +37,7 @@


    @Autowired

    private SysMenuMapper menuMapper;

    


    @Autowired

    private SysRoleMapper roleMapper;



@@ -100,6 +101,27 @@
    }



    /**

     * 根据角色ID查询权限

     * 
     * @param roleId 角色ID

     * @return 权限列表

     */

    @Override

    public Set<String> selectMenuPermsByRoleId(Long roleId)

    {

        List<String> perms = menuMapper.selectMenuPermsByRoleId(roleId);

        Set<String> permsSet = new HashSet<>();

        for (String perm : perms)

        {

            if (StringUtils.isNotEmpty(perm))

            {

                permsSet.addAll(Arrays.asList(perm.trim().split(",")));

            }

        }

        return permsSet;

    }



    /**

     * 根据用户ID查询菜单

     * 

     * @param userId 用户名称

@@ -127,7 +149,7 @@
     * @return 选中菜单列表

     */

    @Override

    public List<Integer> selectMenuListByRoleId(Long roleId)

    public List<Long> selectMenuListByRoleId(Long roleId)

    {

        SysRole role = roleMapper.selectRoleById(roleId);

        return menuMapper.selectMenuListByRoleId(roleId, role.isMenuCheckStrictly());

@@ -150,9 +172,10 @@
            router.setName(getRouteName(menu));

            router.setPath(getRouterPath(menu));

            router.setComponent(getComponent(menu));

            router.setMeta(new MetaVo(menu.getMenuName(), menu.getIcon(), StringUtils.equals("1", menu.getIsCache())));

            router.setQuery(menu.getQuery());

            router.setMeta(new MetaVo(menu.getMenuName(), menu.getIcon(), StringUtils.equals("1", menu.getIsCache()), menu.getPath()));

            List<SysMenu> cMenus = menu.getChildren();

            if (!cMenus.isEmpty() && cMenus.size() > 0 && UserConstants.TYPE_DIR.equals(menu.getMenuType()))

            if (StringUtils.isNotEmpty(cMenus) && UserConstants.TYPE_DIR.equals(menu.getMenuType()))

            {

                router.setAlwaysShow(true);

                router.setRedirect("noRedirect");

@@ -160,12 +183,28 @@
            }

            else if (isMenuFrame(menu))

            {

                router.setMeta(null);

                List<RouterVo> childrenList = new ArrayList<RouterVo>();

                RouterVo children = new RouterVo();

                children.setPath(menu.getPath());

                children.setComponent(menu.getComponent());

                children.setName(StringUtils.capitalize(menu.getPath()));

                children.setMeta(new MetaVo(menu.getMenuName(), menu.getIcon(), StringUtils.equals("1", menu.getIsCache())));

                children.setName(getRouteName(menu.getRouteName(), menu.getPath()));

                children.setMeta(new MetaVo(menu.getMenuName(), menu.getIcon(), StringUtils.equals("1", menu.getIsCache()), menu.getPath()));

                children.setQuery(menu.getQuery());

                childrenList.add(children);

                router.setChildren(childrenList);

            }

            else if (menu.getParentId().intValue() == 0 && isInnerLink(menu))

            {

                router.setMeta(new MetaVo(menu.getMenuName(), menu.getIcon()));

                router.setPath("/");

                List<RouterVo> childrenList = new ArrayList<RouterVo>();

                RouterVo children = new RouterVo();

                String routerPath = innerLinkReplaceEach(menu.getPath());

                children.setPath(routerPath);

                children.setComponent(UserConstants.INNER_LINK);

                children.setName(getRouteName(menu.getRouteName(), routerPath));

                children.setMeta(new MetaVo(menu.getMenuName(), menu.getIcon(), menu.getPath()));

                childrenList.add(children);

                router.setChildren(childrenList);

            }

@@ -184,11 +223,7 @@
    public List<SysMenu> buildMenuTree(List<SysMenu> menus)

    {

        List<SysMenu> returnList = new ArrayList<SysMenu>();

        List<Long> tempList = new ArrayList<Long>();

        for (SysMenu dept : menus)

        {

            tempList.add(dept.getMenuId());

        }

        List<Long> tempList = menus.stream().map(SysMenu::getMenuId).collect(Collectors.toList());

        for (Iterator<SysMenu> iterator = menus.iterator(); iterator.hasNext();)

        {

            SysMenu menu = (SysMenu) iterator.next();

@@ -241,7 +276,7 @@
    public boolean hasChildByMenuId(Long menuId)

    {

        int result = menuMapper.hasChildByMenuId(menuId);

        return result > 0 ? true : false;

        return result > 0;

    }



    /**

@@ -254,7 +289,7 @@
    public boolean checkMenuExistRole(Long menuId)

    {

        int result = roleMenuMapper.checkMenuExistRole(menuId);

        return result > 0 ? true : false;

        return result > 0;

    }



    /**

@@ -300,7 +335,7 @@
     * @return 结果

     */

    @Override

    public String checkMenuNameUnique(SysMenu menu)

    public boolean checkMenuNameUnique(SysMenu menu)

    {

        Long menuId = StringUtils.isNull(menu.getMenuId()) ? -1L : menu.getMenuId();

        SysMenu info = menuMapper.checkMenuNameUnique(menu.getMenuName(), menu.getParentId());

@@ -319,13 +354,25 @@
     */

    public String getRouteName(SysMenu menu)

    {

        String routerName = StringUtils.capitalize(menu.getPath());

        // 非外链并且是一级目录（类型为目录）

        if (isMenuFrame(menu))

        {

            routerName = StringUtils.EMPTY;

            return StringUtils.EMPTY;

        }

        return routerName;

        return getRouteName(menu.getRouteName(), menu.getPath());

    }



    /**

     * 获取路由名称，如没有配置路由名称则取路由地址

     * 
     * @param name 路由名称

     * @param path 路由地址

     * @return 路由名称（驼峰格式）

     */

    public String getRouteName(String name, String path)

    {

        String routerName = StringUtils.isNotEmpty(name) ? name : path;

        return StringUtils.capitalize(routerName);

    }



    /**

@@ -337,6 +384,11 @@
    public String getRouterPath(SysMenu menu)

    {

        String routerPath = menu.getPath();

        // 内链打开外网方式

        if (menu.getParentId().intValue() != 0 && isInnerLink(menu))

        {

            routerPath = innerLinkReplaceEach(routerPath);

        }

        // 非外链并且是一级目录（类型为目录）

        if (0 == menu.getParentId().intValue() && UserConstants.TYPE_DIR.equals(menu.getMenuType())

                && UserConstants.NO_FRAME.equals(menu.getIsFrame()))

@@ -364,6 +416,10 @@
        {

            component = menu.getComponent();

        }

        else if (StringUtils.isEmpty(menu.getComponent()) && menu.getParentId().intValue() != 0 && isInnerLink(menu))

        {

            component = UserConstants.INNER_LINK;

        }

        else if (StringUtils.isEmpty(menu.getComponent()) && isParentView(menu))

        {

            component = UserConstants.PARENT_VIEW;

@@ -382,7 +438,7 @@
        return menu.getParentId().intValue() == 0 && UserConstants.TYPE_MENU.equals(menu.getMenuType())

                && menu.getIsFrame().equals(UserConstants.NO_FRAME);

    }

    


    /**

     * 是否为parent_view组件

     * 

@@ -392,6 +448,17 @@
    public boolean isParentView(SysMenu menu)

    {

        return menu.getParentId().intValue() != 0 && UserConstants.TYPE_DIR.equals(menu.getMenuType());

    }



    /**

     * 是否为内链组件

     * 
     * @param menu 菜单信息

     * @return 结果

     */

    public boolean isInnerLink(SysMenu menu)

    {

        return menu.getIsFrame().equals(UserConstants.NO_FRAME) && StringUtils.ishttp(menu.getPath());

    }



    /**

@@ -420,8 +487,8 @@
    /**

     * 递归列表

     * 

     * @param list

     * @param t

     * @param list 分类表

     * @param t 子节点

     */

    private void recursionFn(List<SysMenu> list, SysMenu t)

    {

@@ -460,6 +527,17 @@
     */

    private boolean hasChild(List<SysMenu> list, SysMenu t)

    {

        return getChildList(list, t).size() > 0 ? true : false;

        return getChildList(list, t).size() > 0;

    }



    /**

     * 内链域名特殊字符替换

     * 
     * @return 替换后的内链域名

     */

    public String innerLinkReplaceEach(String path)

    {

        return StringUtils.replaceEach(path, new String[] { Constants.HTTP, Constants.HTTPS, Constants.WWW, ".", ":" },

                new String[] { "", "", "", "/", "/" });

    }

}