服务之间feign调用传递用户请求头

RuoYi

2020-11-30 3a9d45a7ddb9d82d961fb88c7ec1a14249455234

服务之间feign调用传递用户请求头

 ruoyi-auth/src/main/java/com/ruoyi/auth/service/SysLoginService.java

@@ -7,8 +7,8 @@
import com.ruoyi.common.core.domain.R;
import com.ruoyi.common.core.enums.UserStatus;
import com.ruoyi.common.core.exception.BaseException;
import com.ruoyi.common.core.utils.SecurityUtils;
import com.ruoyi.common.core.utils.StringUtils;
import com.ruoyi.common.security.utils.SecurityUtils;
import com.ruoyi.system.api.RemoteLogService;
import com.ruoyi.system.api.RemoteUserService;
import com.ruoyi.system.api.domain.SysUser;

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

File was renamed from ruoyi-common/ruoyi-common-security/src/main/java/com/ruoyi/common/security/utils/SecurityUtils.java
@@ -1,66 +1,65 @@
package com.ruoyi.common.security.utils;



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

import com.ruoyi.common.core.constant.CacheConstants;

import com.ruoyi.common.core.text.Convert;

import com.ruoyi.common.core.utils.ServletUtils;



/**

 * 权限获取工具类

 * 
 * @author ruoyi

 */

public class SecurityUtils

{

    /**

     * 获取用户

     */

    public static String getUsername()

    {

        return ServletUtils.getRequest().getHeader(CacheConstants.DETAILS_USERNAME);

    }



    /**

     * 获取用户ID

     */

    public static Long getUserId()

    {

        return Convert.toLong(ServletUtils.getRequest().getHeader(CacheConstants.DETAILS_USER_ID));

    }



    /**

     * 是否为管理员

     * 
     * @param userId 用户ID

     * @return 结果

     */

    public static boolean isAdmin(Long userId)

    {

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

    }



    /**

     * 生成BCryptPasswordEncoder密码

     *

     * @param password 密码

     * @return 加密字符串

     */

    public static String encryptPassword(String password)

    {

        BCryptPasswordEncoder passwordEncoder = new BCryptPasswordEncoder();

        return passwordEncoder.encode(password);

    }



    /**

     * 判断密码是否相同

     *

     * @param rawPassword 真实密码

     * @param encodedPassword 加密后字符

     * @return 结果

     */

    public static boolean matchesPassword(String rawPassword, String encodedPassword)

    {

        BCryptPasswordEncoder passwordEncoder = new BCryptPasswordEncoder();

        return passwordEncoder.matches(rawPassword, encodedPassword);

    }

}

package com.ruoyi.common.core.utils;

import org.springframework.security.crypto.bcrypt.BCryptPasswordEncoder;
import com.ruoyi.common.core.constant.CacheConstants;
import com.ruoyi.common.core.text.Convert;

/**
 * 权限获取工具类
 * 
 * @author ruoyi
 */
public class SecurityUtils
{
    /**
     * 获取用户
     */
    public static String getUsername()
    {
        return ServletUtils.getRequest().getHeader(CacheConstants.DETAILS_USERNAME);
    }

    /**
     * 获取用户ID
     */
    public static Long getUserId()
    {
        return Convert.toLong(ServletUtils.getRequest().getHeader(CacheConstants.DETAILS_USER_ID));
    }

    /**
     * 是否为管理员
     * 
     * @param userId 用户ID
     * @return 结果
     */
    public static boolean isAdmin(Long userId)
    {
        return userId != null && 1L == userId;
    }

    /**
     * 生成BCryptPasswordEncoder密码
     *
     * @param password 密码
     * @return 加密字符串
     */
    public static String encryptPassword(String password)
    {
        BCryptPasswordEncoder passwordEncoder = new BCryptPasswordEncoder();
        return passwordEncoder.encode(password);
    }

    /**
     * 判断密码是否相同
     *
     * @param rawPassword 真实密码
     * @param encodedPassword 加密后字符
     * @return 结果
     */
    public static boolean matchesPassword(String rawPassword, String encodedPassword)
    {
        BCryptPasswordEncoder passwordEncoder = new BCryptPasswordEncoder();
        return passwordEncoder.matches(rawPassword, encodedPassword);
    }
}

 ruoyi-common/ruoyi-common-core/src/main/java/com/ruoyi/common/core/utils/ServletUtils.java

@@ -1,6 +1,9 @@
package com.ruoyi.common.core.utils;



