使用Feign動態設定header和原理分析

2022-03-04 19:01:12

Feign動態設定header和原理

專案中用到了Feign做遠端呼叫, 有部分場景需要動態設定header

開始的做法是通過 @RequestHeader 設定引數來實現動態的header設定

例如

@GetMapping(value = "/test", consumes = {MediaType.APPLICATION_JSON_UTF8_VALUE})    
String access(@RequestHeader("Auth") String auth, @RequestBody Expression expression);

這種方式雖然可以達到header的動態設定, 但是當引數過多時會降低介面可用性, 所以想通過傳遞bean的方式來設定header

先說解決辦法

public class HeaderInterceptor implements RequestInterceptor {
    @Override
    public void apply(RequestTemplate requestTemplate) {
        byte[] bytes = requestTemplate.requestBody().asBytes();
        Identity identity = JSONObject.parseObject(bytes, Identity.class);
        requestTemplate.header("Auth", identity.getSecret());
    }
} 
/**
 * configuration指定Interceptor
**/
@FeignClient(name = "test", url = "127.0.0.1:8300", configuration = HeaderInterceptor.class)
public interface GolangTestHandle2 {
    @GetMapping(value = "/handler", consumes = {MediaType.APPLICATION_JSON_UTF8_VALUE})
    String handle(Identity identity);
}

自定義Interceptor實現RequestInterceptor介面, 回撥方法apply提供了RequestTemplate物件, 物件內部封裝了request的所有資訊, 最後通過configuration指定介面, 之後就隨便你怎麼玩了(例如通過body獲取介面引數並動態設定header)

值得注意的一點是HeaderInterceptor如果注入到Springboot容器的話會全域性生效, 就是說及時沒有指定configuration也會對全域性feign介面生效, 為什麼呢? 這裡簡單說明一下

首先Feign為每個feign class建立springcontext上下文

spring通過呼叫getObject獲取feign工廠範例

    @Override
    public Object getObject() throws Exception {
        return getTarget();
    }

內部呼叫FeignClientFatoryBean.getTarget()方法

<T> T getTarget() {
        //獲取feign上下文
        FeignContext context = this.applicationContext.getBean(FeignContext.class);
        //構建feign Builder
        Feign.Builder builder = feign(context);
        ...
    }

根據feign(FeignContext context)構建Builder

protected Feign.Builder feign(FeignContext context) {
        FeignLoggerFactory loggerFactory = get(context, FeignLoggerFactory.class);
        Logger logger = loggerFactory.create(this.type);
        // @formatter:off
        Feign.Builder builder = get(context, Feign.Builder.class)
                // required values
                .logger(logger)
                //預設springEncoder
                .encoder(get(context, Encoder.class))
                //預設OptionalDecoder
                .decoder(get(context, Decoder.class))
                //預設SpringMvcContrat
                .contract(get(context, Contract.class));
        // @formatter:on
        //設定該feign的context
        configureFeign(context, builder);
        return builder;
    }

在構建過程中通過FeignClientFactoryBean.configureUsingConfiguration為feign class註冊基本的設定項, 其中也包括了Interceptor的註冊

    protected void configureUsingConfiguration(FeignContext context,
            Feign.Builder builder) {
        Logger.Level level = getOptional(context, Logger.Level.class);
        if (level != null) {
            builder.logLevel(level);
        }
        Retryer retryer = getOptional(context, Retryer.class);
        if (retryer != null) {
            builder.retryer(retryer);
        }
        ErrorDecoder errorDecoder = getOptional(context, ErrorDecoder.class);
        if (errorDecoder != null) {
            builder.errorDecoder(errorDecoder);
        }
        Request.Options options = getOptional(context, Request.Options.class);
        if (options != null) {
            builder.options(options);
        }
        //從feign context獲取interceptors
        Map<String, RequestInterceptor> requestInterceptors = context
                .getInstances(this.contextId, RequestInterceptor.class);
        if (requestInterceptors != null) {
            builder.requestInterceptors(requestInterceptors.values());
        }
        if (this.decode404) {
            builder.decode404();
        }
    }

contextId為具體的feign class id, RequestInterceptor為具體的介面, 即是說通過context.getInstances獲取所有RequestInterceptor範例並註冊到builder中.

    public <T> Map<String, T> getInstances(String name, Class<T> type) {
        AnnotationConfigApplicationContext context = getContext(name);
        //使用beanNamesForTypeIncludingAncestors
        if (BeanFactoryUtils.beanNamesForTypeIncludingAncestors(context,
                type).length > 0) {
            return BeanFactoryUtils.beansOfTypeIncludingAncestors(context, type);
        }
        return null;
    }

獲取工廠中的範例使用的是beanNamesForTypeIncludingAncestors方法, 該方法不僅會從feign的factory中查詢, 也會通過父級別spring工廠查詢相應範例(類似於springmvc的工廠)

也是因為該方法, 即使你沒有在FeignClient中設定configuration, 但是你的Interceptor通過@Component等方法注入容器的話也會全域性生效的, 所以如果指向讓你的Interceptor部分生效不讓它注入到Spring容器就好

設定Feign的header資訊(兩種形式)

在使用微服務SpringCloud全家桶元件Fegin的時候，我們在進行遠端服務之間呼叫的同時，為了防止使用者端劫持資訊，我們需要將一些敏感資訊新增到我們的Fegin頭部(Header)當中，今天朋友問起，總結一下：那麼工作中常見的方式有兩種

1.在方法引數前面新增@RequestHeader註解

@PostMapping(value = "/getPersonDetail") 
public ServerResponse getPersonDetail(@RequestBody Map map,@RequestHeader(name = "id") String id);

使用@RequestHeader(name = "id")可以傳遞動態header屬性

2.實現RequestInterceptor介面

設定Header(所有的Fegin請求)

import org.springframework.context.annotation.Configuration; 
import org.springframework.web.context.request.RequestContextHolder; 
import org.springframework.web.context.request.ServletRequestAttributes; 
import feign.RequestInterceptor; 
import feign.RequestTemplate; 
@Configuration 
public class FeignConfiguration implements RequestInterceptor {    
        @Override    
        public void apply(RequestTemplate template) {      
                ServletRequestAttributes attributes = (ServletRequestAttributes) RequestContextHolder.getRequestAttributes();       
                HttpServletRequest request = attributes.getRequest();        
                Enumeration<String> headerNames = request.getHeaderNames();        
                if (headerNames != null) {            
                        while (headerNames.hasMoreElements()) {             
                                String name = headerNames.nextElement();              
                                String values = request.getHeader(name);             
                                template.header(name, values);            
                        }            
                }    
        } 
} 
 
@Component
@FeignClient(value = "abc",fallback = abcServiceHystric.class ,configuration = FeignConfiguration.class) public interface AbcService { }

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