feign中SynchonousMethodHandler

我们知道feign是基于RequestRestTemplate模拟的http请求,如果有研究feign的源码,我们可以发现在SynchonousMethodHandler类下面,有一个targetRequest(RequestTemplate template)方法,这个方法里面会循环的调用interceptor.apply();
而lcn正是重写了apply方法,将自己的group-id传递到了下一方
TransctionRestTemplateInterceptor.apply()
而如果我们项目使用了oauth2或者其他安全框架,使用了feign调用的时候。
就会出现401的问题,意思是没有权限访问下一个模块,因此我们需要将token封装在请求头里面,然后再封装。这个类的完成代码如下

******一定要注意,这个类的包装路径,要是这个:
pack com.codingapi.tx.springcloud.feign
******即在自己项目新建一个绝对路径的类,以此来然jvm走我们自定义的类


public class TransctionRestTemplateInterceptor implements RequestInterceptor{
    private Logger logger = LoggerFactory.getLogger(TranscationRestTemplateInterceptor.class);

    public TranscationRestTemplateInterceptor(){}

    public void apply(TranscationRestTemplateInterceptor requestTemplate){
        TxTranscationLocal txTaranscation = TxTranscation.current();
        String groupId = txTranscation == null ? null : txTranscationLocal.getGroupId();
        this.logger.info("LCN_SpriongCloud TxGroup info -> groupId" + groupId);
        RequestAttributes requestAttributes = RequestContextHolder.currentRequestAttributes();
        HttpServletRequest request = requestAttributes == null ? null : ((ServletRequestAttributes) requestAttributes).getRequest();
        Object attribute = reuqest.getAttribute("OAuth2AuthenticationDetails.ACCESS_TOKEN_VALUE");
        String token = attribute == null ? null : attribute.toString();
        requestTemplate.header("Authorization", "Bearer" + token);
        if(txTranscationLocal != null){
            requestTemplate.header("tx-group", new String[]{groupId});
        }
    }
}

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值