Examples with SpanInScope brave.Tracer.SpanInScope used on opensource projects

Example 51 with SpanInScope

use of brave.Tracer.SpanInScope in project brave by openzipkin.

the class TracingMessageProducer method publish.

@Override
public void publish(Message message) throws JMSException {
    checkTopicPublisher();
    TopicPublisher tp = (TopicPublisher) delegate;
    Span span = createAndStartProducerSpan(message, destination(message));
    SpanInScope ws = tracer.withSpanInScope(span);
    Throwable error = null;
    try {
        tp.publish(message);
    } catch (Throwable t) {
        propagateIfFatal(t);
        error = t;
        throw t;
    } finally {
        if (error != null)
            span.error(error);
        span.finish();
        ws.close();
    }
}

Example 52 with SpanInScope

use of brave.Tracer.SpanInScope in project brave by openzipkin.

the class TracingMessageProducer method send.

@Override
public void send(Message message, int deliveryMode, int priority, long timeToLive) throws JMSException {
    Span span = createAndStartProducerSpan(message, destination(message));
    // animal-sniffer mistakes this for AutoCloseable
    SpanInScope ws = tracer.withSpanInScope(span);
    Throwable error = null;
    try {
        delegate.send(message, deliveryMode, priority, timeToLive);
    } catch (Throwable t) {
        propagateIfFatal(t);
        error = t;
        throw t;
    } finally {
        if (error != null)
            span.error(error);
        span.finish();
        ws.close();
    }
}

Example 53 with SpanInScope

use of brave.Tracer.SpanInScope in project brave by openzipkin.

the class TracingMessageProducer method send.

/* @Override JMS 2.0 method: Intentionally no override to ensure JMS 1.1 works! */
@JMS2_0
public void send(Destination destination, Message message, int deliveryMode, int priority, long timeToLive, CompletionListener completionListener) throws JMSException {
    Span span = createAndStartProducerSpan(message, destination);
    completionListener = TracingCompletionListener.create(completionListener, destination, span, current);
    SpanInScope ws = tracer.withSpanInScope(span);
    Throwable error = null;
    try {
        delegate.send(destination, message, deliveryMode, priority, timeToLive, completionListener);
    } catch (Throwable t) {
        propagateIfFatal(t);
        error = t;
        throw t;
    } finally {
        if (error != null)
            span.error(error).finish();
        ws.close();
    }
}

Example 54 with SpanInScope

use of brave.Tracer.SpanInScope in project java-chassis by ServiceComb.

the class ZipkinTracingAdviserTest method startsNewChildSpan.

@SuppressWarnings({ "unused", "try" })
@Test
public void startsNewChildSpan() {
    CyclicBarrier cyclicBarrier = new CyclicBarrier(nThreads);
    CompletableFuture<?>[] futures = (CompletableFuture<?>[]) Array.newInstance(CompletableFuture.class, nThreads);
    for (int i = 0; i < nThreads; i++) {
        futures[i] = CompletableFuture.runAsync(() -> {
            Span currentSpan = tracing.tracer().newTrace().start();
            waitTillAllAreReady(cyclicBarrier);
            try (SpanInScope spanInScope = tracing.tracer().withSpanInScope(currentSpan)) {
                assertThat(tracingAdviser.invoke(spanName, path, supplier), is(expected));
            } catch (Throwable throwable) {
                fail(throwable.getMessage());
            } finally {
                currentSpan.finish();
            }
        }, Executors.newFixedThreadPool(nThreads));
    }
    CompletableFuture.allOf(futures).join();
    assertThat(traces.size(), is(nThreads));
    for (Queue<zipkin2.Span> queue : traces.values()) {
        zipkin2.Span child = queue.poll();
        assertThat(child.name(), is(spanName));
        zipkin2.Span parent = queue.poll();
        assertThat(child.parentId(), is(parent.id()));
        assertThat(child.traceId(), is(parent.traceId()));
        assertThat(tracedValues(child), contains(this.getClass().getCanonicalName()));
    }
}

Example 55 with SpanInScope

use of brave.Tracer.SpanInScope in project java-chassis by ServiceComb.

the class ZipkinTracingHandler method handle.

@SuppressWarnings({ "try", "unused" })
@Override
public void handle(Invocation invocation, AsyncResponse asyncResp) throws Exception {
    Span span = tracingDelegate.createSpan(invocation);
    try (SpanInScope scope = tracer.tracer().withSpanInScope(span)) {
        LOGGER.debug("{}: Generated tracing span for {}", tracingDelegate.name(), invocation.getOperationName());
        invocation.next(onResponse(invocation, asyncResp, span));
    } catch (Exception e) {
        LOGGER.debug("{}: Failed invocation on {}", tracingDelegate.name(), invocation.getOperationName(), e);
        tracingDelegate.onResponse(span, null, e);
        throw e;
    }
}