FeignClient共享Header及踩坑
问题
我们在调用feign的情况下,经常需要用到客户端所持有的header比如说auth-token,我们需要将这些header传递下去
思路
使用拦截器,在客户端拦截feign的请求,并把所需要的header传递下去
代码:
定义拦截器
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
|
@Configuration public class FeignInterceptor implements RequestInterceptor { @Override public void apply(RequestTemplate requestTemplate) { Map<String,String> headers = getHeaders(getHttpServletRequest()); for (String headerName : headers.keySet()){ requestTemplate.header(headerName, getHeaders(getHttpServletRequest()).get(headerName)); } } private HttpServletRequest getHttpServletRequest() { try { return ((ServletRequestAttributes) RequestContextHolder.getRequestAttributes()).getRequest(); } catch (Exception e) { e.printStackTrace(); return null ; } } private Map<String, String> getHeaders(HttpServletRequest request) { Map<String, String> map = new LinkedHashMap<>(); Enumeration<String> enumeration = request.getHeaderNames(); while (enumeration.hasMoreElements()) { String key = enumeration.nextElement(); if (needThisHeader(key)){ String value = request.getHeader(key); map.put(key, value); } } return map; } private boolean needThisHeader(String headerName){ //todo:这里写你的逻辑,哪些header需要传递,千万不能把所有的header传下去 return true ; } } |
将拦截器注入到Spring
1
2
3
4
5
6
7
8
9
10
11
12
|
@Configuration public class FeignSupportConfig { /** * feign请求拦截器 * * @return */ @Bean public RequestInterceptor requestInterceptor(){ return new FeignInterceptor(); } } |
说说坑吧
一定不能把所有的header都往下传,这样很可能会导致在调用feign的时候,请求类型混乱,应该需要哪些header就穿哪些header
Feign设置Header头部
今天尝试用feign去调用其他部门提供的一个HTTP接口,该接口要求在请求中设置一个username头部,用于身份鉴权。
1
2
3
4
5
6
7
|
@FeignClient (name = "kafka-client" , url = "http://kafka.xxx.com" ) public interface KafkaClient { @GetMapping (value = "/api/clusterManager/listTopics" ) @Headers ( { "cache-control: no-cache" , "username: wangyong@xxx.com" }) TopicsResponse listTopics( @RequestParam ( "clusterName" ) String clusterName, @RequestParam ( "clusterArea" ) String clusterArea); } |
代码写好好,尝试去进行访问
抛出如下错误信息
1
2
3
4
5
6
7
|
{ "timestamp" : 1551768926170 , "status" : 500 , "error" : "Internal Server Error" , "exception" : "com.netflix.zuul.exception.ZuulException" , "message" : "pre:AuthAccessFilter" } |
异常堆栈信息如下:
at feign.FeignException.errorStatus(FeignException.java:62)
at feign.codec.ErrorDecoder$Default.decode(ErrorDecoder.java:91)
at feign.SynchronousMethodHandler.executeAndDecode(SynchronousMethodHandler.java:134)
at feign.SynchronousMethodHandler.invoke(SynchronousMethodHandler.java:76)
at feign.ReflectiveFeign$FeignInvocationHandler.invoke(ReflectiveFeign.java:103)
at com.sun.proxy.$Proxy147.listTopics(Unknown Source)
at com.yidian.data.exptmgr.controller.KafkaController.getKafkaTopics(KafkaController.java:35)
// 省略更多异常堆栈信息
于是我尝试通过debug来找到问题的根源,通过debug我发现feign会先将请求参数构建成Request对象,request信息如下:
request
从图中可看到,该Request实例的headers属性为空,而Request又是根据RequestTemplate模板对象生成的,RequestTemplate实例信息如下:
RequestTemplate
到这里可以看出,问题是出在RequestTemplate的构建构成中,于是我就去跟踪RequestTemplate构建的代码,发现RequestTemplate是根据MethodMetadata构建而成,而MethodMetadata就是对方法配置的抽象。
1
2
3
4
5
6
7
8
9
|
RequestTemplate template = resolve(argv, mutable, varBuilder); if (metadata.queryMapIndex() != null ) { // add query map parameters after initial resolve so that they take // precedence over any predefined values template = addQueryMapQueryParameters(argv, template); } if (metadata.headerMapIndex() != null ) { template = addHeaderMapHeaders(argv, template); } |
从上述代码可以看到,header的设置是由metadata的headerMapIndex 属性决定的,那么,设置headerMapIndex的位置,必然就和Header的解析相关,于是通过查看方法引用
我找到了下面的代码
1
2
3
4
5
6
7
8
9
10
11
12
13
|
private void parseHeaders(MethodMetadata md, Method method, RequestMapping annotation) { // TODO: only supports one header value per key if (annotation.headers() != null && annotation.headers().length > 0 ) { for (String header : annotation.headers()) { int index = header.indexOf( '=' ); if (!header.contains( "!=" ) && index >= 0 ) { md.template().header(resolve(header.substring( 0 , index)), resolve(header.substring(index + 1 ).trim())); } } } } |
从代码中我们可以清晰的看到,解析过程中是从@RequestMapping或其派生注解的header属性中解析Header的,并且Header的key和value需要用“=”进行分割。
于是我修改成下面的形式,
问题就解决了
1
2
3
4
|
@GetMapping (value = "/api/clusterManager/listTopics" , headers = { "cache-control=no-cache" , "username=wangyong@xxx.com" }) TopicsResponse listTopics( @RequestParam ( "clusterName" ) String clusterName, @RequestParam ( "clusterArea" ) String clusterArea); |
以上为个人经验,希望能给大家一个参考,也希望大家多多支持服务器之家。
原文链接:https://blog.csdn.net/Leanky/article/details/104883622