How to use

ClientResponseInterceptor

in

com.github.kristofa.brave

  @Override
  public void filter(ClientRequestContext clientRequestContext, ClientResponseContext clientResponseContext) throws IOException {

    final HttpResponse response = new JaxRs2HttpResponse(clientResponseContext);
    responseInterceptor.handle(new HttpClientResponseAdapter(response));
  }
}

serverResponseInterceptor = new ServerResponseInterceptor(serverTracer);
clientRequestInterceptor = new ClientRequestInterceptor(clientTracer);
clientResponseInterceptor = new ClientResponseInterceptor(clientTracer);

  private Brave(Builder builder) {
    serverTracer = ServerTracer.builder()
        .randomGenerator(builder.random)
        .spanCollector(builder.spanCollector)
        .state(builder.state)
        .traceSampler(builder.sampler).build();

    clientTracer = ClientTracer.builder()
        .randomGenerator(builder.random)
        .spanCollector(builder.spanCollector)
        .state(builder.state)
        .traceSampler(builder.sampler).build();

    localTracer = LocalTracer.builder()
        .randomGenerator(builder.random)
        .spanCollector(builder.spanCollector)
        .spanAndEndpoint(SpanAndEndpoint.LocalSpanAndEndpoint.create(builder.state))
        .traceSampler(builder.sampler).build();
    
    serverRequestInterceptor = new ServerRequestInterceptor(serverTracer);
    serverResponseInterceptor = new ServerResponseInterceptor(serverTracer);
    clientRequestInterceptor = new ClientRequestInterceptor(clientTracer);
    clientResponseInterceptor = new ClientResponseInterceptor(clientTracer);
    serverSpanAnnotationSubmitter = AnnotationSubmitter.create(SpanAndEndpoint.ServerSpanAndEndpoint.create(builder.state));
    serverSpanThreadBinder = new ServerSpanThreadBinder(builder.state);
    clientSpanThreadBinder = new ClientSpanThreadBinder(builder.state);
  }
}

synchronized void complete(HttpResponse response) {
 ClientResponseAdapter adapter = new HttpClientResponseAdapter(response);
 if (span != null) {
  spanThreadBinder.setCurrentSpan(span);
 }
 responseInterceptor.handle(adapter);
}

  @Override
  public Result invoke(Invoker<?> invoker, Invocation invocation) throws RpcException {
    clientRequestInterceptor.handle(new DubboClientRequestAdapter(invoker,invocation));
    try{
      Result rpcResult = invoker.invoke(invocation);
      clientResponseInterceptor.handle(new DubboClientResponseAdapter(rpcResult));
      return rpcResult;
    }catch (Exception ex){
      clientResponseInterceptor.handle(new DubboClientResponseAdapter(ex));
      throw  ex;
    }finally {
      clientSpanThreadBinder.setCurrentSpan(null);
    }
  }
}

public Result invoke(Invoker<?> invoker, Invocation invocation) throws RpcException {
  if ("com.alibaba.dubbo.monitor.MonitorService".equals(invoker.getInterface().getName())) {
    return invoker.invoke(invocation);
  }
  RpcContext context = RpcContext.getContext();
  clientRequestInterceptor.handle(new DubboClientRequestAdapter(context, invocation));
  Result result = invoker.invoke(invocation);
  clientResponseInterceptor.handle(new DubboClientResponseAdapter(result));
  return result;
}

public Result invoke(Invoker<?> arg0, Invocation arg1) throws RpcException {
  clientRequestInterceptor.handle(new GrpcClientRequestAdapter(arg1));
  Map<String,String> att = arg1.getAttachments();
  final Span currentClientSpan = clientSpanThreadBinder.getCurrentClientSpan();
  Result result ;
  try {
    result =  arg0.invoke(arg1);
    clientSpanThreadBinder.setCurrentSpan(currentClientSpan);
    clientResponseInterceptor.handle(new GrpcClientResponseAdapter(result));
  } finally {
    clientSpanThreadBinder.setCurrentSpan(null);
  }
  return result;
}

  @Override
  public Response execute(Request request, Request.Options options) throws IOException {
    Map<String, Collection<String>> traceHeaders = new HashMap<>();
    traceHeaders.putAll(request.headers());
    Request wrappedRequest =
        Request.create(request.method(), request.url(), traceHeaders, request.body(), request.charset());
    HttpClientRequest httpClientRequest = new TraceableHttpClientRequest(wrappedRequest);
    ClientRequestAdapter adapter = new HttpClientRequestAdapter(httpClientRequest, new DefaultSpanNameProvider());
    requestInterceptor.handle(adapter);

    Response response = clientDelegate.execute(wrappedRequest, options);

    HttpResponse httpResponse = new TraceableHttpClientResponse(response);
    ClientResponseAdapter responseAdapter = new HttpClientResponseAdapter(httpResponse);
    responseInterceptor.handle(responseAdapter);
    return response;
  }
}