brave.propagation.TraceContext.Extractor Java Examples

<R> TraceContextOrSamplingFlags extractAndClearTraceIdHeaders(
  Extractor<R> extractor, R request, Message message
) {
  TraceContextOrSamplingFlags extracted = extractor.extract(request);
  // Clear any propagation keys present in the headers
  if (extracted.samplingFlags() == null) { // then trace IDs were extracted
    MessageProperties properties = message.getMessageProperties();
    if (properties != null) clearTraceIdHeaders(properties.getHeaders());
  }
  return extracted;
}
 

/**
 * Extracts the underlying context using B3 encoding by default. Null is returned when there is no
 * encoded context in the carrier, or upon error extracting it.
 */
@Nullable @Override public <C> BraveSpanContext extract(Format<C> format, C carrier) {
  TraceContextOrSamplingFlags extractionResult;
  if (carrier instanceof BinaryExtract) {
    extractionResult = BinaryCodec.INSTANCE.extract((BinaryExtract) carrier);
  } else {
    Extractor<C> extractor = (Extractor<C>) formatToExtractor.get(format);
    if (extractor == null) {
      throw new UnsupportedOperationException(format + " not in " + formatToExtractor.keySet());
    }
    extractionResult = extractor.extract(carrier);
  }
  if (emptyExtractions.contains(extractionResult)) return null;
  return BraveSpanContext.create(extractionResult);
}
 

<R> TraceContextOrSamplingFlags extractAndClearTraceIdHeaders(
  Extractor<R> extractor, R request, Headers headers
) {
  TraceContextOrSamplingFlags extracted = extractor.extract(request);
  // Clear any propagation keys present in the headers
  if (extracted.samplingFlags() == null) { // then trace IDs were extracted
    clearTraceIdHeaders(headers);
  }
  return extracted;
}
 

<R> TraceContextOrSamplingFlags extractAndClearTraceIdProperties(
  Extractor<R> extractor, R request, Message message
) {
  TraceContextOrSamplingFlags extracted = extractor.extract(request);
  // Clear propagation regardless of extraction as JMS requires clearing as a means to make the
  // message writable
  PropertyFilter.filterProperties(message, traceIdProperties);
  return extracted;
}
 

TracingProducer(Extractor<R> extractor, Injector<R> injector, JmsTracing jmsTracing) {
  this.jmsTracing = jmsTracing;
  this.tracer = jmsTracing.tracing.tracer();
  this.current = jmsTracing.tracing.currentTraceContext();
  this.extractor = extractor;
  this.injector = injector;
  this.sampler = jmsTracing.producerSampler;
  this.remoteServiceName = jmsTracing.remoteServiceName;
}
 

/**
 * @since 4.3
 * @deprecated Since 5.7, use {@link #handleReceive(HttpServerRequest)}
 */
@Deprecated public <C> Span handleReceive(Extractor<C> extractor, C carrier, Req request) {
  if (carrier == null) throw new NullPointerException("request == null");

  HttpServerRequest serverRequest;
  if (request instanceof HttpServerRequest) {
    serverRequest = (HttpServerRequest) request;
  } else {
    serverRequest = new HttpServerAdapters.FromRequestAdapter<>(adapter, request);
  }

  Span span = nextSpan(extractor.extract(carrier), serverRequest);
  return handleStart(serverRequest, span);
}
 

@Override public <R> Extractor<R> extractor(Getter<R, K> getter) {
  // No check if Setter is a RemoteGetter because this instance cannot have String keys while
  // RemoteGetter must have String keys
  return delegate.extractor(new GetterAdapter<>(getter, map));
}
 

@Override public <R> Extractor<R> extractor(Getter<R, String> getter) {
  return new GrpcExtractor<>(this, getter);
}
 

@Override public <R> Extractor<R> extractor(Propagation.Getter<R, String> getter) {
  return delegate.extractor(new Getter<>(getter, customTraceIdName));
}
 

@Override
public <R> Extractor<R> extractor(Getter<R, String> getter) {
  if (getter == null) throw new NullPointerException("getter == null");
  return new AWSExtractor<>(this, getter);
}
 

@Override public <R> Extractor<R> extractor(Getter<R, K> getter) {
  return delegate.extractor(getter);
}
 

@Override public <R> Extractor<R> extractor(Getter<R, String> getter) {
  return new BaggageExtractor<>(this, getter);
}
 

@Override public <R> Extractor<R> extractor(Getter<R, String> getter) {
  if (getter == null) throw new NullPointerException("getter == null");
  return new B3Extractor<>(this, getter);
}
 

@Override public <R> Extractor<R> extractor(Getter<R, K> getter) {
  return delegate.extractor(getter);
}
 

@Override public <R> Extractor<R> extractor(Getter<R, String> getter) {
  if (getter == null) throw new NullPointerException("getter == null");
  return new XCloudTraceContextExtractor<>(primary, getter);
}
 

@Override public <C> Extractor<C> extractor(Getter<C, K> getter) {
  return delegate.extractor(getter);
}
 

/**
 * @since 4.3
 * @deprecated Since 5.7, use {@link #handleReceive(HttpServerRequest)}
 */
@Deprecated public Span handleReceive(Extractor<Req> extractor, Req request) {
  return handleReceive(extractor, request, request);
}