FeignClient如何共用Header及踩坑過程記錄

2022-03-04 19:00:06

FeignClient共用Header及踩坑

問題

我們在呼叫feign的情況下，經常需要用到使用者端所持有的header比如說auth-token，我們需要將這些header傳遞下去

思路

使用攔截器，在使用者端攔截feign的請求，並把所需要的header傳遞下去

程式碼：

定義攔截器

@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

@Configuration
public class FeignSupportConfig {
    /**
     * feign請求攔截器
     *
     * @return
     */
    @Bean
    public RequestInterceptor requestInterceptor(){
        return new FeignInterceptor();
    }
}

說說坑吧

一定不能把所有的header都往下傳，這樣很可能會導致在呼叫feign的時候，請求型別混亂，應該需要哪些header就穿哪些header

Feign設定Header頭部

今天嘗試用feign去呼叫其他部門提供的一個HTTP介面，該介面要求在請求中設定一個username頭部，用於身份鑑權。

@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);
}

程式碼寫好好，嘗試去進行存取

丟擲如下錯誤資訊

{
    "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就是對方法設定的抽象。

    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的解析相關，於是通過檢視方法參照

我找到了下面的程式碼

    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需要用“=”進行分割。

於是我修改成下面的形式，

問題就解決了

   @GetMapping(value = "/api/clusterManager/listTopics",
        headers = {"cache-control=no-cache", "username=wangyong@xxx.com"})
    TopicsResponse listTopics(@RequestParam("clusterName") String clusterName,
                              @RequestParam("clusterArea") String clusterArea);

以上為個人經驗，希望能給大家一個參考，也希望大家多多支援it145.com。