import java.io.IOException;

import java.util.Enumeration;

import java.util.LinkedHashMap;

import java.util.Map;

import javax.servlet.http.HttpServletRequest;

import javax.servlet.http.HttpServletResponse;

import javax.servlet.http.HttpSession;

@@ -53,7 +56,14 @@
     */

    public static HttpServletRequest getRequest()

    {

        return getRequestAttributes().getRequest();

        try

        {

            return getRequestAttributes().getRequest();

        }

        catch (Exception e)

        {

            return null;

        }

    }



    /**

@@ -61,7 +71,14 @@
     */

    public static HttpServletResponse getResponse()

    {

        return getRequestAttributes().getResponse();

        try

        {

            return getRequestAttributes().getResponse();

        }

        catch (Exception e)

        {

            return null;

        }

    }



    /**

@@ -74,8 +91,31 @@


    public static ServletRequestAttributes getRequestAttributes()

    {

        RequestAttributes attributes = RequestContextHolder.getRequestAttributes();

        return (ServletRequestAttributes) attributes;

        try

        {

            RequestAttributes attributes = RequestContextHolder.getRequestAttributes();

            return (ServletRequestAttributes) attributes;

        }

        catch (Exception e)

        {

            return null;

        }

    }



    public static Map<String, String> getHeaders(HttpServletRequest request)

    {

        Map<String, String> map = new LinkedHashMap<>();

        Enumeration<String> enumeration = request.getHeaderNames();

        if (enumeration != null)

        {

            while (enumeration.hasMoreElements())

            {

                String key = enumeration.nextElement();

                String value = request.getHeader(key);

                map.put(key, value);

            }

        }

        return map;

    }



    /**


 ruoyi-common/ruoyi-common-security/src/main/java/com/ruoyi/common/security/annotation/EnableCustomConfig.java

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



import java.lang.annotation.*;

import java.lang.annotation.Documented;

import java.lang.annotation.ElementType;

import java.lang.annotation.Inherited;

import java.lang.annotation.Retention;

import java.lang.annotation.RetentionPolicy;

import java.lang.annotation.Target;

import org.mybatis.spring.annotation.MapperScan;

import org.springframework.context.annotation.EnableAspectJAutoProxy;

import org.springframework.context.annotation.Import;

import org.springframework.scheduling.annotation.EnableAsync;

import com.ruoyi.common.security.config.ApplicationConfig;

import com.ruoyi.common.security.feign.FeignAutoConfiguration;



@Target(ElementType.TYPE)

@Retention(RetentionPolicy.RUNTIME)

@@ -18,7 +24,7 @@
// 开启线程异步执行

@EnableAsync

// 自动加载类

@Import({ApplicationConfig.class})

@Import({ ApplicationConfig.class, FeignAutoConfiguration.class })

public @interface EnableCustomConfig

{




 ruoyi-common/ruoyi-common-security/src/main/java/com/ruoyi/common/security/feign/FeignAutoConfiguration.java

New file
@@ -0,0 +1,20 @@
package com.ruoyi.common.security.feign;



import org.springframework.context.annotation.Bean;

import org.springframework.context.annotation.Configuration;

import feign.RequestInterceptor;



/**

 * Feign 配置注册

 *

 * @author ruoyi

 **/

@Configuration

public class FeignAutoConfiguration

{

    @Bean

    public RequestInterceptor requestInterceptor()

    {

        return new FeignRequestInterceptor();

    }

}


 ruoyi-common/ruoyi-common-security/src/main/java/com/ruoyi/common/security/feign/FeignRequestInterceptor.java

New file
@@ -0,0 +1,40 @@
package com.ruoyi.common.security.feign;



import java.util.Map;

import javax.servlet.http.HttpServletRequest;

import org.springframework.stereotype.Component;

import com.ruoyi.common.core.constant.CacheConstants;

import com.ruoyi.common.core.utils.ServletUtils;

import com.ruoyi.common.core.utils.StringUtils;

import feign.RequestInterceptor;

import feign.RequestTemplate;



/**

 * feign 请求拦截器

 * 
 * @author ruoyi

 */

@Component

public class FeignRequestInterceptor implements RequestInterceptor

{

    @Override

    public void apply(RequestTemplate requestTemplate)

