Java Code Examples for io.opentracing.Span#context()

private SpanContext getActiveSpanContext() {
  if (activeSpanSource != null) {
    Span activeSpan = activeSpanSource.getActiveSpan();
    if (activeSpan != null) {
      return activeSpan.context();
    }
  }
  if (activeSpanContextSource != null) {
    final SpanContext spanContext = activeSpanContextSource.getActiveSpanContext();
    if (spanContext != null) {
      return spanContext;
    }
  }
  if (tracer.activeSpan() != null) {
    return tracer.activeSpan().context();
  }
  return null;
}
 

@Override
public void subscribe(final CoreSubscriber<? super Void> subscriber) {
    final Context context = subscriber.currentContext();
    final Span parentSpan = context.<Span>getOrEmpty(Span.class).orElseGet(tracer::activeSpan);
    final ServerHttpRequest request = exchange.getRequest();

    final SpanContext extractedContext;
    if (parentSpan != null) {
        extractedContext = parentSpan.context();
    } else {
        extractedContext = tracer.extract(Format.Builtin.HTTP_HEADERS, new HttpHeadersExtractAdapter(request.getHeaders()));
    }

    final Span span = tracer.buildSpan(request.getMethodValue())
            .asChildOf(extractedContext)
            .withTag(Tags.SPAN_KIND.getKey(), Tags.SPAN_KIND_SERVER)
            .start();

    try (final Scope scope = tracer.scopeManager().activate(span)) {
        exchange.getAttributes().put(TracingWebFilter.SERVER_SPAN_CONTEXT, span.context());
        source.subscribe(new TracingSubscriber(subscriber, exchange, context, span, spanDecorators));
    }
}
 

private CompletableFuture<ClientHttpResponse> inject(
        final RequestExecution execution,
        final RequestArguments arguments) throws IOException {

    @Nullable final Span span = arguments.getAttribute(internalSpan)
            .orElse(null);

    if (span == null) {
        return execution.execute(arguments);
    }

    decorator.onRequest(span, arguments);

    final SpanContext context = span.context();
    return execution.execute(injection.inject(tracer, arguments, context))
            .whenComplete(decorateOnResponse(span, arguments))
            .whenComplete(decorateOnError(span, arguments));
}
 

@Test
public void testFindPrimaryReferenceSingleChildOfSpanContextWithOtherRefs() {
    Tracer tracer = new APMTracer();

    SpanContext spanCtx1 = extractSpanContext(tracer, TEST_APM_ID1);

    Reference ref1 = new Reference(References.CHILD_OF, spanCtx1);

    Span span2 = tracer.buildSpan("test2").start();
    Reference ref2 = new Reference(References.FOLLOWS_FROM, span2.context());

    Span span3 = tracer.buildSpan("test3").start();
    Reference ref3 = new Reference(References.CHILD_OF, span3.context());
    Span span4 = tracer.buildSpan("test4").start();
    Reference ref4 = new Reference(References.CHILD_OF, span4.context());

    assertEquals(ref1, APMSpan.findPrimaryReference(Arrays.asList(ref1, ref2, ref3, ref4)));
}
 

@Override
public void subscribe(final CoreSubscriber<? super Void> subscriber) {
  final Context context = subscriber.currentContext();
  final Span parentSpan = context.<Span>getOrEmpty(Span.class).orElseGet(tracer::activeSpan);
  final ServerHttpRequest request = exchange.getRequest();

  final SpanContext extractedContext;
  if (parentSpan != null) {
    extractedContext = parentSpan.context();
  } else {
    extractedContext = tracer.extract(Format.Builtin.HTTP_HEADERS, new HttpHeadersExtractAdapter(request.getHeaders()));
  }

  final Span span = tracer.buildSpan(request.getMethodValue())
      .asChildOf(extractedContext)
      .withTag(Tags.SPAN_KIND.getKey(), Tags.SPAN_KIND_SERVER)
      .start();

  try (final Scope scope = tracer.scopeManager().activate(span, false)) {
    exchange.getAttributes().put(TracingWebFilter.SERVER_SPAN_CONTEXT, span.context());
    source.subscribe(new TracingSubscriber(subscriber, exchange, context, span, spanDecorators));
  }
}
 

@Test
public void testFindPrimaryReferenceMultipleChildOfSpan() {
    Tracer tracer = new APMTracer();
    Span span1 = tracer.buildSpan("test1").start();
    Reference ref1 = new Reference(References.CHILD_OF, span1.context());
    Span span2 = tracer.buildSpan("test2").start();
    Reference ref2 = new Reference(References.CHILD_OF, span2.context());
    assertNull(APMSpan.findPrimaryReference(Arrays.asList(ref1, ref2)));
}
 

public static SpanContext useOrCreateSpanContext(SpanContext spanContext, Span span) {
    LOGGER.debug("Creating a new span. {}", span.context());
    if (spanContext == null) {
        LOGGER.debug("Using the new span context. {}", span.context());
        spanContext = span.context();
    }
    return spanContext;
}
 

@Test
public void testInboundRequestWithNonNullTrace() {
    Trace trace = new Trace(42, 0, 42, (byte) 1);
    request.setTrace(trace);

    Span span = Tracing.startInboundSpan(request,tracer, new TracingContext.Default());
    JaegerSpanContext spanContext = (JaegerSpanContext) span.context();
    assertEquals(trace.traceId, spanContext.getTraceIdLow());
    assertEquals(trace.spanId, spanContext.getParentId());
    assertEquals(trace.traceFlags, spanContext.getFlags());
    assertNotEquals(trace.spanId, spanContext.getSpanId());
}
 

