springCloud微服务系列——链路跟踪第三篇——feign链路跟踪器

目录

 

一、简介

二、思路

三、获取riboon负载均衡结果

  源码分析

  扩展点

四、示例代码

---

一、简介

   这篇文章总结feign链路跟踪器的实现

二、思路

   上篇文章中总结了mvc的链路跟踪器，我们可以知道要实现链路跟踪器需要在前后拦截请求，那么这里我们应该怎么处理呢？既然使用的是spring，那么很容易想到使用AOP来进行拦截。

   还有个细节需要注意，因为feign可能会调用hystrix，这个时候hystrixCommand会新产生一个线程，这时候需要通过http将链路信息传递过去。参看《springCloud微服务系列——OAuth2+JWT模式下的feign+hystrix处理》

@Slf4j
public class LuminaryRequestInterceptor implements RequestInterceptor {

	public static String TOKEN_HEADER = "authorization";
	
	/* (non-Javadoc)  
	 * <p>Title: apply</p>  
	 * <p>Description: </p>  
	 * @param arg0  
	 * @see feign.RequestInterceptor#apply(feign.RequestTemplate)  
	 */
	@Override
	public void apply(RequestTemplate template) {
		HttpServletRequest request = getHttpServletRequest();
		if(request == null) {
			log.warn("request is null in feign request interceptor");
			return;
		}
		Map<String, String> headerMap = getHeaders(request);
		if(headerMap.get(TOKEN_HEADER) != null) {
			template.header(TOKEN_HEADER, headerMap.get(TOKEN_HEADER));
		}
		if(request.getAttribute(TraceInfo.TRACE_ID_KEY) != null) {
			template.header(TraceInfo.TRACE_ID_KEY, (String) request.getAttribute(TraceInfo.TRACE_ID_KEY));
		}
		if(request.getAttribute(TraceInfo.RPC_ID_KEY) != null) {
			template.header(TraceInfo.RPC_ID_KEY, (String) request.getAttribute(TraceInfo.RPC_ID_KEY));
		}
	}
	
	private HttpServletRequest getHttpServletRequest() {
        try {
            return ((ServletRequestAttributes) RequestContextHolder.getRequestAttributes()).getRequest();
        } catch (Exception e) {
            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();
            String value = request.getHeader(key);
            map.put(key, value);
        }
        return map;
    }

}

三、获取riboon负载均衡结果

   feign链路跟踪的时候还有个获取负载均衡结果的问题，我们需要在跟踪的时候拿到具体调用的是哪个服务。feign的负载均衡使用的是riboon，在riboon中通过eureka拿到服务列表，通过ping的方式筛选出可用的服务。

  源码分析

 其中紫色表示类，绿色表示方法，蓝色表示内部方法

 我们可以看到最终是注册成了一个观察者，在特定的事件中通过loadBalancerContext的getServerFromLoadBalancer方法，loadBalancerContext中又通过ILoadBalancer调用chooseServer方法，最终通过IRule调用choose获得负载均衡结果。

LoadBalancerFeignClient

@Override
	public Response execute(Request request, Request.Options options) throws IOException {
		try {
			URI asUri = URI.create(request.url());
			String clientName = asUri.getHost();
			URI uriWithoutHost = cleanUrl(request.url(), clientName);
			FeignLoadBalancer.RibbonRequest ribbonRequest = new FeignLoadBalancer.RibbonRequest(
					this.delegate, request, uriWithoutHost);

			IClientConfig requestConfig = getClientConfig(options, clientName);
			return lbClient(clientName).executeWithLoadBalancer(ribbonRequest,
					requestConfig).toResponse();
		}
		catch (ClientException e) {
			IOException io = findIOException(e);
			if (io != null) {
				throw io;
			}
			throw new RuntimeException(e);
		}
	}

AbstractLoadBalancerAwareClient

public T executeWithLoadBalancer(final S request, final IClientConfig requestConfig) throws ClientException {
        RequestSpecificRetryHandler handler = getRequestSpecificRetryHan