    {

        HttpServletRequest httpServletRequest = ServletUtils.getRequest();

        if (StringUtils.isNotNull(httpServletRequest))

        {

            Map<String, String> headers = ServletUtils.getHeaders(httpServletRequest);

            // 传递用户信息请求头，防止丢失

            String userId = headers.get(CacheConstants.DETAILS_USER_ID);

            if (StringUtils.isNotEmpty(userId))

            {

                requestTemplate.header(CacheConstants.DETAILS_USER_ID, userId);

            }

            String userName = headers.get(CacheConstants.DETAILS_USERNAME);

            if (StringUtils.isNotEmpty(userName))

            {

                requestTemplate.header(CacheConstants.DETAILS_USERNAME, userName);

            }

        }

    }

}

 ruoyi-modules/ruoyi-gen/src/main/java/com/ruoyi/gen/service/GenTableServiceImpl.java

@@ -25,9 +25,9 @@
import com.ruoyi.common.core.constant.GenConstants;

import com.ruoyi.common.core.exception.CustomException;

import com.ruoyi.common.core.text.CharsetKit;

import com.ruoyi.common.core.utils.SecurityUtils;

import com.ruoyi.common.core.utils.StringUtils;

import com.ruoyi.common.core.utils.file.FileUtils;

import com.ruoyi.common.security.utils.SecurityUtils;

import com.ruoyi.gen.domain.GenTable;

import com.ruoyi.gen.domain.GenTableColumn;

import com.ruoyi.gen.mapper.GenTableColumnMapper;


 ruoyi-modules/ruoyi-job/src/main/java/com/ruoyi/job/controller/SysJobController.java

@@ -14,6 +14,7 @@
import org.springframework.web.bind.annotation.RequestMapping;

import org.springframework.web.bind.annotation.RestController;

import com.ruoyi.common.core.exception.job.TaskException;

import com.ruoyi.common.core.utils.SecurityUtils;

import com.ruoyi.common.core.utils.poi.ExcelUtil;

import com.ruoyi.common.core.web.controller.BaseController;

import com.ruoyi.common.core.web.domain.AjaxResult;

@@ -21,7 +22,6 @@
import com.ruoyi.common.log.annotation.Log;

import com.ruoyi.common.log.enums.BusinessType;

import com.ruoyi.common.security.annotation.PreAuthorize;

import com.ruoyi.common.security.utils.SecurityUtils;

import com.ruoyi.job.domain.SysJob;

import com.ruoyi.job.service.ISysJobService;

import com.ruoyi.job.util.CronUtils;


 ruoyi-modules/ruoyi-system/src/main/java/com/ruoyi/system/controller/SysConfigController.java

@@ -14,6 +14,7 @@
import org.springframework.web.bind.annotation.RequestMapping;

import org.springframework.web.bind.annotation.RestController;

import com.ruoyi.common.core.constant.UserConstants;

import com.ruoyi.common.core.utils.SecurityUtils;

import com.ruoyi.common.core.utils.poi.ExcelUtil;

import com.ruoyi.common.core.web.controller.BaseController;

import com.ruoyi.common.core.web.domain.AjaxResult;

@@ -21,7 +22,6 @@
import com.ruoyi.common.log.annotation.Log;

import com.ruoyi.common.log.enums.BusinessType;

import com.ruoyi.common.security.annotation.PreAuthorize;

import com.ruoyi.common.security.utils.SecurityUtils;

import com.ruoyi.system.domain.SysConfig;

import com.ruoyi.system.service.ISysConfigService;




 ruoyi-modules/ruoyi-system/src/main/java/com/ruoyi/system/controller/SysDeptController.java

@@ -14,13 +14,13 @@
import org.springframework.web.bind.annotation.RequestMapping;

import org.springframework.web.bind.annotation.RestController;

import com.ruoyi.common.core.constant.UserConstants;

import com.ruoyi.common.core.utils.SecurityUtils;

import com.ruoyi.common.core.utils.StringUtils;

import com.ruoyi.common.core.web.controller.BaseController;

import com.ruoyi.common.core.web.domain.AjaxResult;

import com.ruoyi.common.log.annotation.Log;

import com.ruoyi.common.log.enums.BusinessType;

import com.ruoyi.common.security.annotation.PreAuthorize;

import com.ruoyi.common.security.utils.SecurityUtils;

import com.ruoyi.system.api.domain.SysDept;

import com.ruoyi.system.service.ISysDeptService;




 ruoyi-modules/ruoyi-system/src/main/java/com/ruoyi/system/controller/SysDictDataController.java