private static TraceResponse observeSpanAndDownstream(String encodings, Trace requestTrace) {
    Span span = tracingContext.currentSpan();
    TraceResponse response = new TraceResponse();
    JaegerSpanContext context = (JaegerSpanContext) span.context();
    response.requestTrace = requestTrace;
    response.traceId = context.toTraceId();
    response.sampled = context.isSampled();
    response.baggage = span.getBaggageItem(BAGGAGE_KEY);
    try {
        response.downstream = callDownstream(encodings);
    } catch (Exception e) {
        e.printStackTrace();
    }
    return response;
}
 

private SpanContext activeSpanContext() {
    Span span = activeSpan();
    if (span == null) {
        return null;
    }

    return span.context();
}
 

@Test
public void testFindPrimaryReferenceMultipleFollowsFrom() {
    Tracer tracer = new APMTracer();
    Span span1 = tracer.buildSpan("test1").start();
    Reference ref1 = new Reference(References.FOLLOWS_FROM, span1.context());
    Span span2 = tracer.buildSpan("test2").start();
    Reference ref2 = new Reference(References.FOLLOWS_FROM, span2.context());
    assertNull(APMSpan.findPrimaryReference(Arrays.asList(ref1, ref2)));
}
 

@Test
public void testFindPrimaryReferenceSingleFollowsFrom() {
    Tracer tracer = new APMTracer();
    Span span = tracer.buildSpan("test").start();

    Reference ref = new Reference(References.FOLLOWS_FROM, span.context());
    assertEquals(ref, APMSpan.findPrimaryReference(Arrays.asList(ref)));
}
 

@Test
public void testFindPrimaryReferenceSingleChildOfSpan() {
    Tracer tracer = new APMTracer();
    Span span = tracer.buildSpan("test").start();

    Reference ref = new Reference(References.CHILD_OF, span.context());
    assertEquals(ref, APMSpan.findPrimaryReference(Arrays.asList(ref)));
}
 

/**
 * Returns a parent for a span created by this filter (jax-rs span).
 * The context from the active span takes precedence over context in the request,
 * but only if joinExistingActiveSpan has been set.
 * The current active span should be child-of extracted context and for example
 * created at a lower level e.g. jersey filter.
 */
private SpanContext parentSpanContext(ContainerRequestContext requestContext) {
    Span activeSpan = tracer.activeSpan();
    if (activeSpan != null && this.joinExistingActiveSpan) {
        return activeSpan.context();
    } else {
        return tracer.extract(
                Format.Builtin.HTTP_HEADERS,
                new ServerHeadersExtractTextMap(requestContext.getHeaders())
        );
    }
}
 

/**
 * Helper method for accessing server span context associated with current request.
 *
 * @param routingContext routing context
 * @return server span context or null if not present
 */
public static SpanContext serverSpanContext(final RoutingContext routingContext) {
    SpanContext serverContext = null;

    final Object object = routingContext.get(CURRENT_SPAN);
    if (object instanceof Span) {
        final Span span = (Span) object;
        serverContext = span.context();
    } else {
        log.error("Sever SpanContext is null or not an instance of SpanContext");
    }

    return serverContext;
}
 

/**
 * Helper method for accessing server span context associated with current request.
 *
 * @param routingContext routing context
 * @return server span context or null if not present
 */
public static SpanContext serverSpanContext(RoutingContext routingContext) {
    SpanContext serverContext = null;

    Object object = routingContext.get(CURRENT_SPAN);
    if (object instanceof Span) {
        Span span = (Span) object;
        serverContext = span.context();
    } else {
        log.error("Sever SpanContext is null or not an instance of SpanContext");
    }

    return serverContext;
}
 

private void doFullRegisterOrderAndRemove() throws InterruptedException, ExecutionException {
	SpanBuilder spanBuilder = getTracer().buildSpan("fullSystemTest");
	final Span span = spanBuilder.start();
	try {
		SpanContext parentContext = span.context();

		// Register 
		CompletableFuture<Customer> newCustomer = CompletableFuture
				.supplyAsync(() -> registerRandomCustomer(parentContext));
		// Maybe not get the types and colors over and over. Looks pretty in the traces though...
		CompletableFuture<RobotType[]> availableTypes = CompletableFuture
				.supplyAsync(() -> getAllTypes(parentContext));
		CompletableFuture<Color[]> availableColors = CompletableFuture
				.supplyAsync(() -> getAllColors(parentContext));
		CompletableFuture.allOf(newCustomer, availableTypes, availableColors);

		Customer customer = newCustomer.get();

		// First completion stage done. Now we can create the order
		List<RobotOrderLineItem> lineItems = createRandomOrder(availableTypes.get(), availableColors.get());
		CompletableFuture<RobotOrder> robotOrderCompletable = CompletableFuture
				.supplyAsync(() -> postOrder(customer, lineItems, parentContext));

		// Rest will happen asynchrously when data is available...
		CompletableFuture<RealizedOrder> realizedOrderFuture = new CompletableFuture<RealizedOrder>();
		// When we have the order, we schedule the polling for an available order...
		robotOrderCompletable
				.thenAccept((order) -> awaitOrderCompletion(order, realizedOrderFuture, parentContext));
		// Once the order is realized, we will remove the customer.
		realizedOrderFuture.thenApply((realizedOrder) -> removeOwner(realizedOrder, parentContext))
				.thenAccept((customerId) -> span.finish());
	} catch (Throwable t) {
		span.log(OpenTracingUtil.getSpanLogMap(t));
		throw t;
	}
}
 

private SpanContext activeSpanContext() {
    Span span = activeSpan();
    if (span == null) {
        return null;
    }

    return span.context();
}