Examples with SpanContext io.opentelemetry.api.trace.SpanContext used on opensource projects

Example 61 with SpanContext

use of io.opentelemetry.api.trace.SpanContext in project smallrye-reactive-messaging by smallrye.

the class TracingAmqpToAppToAmqpTest method testFromAmqpToAppToAmqp.

@Test
public void testFromAmqpToAppToAmqp() {
    List<Integer> payloads = new CopyOnWriteArrayList<>();
    List<Context> receivedContexts = new CopyOnWriteArrayList<>();
    usage.consumeIntegersWithTracing("result-topic", payloads::add, receivedContexts::add);
    weld.addBeanClass(MyAppProcessingData.class);
    new MapBasedConfig().put("mp.messaging.outgoing.result-topic.connector", AmqpConnector.CONNECTOR_NAME).put("mp.messaging.outgoing.result-topic.durable", false).put("mp.messaging.outgoing.result-topic.host", host).put("mp.messaging.outgoing.result-topic.port", port).put("mp.messaging.incoming.parent-topic.connector", AmqpConnector.CONNECTOR_NAME).put("mp.messaging.incoming.parent-topic.host", host).put("mp.messaging.incoming.parent-topic.port", port).put("amqp-username", username).put("amqp-password", password).write();
    container = weld.initialize();
    MyAppProcessingData bean = container.getBeanManager().createInstance().select(MyAppProcessingData.class).get();
    await().until(() -> isAmqpConnectorReady(container));
    AtomicInteger count = new AtomicInteger();
    List<SpanContext> producedSpanContexts = new CopyOnWriteArrayList<>();
    usage.produce("parent-topic", 10, () -> AmqpMessage.create().durable(false).ttl(10000).withIntegerAsBody(count.getAndIncrement()).applicationProperties(createTracingSpan(producedSpanContexts, "parent-topic")).build());
    await().atMost(Duration.ofMinutes(5)).until(() -> payloads.size() >= 10);
    assertThat(payloads).containsExactly(1, 2, 3, 4, 5, 6, 7, 8, 9, 10);
    List<String> producedTraceIds = producedSpanContexts.stream().map(SpanContext::getTraceId).collect(Collectors.toList());
    assertThat(producedTraceIds).hasSize(10);
    assertThat(receivedContexts).hasSize(10);
    assertThat(receivedContexts).doesNotContainNull().doesNotHaveDuplicates();
    List<String> receivedSpanIds = receivedContexts.stream().map(context -> Span.fromContextOrNull(context).getSpanContext().getSpanId()).collect(Collectors.toList());
    assertThat(receivedSpanIds).doesNotContainNull().doesNotHaveDuplicates().hasSize(10);
    List<String> receivedTraceIds = receivedContexts.stream().map(context -> Span.fromContextOrNull(context).getSpanContext().getTraceId()).collect(Collectors.toList());
    assertThat(receivedTraceIds).doesNotContainNull().doesNotHaveDuplicates().hasSize(10);
    assertThat(receivedTraceIds).containsExactlyInAnyOrderElementsOf(producedTraceIds);
    List<String> receivedParentSpanIds = new ArrayList<>();
    assertThat(bean.tracing()).hasSizeGreaterThanOrEqualTo(10);
    assertThat(bean.tracing()).doesNotContainNull().doesNotHaveDuplicates();
    List<String> spanIds = new ArrayList<>();
    for (TracingMetadata tracing : bean.tracing()) {
        Span span = Span.fromContext(tracing.getCurrentContext());
        spanIds.add(span.getSpanContext().getSpanId());
        assertThat(Span.fromContextOrNull(tracing.getPreviousContext())).isNotNull();
    }
    await().atMost(Duration.ofMinutes(2)).until(() -> testExporter.getFinishedSpanItems().size() >= 10);
    List<String> outgoingParentIds = new ArrayList<>();
    List<String> incomingParentIds = new ArrayList<>();
    for (SpanData data : testExporter.getFinishedSpanItems()) {
        if (data.getKind().equals(SpanKind.CONSUMER)) {
            incomingParentIds.add(data.getParentSpanId());
            // Need to skip the spans created during @Incoming processing
            continue;
        }
        assertThat(data.getSpanId()).isIn(receivedSpanIds);
        assertThat(data.getSpanId()).isNotEqualTo(data.getParentSpanId());
        assertThat(data.getTraceId()).isIn(producedTraceIds);
        assertThat(data.getKind()).isEqualByComparingTo(SpanKind.PRODUCER);
        outgoingParentIds.add(data.getParentSpanId());
    }
    // Assert span created on AMQP record is the parent of consumer span we create
    assertThat(producedSpanContexts.stream().map(SpanContext::getSpanId)).containsExactlyElementsOf(incomingParentIds);
    // Assert consumer span is the parent of the producer span we received in AMQP
    assertThat(spanIds.stream()).containsExactlyElementsOf(outgoingParentIds);
}

Example 62 with SpanContext

use of io.opentelemetry.api.trace.SpanContext in project smallrye-reactive-messaging by smallrye.

the class TracingAmqpToAppWithParentTest method createTracingSpan.

private JsonObject createTracingSpan(List<SpanContext> spanContexts, String topic) {
    Properties properties = new Properties();
    final Span span = AmqpConnector.TRACER.spanBuilder(topic).setSpanKind(SpanKind.PRODUCER).startSpan();
    final Context context = span.storeInContext(Context.current());
    GlobalOpenTelemetry.getPropagators().getTextMapPropagator().inject(context, properties, (headers, key, value) -> {
        if (headers != null) {
            headers.put(key, value.getBytes(StandardCharsets.UTF_8));
        }
    });
    spanContexts.add(span.getSpanContext());
    return JsonObject.mapFrom(properties);
}