@@ -13,6 +13,7 @@
import org.springframework.web.bind.annotation.RequestBody;

import org.springframework.web.bind.annotation.RequestMapping;

import org.springframework.web.bind.annotation.RestController;

import com.ruoyi.common.core.utils.SecurityUtils;

import com.ruoyi.common.core.utils.poi.ExcelUtil;

import com.ruoyi.common.core.web.controller.BaseController;

import com.ruoyi.common.core.web.domain.AjaxResult;

@@ -20,7 +21,6 @@
import com.ruoyi.common.log.annotation.Log;

import com.ruoyi.common.log.enums.BusinessType;

import com.ruoyi.common.security.annotation.PreAuthorize;

import com.ruoyi.common.security.utils.SecurityUtils;

import com.ruoyi.system.domain.SysDictData;

import com.ruoyi.system.service.ISysDictDataService;

import com.ruoyi.system.service.ISysDictTypeService;


 ruoyi-modules/ruoyi-system/src/main/java/com/ruoyi/system/controller/SysDictTypeController.java

@@ -14,6 +14,7 @@
import org.springframework.web.bind.annotation.RequestMapping;

import org.springframework.web.bind.annotation.RestController;

import com.ruoyi.common.core.constant.UserConstants;

import com.ruoyi.common.core.utils.SecurityUtils;

import com.ruoyi.common.core.utils.poi.ExcelUtil;

import com.ruoyi.common.core.web.controller.BaseController;

import com.ruoyi.common.core.web.domain.AjaxResult;

@@ -21,7 +22,6 @@
import com.ruoyi.common.log.annotation.Log;

import com.ruoyi.common.log.enums.BusinessType;

import com.ruoyi.common.security.annotation.PreAuthorize;

import com.ruoyi.common.security.utils.SecurityUtils;

import com.ruoyi.system.domain.SysDictType;

import com.ruoyi.system.service.ISysDictTypeService;




 ruoyi-modules/ruoyi-system/src/main/java/com/ruoyi/system/controller/SysMenuController.java

@@ -13,13 +13,13 @@
import org.springframework.web.bind.annotation.RestController;

import com.ruoyi.common.core.constant.Constants;

import com.ruoyi.common.core.constant.UserConstants;

import com.ruoyi.common.core.utils.SecurityUtils;

import com.ruoyi.common.core.utils.StringUtils;

import com.ruoyi.common.core.web.controller.BaseController;

import com.ruoyi.common.core.web.domain.AjaxResult;

import com.ruoyi.common.log.annotation.Log;

import com.ruoyi.common.log.enums.BusinessType;

import com.ruoyi.common.security.annotation.PreAuthorize;

import com.ruoyi.common.security.utils.SecurityUtils;

import com.ruoyi.system.domain.SysMenu;

import com.ruoyi.system.service.ISysMenuService;




 ruoyi-modules/ruoyi-system/src/main/java/com/ruoyi/system/controller/SysNoticeController.java

@@ -11,13 +11,13 @@
import org.springframework.web.bind.annotation.RequestBody;

import org.springframework.web.bind.annotation.RequestMapping;

import org.springframework.web.bind.annotation.RestController;

import com.ruoyi.common.core.utils.SecurityUtils;

import com.ruoyi.common.core.web.controller.BaseController;

import com.ruoyi.common.core.web.domain.AjaxResult;

import com.ruoyi.common.core.web.page.TableDataInfo;

import com.ruoyi.common.log.annotation.Log;

import com.ruoyi.common.log.enums.BusinessType;

import com.ruoyi.common.security.annotation.PreAuthorize;

import com.ruoyi.common.security.utils.SecurityUtils;

import com.ruoyi.system.domain.SysNotice;

import com.ruoyi.system.service.ISysNoticeService;




 ruoyi-modules/ruoyi-system/src/main/java/com/ruoyi/system/controller/SysPostController.java

@@ -14,6 +14,7 @@
import org.springframework.web.bind.annotation.RequestMapping;

import org.springframework.web.bind.annotation.RestController;

import com.ruoyi.common.core.constant.UserConstants;

import com.ruoyi.common.core.utils.SecurityUtils;

import com.ruoyi.common.core.utils.poi.ExcelUtil;

import com.ruoyi.common.core.web.controller.BaseController;

import com.ruoyi.common.core.web.domain.AjaxResult;

@@ -21,7 +22,6 @@
import com.ruoyi.common.log.annotation.Log;

