快捷導(dǎo)航

OpenFeign實(shí)現(xiàn)攜帶請(qǐng)求頭方案詳細(xì)介紹

更新時(shí)間：2022年11月18日 16:34:07 作者：hi_kong

這篇文章主要介紹了OpenFeign實(shí)現(xiàn)攜帶請(qǐng)求頭方案，在通過(guò)???OpenFeign???進(jìn)行服務(wù)調(diào)用的過(guò)程中，我們需要將用戶(hù)的??user-token???、??lang??等信息放入請(qǐng)求header中。在分布式系統(tǒng)中，往往一個(gè)業(yè)務(wù)接口內(nèi)部會(huì)發(fā)生多次RPC調(diào)用

1. 同步請(qǐng)求

對(duì)于同步請(qǐng)求，無(wú)需另作處理，只需從header中獲取token信息，放入新請(qǐng)求即可。

@Configuration
public class FeignConfig {
    @Bean
    public RequestInterceptor requestInterceptor(){
        return new RequestInterceptor() {
            @Override
            public void apply(RequestTemplate requestTemplate) {
                ServletRequestAttributes requestAttributes = (ServletRequestAttributes) RequestContextHolder.getRequestAttributes();
                if(requestAttributes != null){
                    //獲取請(qǐng)求的token信息
                    String token = requestAttributes.getRequest().getHeader(BaseConstant.TOKEN_HEADER);
                    //同步到請(qǐng)求中
                    requestTemplate.header(BaseConstant.TOKEN_HEADER,token);
                    return;
                }
            }
        };
    }
}

2. 異步請(qǐng)求

對(duì)于異步請(qǐng)求（例如A線(xiàn)程接到了HTTP請(qǐng)求，然后開(kāi)啟子線(xiàn)程B，B處理完成后調(diào)用openfeign接口），是無(wú)法使用上述方法的。因?yàn)镽equestContextHolder.getRequestAttributes()方法獲取的requestAttributesHolder變量，是ThreadLocal類(lèi)型的：

private static final ThreadLocal<RequestAttributes> requestAttributesHolder = new NamedThreadLocal("Request attributes");

所以想到了一個(gè)簡(jiǎn)單的辦法，在創(chuàng)建子線(xiàn)程時(shí)，將當(dāng)前線(xiàn)程的token信息傳遞到子線(xiàn)程中。子線(xiàn)程在調(diào)用feign接口前，將token存入當(dāng)前線(xiàn)程變量中，token類(lèi)定義以及ThreadLocal定義如下：

//Token類(lèi)定義
@Data
public class TokenInfo implements Serializable {
    private String token;
}
//TokenContext類(lèi)定義
public class TokenContext {
    public static final ThreadLocal<TokenInfo> tokenInfo = new ThreadLocal<>();
    //設(shè)置token信息
    public static void set(TokenInfo info){
        tokenInfo.set(info);
    }
    //獲取token信息
    public static TokenInfo get(){
        return tokenInfo.get();
    }
    //移除token信息
    public static void remove(){
        tokenInfo.remove();
    }
}

調(diào)用接口處理：

    TokenInfo info = new TokenInfo();
    info.setToken(token);
    TokenContext.set(info);
    //調(diào)用接口
    feign.interface;
    //一定要記得刪除，不然后內(nèi)存泄露風(fēng)險(xiǎn)
    TokenContext.remove();

最后，上述創(chuàng)建的Bean改為：

@Configuration
public class FeignConfig {
    @Bean
    public RequestInterceptor requestInterceptor(){
        return new RequestInterceptor() {
            @Override
            public void apply(RequestTemplate requestTemplate) {
                //老請(qǐng)求 獲取當(dāng)前線(xiàn)程請(qǐng)求的請(qǐng)求信息
                ServletRequestAttributes requestAttributes = (ServletRequestAttributes) RequestContextHolder.getRequestAttributes();
                if(requestAttributes != null){
                    String token = requestAttributes.getRequest().getHeader(BaseConstant.TOKEN_HEADER);
                    //同步到請(qǐng)求中
                    requestTemplate.header(BaseConstant.TOKEN_HEADER,token);
                    return;
                }
                TokenInfo tokenInfo = TokenContext.get();
                if(userInfo != null){
                    String token = tokenInfo.getToken();
                    requestTemplate.header(BaseConstant.TOKEN_HEADER,BaseConstant.TOKEN_PREFIX + token);
                    return;
                }
            }
        };
    }
}

到此這篇關(guān)于OpenFeign實(shí)現(xiàn)攜帶請(qǐng)求頭方案詳細(xì)介紹的文章就介紹到這了,更多相關(guān)OpenFeign攜帶請(qǐng)求頭內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: