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

RuoYi

2020-11-30 3a9d45a7ddb9d82d961fb88c7ec1a14249455234

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
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); 
            } 
        } 
    } 
}