优化导入带标题文件关闭清理

RuoYi

2024-12-25 eb9b81f6c14687ec03fa92fe49c82801fa99bd19

 src/main/java/com/ruoyi/common/utils/SecurityUtils.java

@@ -1,11 +1,17 @@
package com.ruoyi.common.utils;



import java.util.Collection;

import java.util.List;

import java.util.stream.Collectors;

import org.springframework.security.core.Authentication;

import org.springframework.security.core.context.SecurityContextHolder;

import org.springframework.security.crypto.bcrypt.BCryptPasswordEncoder;

import org.springframework.util.PatternMatchUtils;

import com.ruoyi.common.constant.Constants;

import com.ruoyi.common.constant.HttpStatus;

import com.ruoyi.common.exception.CustomException;

import com.ruoyi.common.exception.ServiceException;

import com.ruoyi.framework.security.LoginUser;

import com.ruoyi.project.system.domain.SysRole;



/**

 * 安全服务工具类

@@ -14,6 +20,36 @@
 */

public class SecurityUtils

{

    /**

     * 用户ID

     **/

    public static Long getUserId()

    {

        try

        {

            return getLoginUser().getUserId();

        }

        catch (Exception e)

        {

            throw new ServiceException("获取用户ID异常", HttpStatus.UNAUTHORIZED);

        }

    }



    /**

     * 获取部门ID

     **/

    public static Long getDeptId()

    {

        try

        {

            return getLoginUser().getDeptId();

        }

        catch (Exception e)

        {

            throw new ServiceException("获取部门ID异常", HttpStatus.UNAUTHORIZED);

        }

    }



    /**

     * 获取用户账户

     **/

@@ -25,7 +61,7 @@
        }

        catch (Exception e)

        {

            throw new CustomException("获取用户账户异常", HttpStatus.UNAUTHORIZED);

            throw new ServiceException("获取用户账户异常", HttpStatus.UNAUTHORIZED);

        }

    }



@@ -40,7 +76,7 @@
        }

        catch (Exception e)

        {

            throw new CustomException("获取用户信息异常", HttpStatus.UNAUTHORIZED);

            throw new ServiceException("获取用户信息异常", HttpStatus.UNAUTHORIZED);

        }

    }



@@ -87,4 +123,54 @@
    {

        return userId != null && 1L == userId;

    }



    /**

     * 验证用户是否具备某权限

     * 
     * @param permission 权限字符串

     * @return 用户是否具备某权限

     */

    public static boolean hasPermi(String permission)

    {

        return hasPermi(getLoginUser().getPermissions(), permission);

    }



    /**

     * 判断是否包含权限

     * 
     * @param authorities 权限列表

     * @param permission 权限字符串

     * @return 用户是否具备某权限

     */

    public static boolean hasPermi(Collection<String> authorities, String permission)

    {

        return authorities.stream().filter(StringUtils::hasText)

                .anyMatch(x -> Constants.ALL_PERMISSION.equals(x) || PatternMatchUtils.simpleMatch(x, permission));

    }



    /**

     * 验证用户是否拥有某个角色

     * 
     * @param role 角色标识

     * @return 用户是否具备某角色

     */

    public static boolean hasRole(String role)

    {

        List<SysRole> roleList = getLoginUser().getUser().getRoles();

        Collection<String> roles = roleList.stream().map(SysRole::getRoleKey).collect(Collectors.toSet());

        return hasRole(roles, role);

    }



    /**

     * 判断是否包含角色

     * 
     * @param roles 角色列表

     * @param role 角色

     * @return 用户是否具备某角色权限

     */

    public static boolean hasRole(Collection<String> roles, String role)

    {

        return roles.stream().filter(StringUtils::hasText)

                .anyMatch(x -> Constants.SUPER_ADMIN.equals(x) || PatternMatchUtils.simpleMatch(x, role));

    }

}