我们知道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});
}
}
}