import com.ruoyi.common.log.enums.BusinessType;

import com.ruoyi.common.security.annotation.PreAuthorize;

import com.ruoyi.common.security.utils.SecurityUtils;

import com.ruoyi.system.domain.SysPost;

import com.ruoyi.system.service.ISysPostService;




 ruoyi-modules/ruoyi-system/src/main/java/com/ruoyi/system/controller/SysProfileController.java

@@ -11,6 +11,7 @@
import org.springframework.web.bind.annotation.RestController;

import org.springframework.web.multipart.MultipartFile;

import com.ruoyi.common.core.domain.R;

import com.ruoyi.common.core.utils.SecurityUtils;

import com.ruoyi.common.core.utils.ServletUtils;

import com.ruoyi.common.core.utils.StringUtils;

import com.ruoyi.common.core.web.controller.BaseController;

@@ -18,7 +19,6 @@
import com.ruoyi.common.log.annotation.Log;

import com.ruoyi.common.log.enums.BusinessType;

import com.ruoyi.common.security.service.TokenService;

import com.ruoyi.common.security.utils.SecurityUtils;

import com.ruoyi.system.api.RemoteFileService;

import com.ruoyi.system.api.domain.SysFile;

import com.ruoyi.system.api.domain.SysUser;


 ruoyi-modules/ruoyi-system/src/main/java/com/ruoyi/system/controller/SysRoleController.java

@@ -14,6 +14,7 @@
import org.springframework.web.bind.annotation.RequestMapping;

import org.springframework.web.bind.annotation.RestController;

import com.ruoyi.common.core.constant.UserConstants;

import com.ruoyi.common.core.utils.SecurityUtils;

import com.ruoyi.common.core.utils.poi.ExcelUtil;

import com.ruoyi.common.core.web.controller.BaseController;

import com.ruoyi.common.core.web.domain.AjaxResult;

@@ -21,7 +22,6 @@
import com.ruoyi.common.log.annotation.Log;

import com.ruoyi.common.log.enums.BusinessType;

import com.ruoyi.common.security.annotation.PreAuthorize;

import com.ruoyi.common.security.utils.SecurityUtils;

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

import com.ruoyi.system.service.ISysRoleService;




 ruoyi-modules/ruoyi-system/src/main/java/com/ruoyi/system/controller/SysUserController.java

@@ -18,6 +18,7 @@
import org.springframework.web.multipart.MultipartFile;

import com.ruoyi.common.core.constant.UserConstants;

import com.ruoyi.common.core.domain.R;

import com.ruoyi.common.core.utils.SecurityUtils;

import com.ruoyi.common.core.utils.StringUtils;

import com.ruoyi.common.core.utils.poi.ExcelUtil;

import com.ruoyi.common.core.web.controller.BaseController;

@@ -26,7 +27,6 @@
import com.ruoyi.common.log.annotation.Log;

import com.ruoyi.common.log.enums.BusinessType;

import com.ruoyi.common.security.annotation.PreAuthorize;

import com.ruoyi.common.security.utils.SecurityUtils;

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

import com.ruoyi.system.api.domain.SysUser;

import com.ruoyi.system.api.model.LoginUser;


 ruoyi-modules/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/SysMenuServiceImpl.java

@@ -11,8 +11,8 @@
import org.springframework.beans.factory.annotation.Autowired;

import org.springframework.stereotype.Service;

import com.ruoyi.common.core.constant.UserConstants;

import com.ruoyi.common.core.utils.SecurityUtils;

import com.ruoyi.common.core.utils.StringUtils;

import com.ruoyi.common.security.utils.SecurityUtils;

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

import com.ruoyi.system.api.domain.SysUser;

import com.ruoyi.system.domain.SysMenu;


 ruoyi-modules/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/SysUserServiceImpl.java

@@ -2,18 +2,16 @@


import java.util.ArrayList;

import java.util.List;



import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

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

import org.springframework.stereotype.Service;

import org.springframework.transaction.annotation.Transactional;



import com.ruoyi.common.core.constant.UserConstants;

import com.ruoyi.common.core.exception.CustomException;

import com.ruoyi.common.core.utils.SecurityUtils;

import com.ruoyi.common.core.utils.StringUtils;

import com.ruoyi.common.datascope.annotation.DataScope;

import com.ruoyi.common.security.utils.SecurityUtils;

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

import com.ruoyi.system.api.domain.SysUser;

import com.ruoyi.system.domain.SysPost;