io.opentracing.propagation.TextMapInjectAdapter Java Exaples

Source File: Client.java From opentelemetry-java with Apache License 2.0

6 votes

public void send() throws InterruptedException {
  Message message = new Message();

  Span span =
      tracer
          .buildSpan("send")
          .withTag(Tags.SPAN_KIND.getKey(), Tags.SPAN_KIND_CLIENT)
          .withTag(Tags.COMPONENT.getKey(), "example-client")
          .start();
  try (Scope scope = tracer.activateSpan(span)) {
    tracer.inject(span.context(), Builtin.TEXT_MAP_INJECT, new TextMapInjectAdapter(message));
    queue.put(message);
  } finally {
    span.finish();
  }
}

Source File: AsyncService.java From hawkular-apm with Apache License 2.0

6 votes

public void callService(Span span, Handler handler) {
    Span clientSpan = getTracer().buildSpan("Client")
            .withTag(Constants.ZIPKIN_BIN_ANNOTATION_HTTP_URL, "http://localhost:8080/outbound")
            .withTag(Constants.PROP_TRANSACTION_NAME, "AnotherTxnName")     // Should not overwrite the existing name
            .asChildOf(span).start();
    Message mesg = createMessage();
    getTracer().inject(clientSpan.context(), Format.Builtin.TEXT_MAP,
            new TextMapInjectAdapter(mesg.getHeaders()));

    ExecutorService executor = Executors.newSingleThreadExecutor();
    executor.submit(() -> {
        delay(500);

        // Explicit finish
        clientSpan.finish();

        handler.handle("My Response");
    });
}

Source File: SpanSubscriber.java From rsocket-rpc-java with Apache License 2.0

5 votes

SpanSubscriber(
    Subscriber<? super T> subscriber,
    Context ctx,
    Tracer tracer,
    Map<String, String> tracingMetadata,
    SpanContext spanContext,
    String name,
    Tag... tags) {
  this.subscriber = subscriber;
  this.tracer = tracer;
  this.rootSpan = null;

  Tracer.SpanBuilder spanBuilder = this.tracer.buildSpan(name).asChildOf(spanContext);
  if (tags != null && tags.length > 0) {
    for (Tag tag : tags) {
      spanBuilder.withTag(tag.getKey(), tag.getValue());
    }
  }

  this.span = spanBuilder.start();

  if (tracingMetadata != null) {
    TextMapInjectAdapter adapter = new TextMapInjectAdapter(tracingMetadata);
    tracer.inject(span.context(), Format.Builtin.TEXT_MAP, adapter);
  }

  if (log.isTraceEnabled()) {
    log.trace(
        "Created span [{}], with name [{}], child of [{}]",
        this.span,
        name,
        spanContext.toString());
  }

  this.context = ctx.put(Span.class, this.span);
}

Source File: SpanSubscriber.java From rsocket-rpc-java with Apache License 2.0

5 votes

SpanSubscriber(
    Subscriber<? super T> subscriber,
    Context ctx,
    Tracer tracer,
    Map<String, String> tracingMetadata,
    SpanContext spanContext,
    String name,
    Tag... tags) {
  this.subscriber = subscriber;
  this.tracer = tracer;
  this.rootSpan = null;

  Tracer.SpanBuilder spanBuilder = this.tracer.buildSpan(name).asChildOf(spanContext);
  if (tags != null && tags.length > 0) {
    for (Tag tag : tags) {
      spanBuilder.withTag(tag.getKey(), tag.getValue());
    }
  }

  this.span = spanBuilder.start();

  if (tracingMetadata != null) {
    TextMapInjectAdapter adapter = new TextMapInjectAdapter(tracingMetadata);
    tracer.inject(span.context(), Format.Builtin.TEXT_MAP, adapter);
  }

  if (log.isTraceEnabled()) {
    log.trace(
        "Created span [{}], with name [{}], child of [{}]",
        this.span,
        name,
        spanContext.toString());
  }

  this.context = ctx.put(Span.class, this.span);
}

Source File: SyncService.java From hawkular-apm with Apache License 2.0

5 votes

public void callService(Span span) {
    try (Span clientSpan = getTracer().buildSpan("Client")
            .withTag(Constants.ZIPKIN_BIN_ANNOTATION_HTTP_URL, "http://localhost:8080/outbound")
            .withTag(Constants.PROP_TRANSACTION_NAME, SYNC_TXN_NAME_2)
            .asChildOf(span).start()) {
        Message mesg = createMessage();
        getTracer().inject(clientSpan.context(), Format.Builtin.TEXT_MAP,
                new TextMapInjectAdapter(mesg.getHeaders()));

        delay(500);

        // Explicit finish
        clientSpan.finish();
    }
}

Source File: SpawnService.java From hawkular-apm with Apache License 2.0

5 votes

public void callService(Span span) {
    try (Span clientSpan = getTracer().buildSpan("Client")
            .withTag(Constants.ZIPKIN_BIN_ANNOTATION_HTTP_URL, "http://localhost:8080/outbound")
            .withTag(Constants.PROP_TRANSACTION_NAME, "AnotherTxnName")     // Should not overwrite the existing name
            .asChildOf(span).start()) {
        Message mesg = createMessage();
        getTracer().inject(clientSpan.context(), Format.Builtin.TEXT_MAP,
                new TextMapInjectAdapter(mesg.getHeaders()));

        delay(500);

        // Explicit finish
        clientSpan.finish();
    }
}

Source File: ClientService.java From hawkular-apm with Apache License 2.0

5 votes

public void callService(Span span) {
    try (Span clientSpan = getTracer().buildSpan("Client")
            .withTag(Constants.ZIPKIN_BIN_ANNOTATION_HTTP_URL, "http://localhost:8080/outbound")
            .withTag("myTag", myTag)
            .asChildOf(span).start()) {
        Message mesg = createMessage();
        getTracer().inject(clientSpan.context(), Format.Builtin.TEXT_MAP,
                new TextMapInjectAdapter(mesg.getHeaders()));

        delay(500);
    }
}

Source File: MockTracerTest.java From opentracing-java with Apache License 2.0

5 votes

@Test
public void testTextMapPropagatorTextMap() {
    MockTracer tracer = new MockTracer(MockTracer.Propagator.TEXT_MAP);
    HashMap<String, String> injectMap = new HashMap<>();
    injectMap.put("foobag", "donttouch");
    {
        Span parentSpan = tracer.buildSpan("foo")
                .start();
        parentSpan.setBaggageItem("foobag", "fooitem");
        parentSpan.finish();

        tracer.inject(parentSpan.context(), Format.Builtin.TEXT_MAP_INJECT,
                new TextMapInjectAdapter(injectMap));

        SpanContext extract = tracer.extract(Format.Builtin.TEXT_MAP_EXTRACT, new TextMapExtractAdapter(injectMap));

        Span childSpan = tracer.buildSpan("bar")
                .asChildOf(extract)
                .start();
        childSpan.setBaggageItem("barbag", "baritem");
        childSpan.finish();
    }
    List<MockSpan> finishedSpans = tracer.finishedSpans();

    Assert.assertEquals(2, finishedSpans.size());
    Assert.assertEquals(finishedSpans.get(0).context().traceId(), finishedSpans.get(1).context().traceId());
    Assert.assertEquals(finishedSpans.get(0).context().spanId(), finishedSpans.get(1).parentId());
    Assert.assertEquals("fooitem", finishedSpans.get(0).getBaggageItem("foobag"));
    Assert.assertNull(finishedSpans.get(0).getBaggageItem("barbag"));
    Assert.assertEquals("fooitem", finishedSpans.get(1).getBaggageItem("foobag"));
    Assert.assertEquals("baritem", finishedSpans.get(1).getBaggageItem("barbag"));
    Assert.assertEquals("donttouch", injectMap.get("foobag"));
}

Source File: Client.java From opentracing-java with Apache License 2.0

5 votes

public void send() throws InterruptedException {
    Message message = new Message();

    Span span = tracer.buildSpan("send")
        .withTag(Tags.SPAN_KIND.getKey(), Tags.SPAN_KIND_CLIENT)
        .withTag(Tags.COMPONENT.getKey(), "example-client")
        .start();
    try (Scope scope = tracer.activateSpan(span)) {
        tracer.inject(span.context(), Builtin.TEXT_MAP_INJECT, new TextMapInjectAdapter(message));
        queue.put(message);
    } finally {
        span.finish();
    }
}

Source File: Util.java From carbon-apimgt with Apache License 2.0

5 votes

/**
 * Inject tracer specific information to tracerSpecificCarrier
 *
 * @param span
 * @param tracer
 * @param tracerSpecificCarrier
 * */
public static void inject(TracingSpan span, TracingTracer tracer, Map<String, String> tracerSpecificCarrier) {
    Object sp = span.getSpan();
    if (sp instanceof Span) {
        tracer.getTracingTracer().inject(((Span) sp).context(), Format.Builtin.HTTP_HEADERS,
                new TextMapInjectAdapter(tracerSpecificCarrier));
    } else if (sp instanceof SpanContext) {
        tracer.getTracingTracer().inject((SpanContext) sp, Format.Builtin.HTTP_HEADERS,
                new TextMapInjectAdapter(tracerSpecificCarrier));
    }
}

Source File: TraceClientFilter.java From TarsJava with BSD 3-Clause "New" or "Revised" License

4 votes

@Override
public void doFilter(Request request, Response response, FilterChain chain) throws Throwable {
	if (!isTrace) {
		chain.doFilter(request, response);
		return;
	}
	if (!(request instanceof TarsServantRequest) || !TraceUtil.checkServant(((TarsServantRequest)request).getServantName())) {
		chain.doFilter(request, response);
		return;
	}
	ServerConfig config = ConfigurationManager.getInstance().getServerConfig();
	DistributedContext context = DistributedContextManager.getDistributedContext();
	String servantName = context.get(TraceManager.INTERNAL_SERVANT_NAME);
	Tracer tracer = TraceContext.getInstance().getCurrentTracer();
	
       if (tracer == null) {
       	chain.doFilter(request, response);
       } else {
       	TarsServantRequest tarsServantRequest = (TarsServantRequest)request;
       	boolean isSync = tarsServantRequest.getInvokeStatus() == InvokeStatus.SYNC_CALL 
       			|| tarsServantRequest.getInvokeStatus() == InvokeStatus.FUTURE_CALL;
       	
       	String protocol = Constants.TARS_PROTOCOL;
       	Map<String, String> requestContext = tarsServantRequest.getContext();
       	if (requestContext != null && !requestContext.isEmpty()) {
       		protocol = requestContext.get(TraceManager.PROTOCOL);
       		requestContext.remove(TraceManager.PROTOCOL);
       	}
       	try(Scope scope = tracer.buildSpan(tarsServantRequest.getFunctionName()).withTag(Tags.SPAN_KIND.getKey(), Tags.SPAN_KIND_CLIENT).startActive(isSync)) {
       		Map<String, String> status = tarsServantRequest.getStatus();
       		if (status == null) {
       			tarsServantRequest.setStatus(new HashMap<String, String>());
       			status = tarsServantRequest.getStatus();
       		}
       		tracer.inject(scope.span().context(), Format.Builtin.TEXT_MAP, new TextMapInjectAdapter(status));
       		scope.span().setTag("client.ipv4", config.getLocalIP());
       		scope.span().setTag("client.port", config.getServantAdapterConfMap().get(servantName).getEndpoint().port());
       		scope.span().setTag("tars.interface", getObjName(servantName));
       		scope.span().setTag("tars.method", tarsServantRequest.getFunctionName());
       		scope.span().setTag("tars.protocol", protocol);
       		scope.span().setTag("tars.client.version", ClientVersion.getVersion());
       		
       		
       		TarsServantResponse tarsServantResponse = (TarsServantResponse)response;
       		try {
       			chain.doFilter(request, response);
       			if (isSync) {
           			scope.span().setTag("tars.retcode", Integer.toString(tarsServantResponse.getRet()));
           		} else {
           			TraceManager.getInstance().putSpan(request.getTicketNumber(), tracer, scope.span());
           		}
       		} catch (Exception e) {
       			scope.span().log(e.getMessage());
       			throw e;
       		}
       		
       		
       	}
       }
}

Source File: SpanSubscriber.java From rsocket-rpc-java with Apache License 2.0

4 votes

SpanSubscriber(
    Subscriber<? super T> subscriber,
    Context ctx,
    Tracer tracer,
    Map<String, String> tracingMetadata,
    String name,
    Tag... tags) {
  this.subscriber = subscriber;
  this.tracer = tracer;
  Span root = ctx.getOrDefault(Span.class, this.tracer.activeSpan());
  if (log.isTraceEnabled()) {
    log.trace("Span from context [{}]", root);
  }
  this.rootSpan = root;
  if (log.isTraceEnabled()) {
    log.trace("Stored context root span [{}]", this.rootSpan);
  }

  Tracer.SpanBuilder spanBuilder = this.tracer.buildSpan(name);
  if (tags != null && tags.length > 0) {
    for (Tag tag : tags) {
      spanBuilder.withTag(tag.getKey(), tag.getValue());
    }
  }

  if (root != null) {
    spanBuilder.asChildOf(root);
  }

  this.span = spanBuilder.start();

  if (tracingMetadata != null) {
    TextMapInjectAdapter adapter = new TextMapInjectAdapter(tracingMetadata);
    tracer.inject(span.context(), Format.Builtin.TEXT_MAP, adapter);
  }

  if (log.isTraceEnabled()) {
    log.trace("Created span [{}], with name [{}]", this.span, name);
  }
  this.context = ctx.put(Span.class, this.span);
}

Source File: SpanSubscriber.java From rsocket-rpc-java with Apache License 2.0

4 votes

SpanSubscriber(
    Subscriber<? super T> subscriber,
    Context ctx,
    Tracer tracer,
    Map<String, String> tracingMetadata,
    String name,
    Tag... tags) {
  this.subscriber = subscriber;
  this.tracer = tracer;
  Span root = ctx.getOrDefault(Span.class, this.tracer.activeSpan());
  if (log.isTraceEnabled()) {
    log.trace("Span from context [{}]", root);
  }
  this.rootSpan = root;
  if (log.isTraceEnabled()) {
    log.trace("Stored context root span [{}]", this.rootSpan);
  }

  Tracer.SpanBuilder spanBuilder = this.tracer.buildSpan(name);
  if (tags != null && tags.length > 0) {
    for (Tag tag : tags) {
      spanBuilder.withTag(tag.getKey(), tag.getValue());
    }
  }

  if (root != null) {
    spanBuilder.asChildOf(root);
  }

  this.span = spanBuilder.start();

  if (tracingMetadata != null) {
    TextMapInjectAdapter adapter = new TextMapInjectAdapter(tracingMetadata);
    tracer.inject(span.context(), Format.Builtin.TEXT_MAP, adapter);
  }

  if (log.isTraceEnabled()) {
    log.trace("Created span [{}], with name [{}]", this.span, name);
  }
  this.context = ctx.put(Span.class, this.span);
}

Source File: SamplingTest.java From hawkular-apm with Apache License 2.0

4 votes

@Test
public void testSamplingPriorityChangedToZero() {
    APMTracerTest.TestTraceRecorder traceRecorder = new APMTracerTest.TestTraceRecorder();
    Tracer tracer = new APMTracer(traceRecorder, Sampler.ALWAYS_SAMPLE);

    Span rootSpan = tracer.buildSpan("foo")
            .asChildOf(extractedTraceState(tracer, ReportingLevel.All))
            .start();

    Span descendant = tracer.buildSpan("foo")
            .asChildOf(rootSpan)
            .start();

    Map<String, String> carrier = new HashMap<>();
    tracer.inject(descendant.context(), Format.Builtin.TEXT_MAP, new TextMapInjectAdapter(carrier));
    Assert.assertEquals(ReportingLevel.All.name(), carrier.get(Constants.HAWKULAR_APM_LEVEL));

    Span descendantZeroSampling = tracer.buildSpan("foo")
            .asChildOf(rootSpan)
            .start();

    descendantZeroSampling.setTag(Tags.SAMPLING_PRIORITY.getKey(), 0);
    carrier.clear();
    tracer.inject(descendantZeroSampling.context(), Format.Builtin.TEXT_MAP, new TextMapInjectAdapter(carrier));
    Assert.assertEquals(ReportingLevel.None.name(), carrier.get(Constants.HAWKULAR_APM_LEVEL));

    descendantZeroSampling.finish();
    descendant.finish();
    rootSpan.finish();

    Assert.assertEquals(1, traceRecorder.getTraces().size());
    traceRecorder.clear();

    Span descendantDescendantZeroSampling = tracer.buildSpan("foo")
            .addReference(References.FOLLOWS_FROM, descendantZeroSampling.context())
            .start();

    carrier.clear();
    tracer.inject(descendantDescendantZeroSampling.context(), Format.Builtin.TEXT_MAP, new TextMapInjectAdapter(carrier));
    Assert.assertEquals(ReportingLevel.None.name(), carrier.get(Constants.HAWKULAR_APM_LEVEL));

    descendantDescendantZeroSampling.finish();
    Assert.assertEquals(0, traceRecorder.getTraces().size());
}

Source File: SamplingTest.java From hawkular-apm with Apache License 2.0

4 votes

@Test
public void testSamplingPriorityChangedToOne() {
    APMTracerTest.TestTraceRecorder traceRecorder = new APMTracerTest.TestTraceRecorder();
    Tracer tracer = new APMTracer(traceRecorder, Sampler.ALWAYS_SAMPLE);

    Span rootSpan = tracer.buildSpan("foo")
            .asChildOf(extractedTraceState(tracer, ReportingLevel.None))
            .start();

    Span descendant = tracer.buildSpan("foo")
            .asChildOf(rootSpan)
            .start();

    Map<String, String> carrier = new HashMap<>();
    tracer.inject(descendant.context(), Format.Builtin.TEXT_MAP, new TextMapInjectAdapter(carrier));
    Assert.assertEquals(ReportingLevel.None.name(), carrier.get(Constants.HAWKULAR_APM_LEVEL));

    Span descendantOneSampling = tracer.buildSpan("foo")
            .asChildOf(rootSpan)
            .start();

    descendantOneSampling.setTag(Tags.SAMPLING_PRIORITY.getKey(), 1);
    carrier.clear();
    tracer.inject(descendantOneSampling.context(), Format.Builtin.TEXT_MAP, new TextMapInjectAdapter(carrier));
    Assert.assertEquals(ReportingLevel.All.name(), carrier.get(Constants.HAWKULAR_APM_LEVEL));

    descendantOneSampling.finish();
    descendant.finish();
    rootSpan.finish();

    Span descendantDescendantOneSampling = tracer.buildSpan("foo")
            .addReference(References.FOLLOWS_FROM, descendantOneSampling.context())
            .start();

    carrier.clear();
    tracer.inject(descendantDescendantOneSampling.context(), Format.Builtin.TEXT_MAP, new TextMapInjectAdapter(carrier));
    Assert.assertEquals(ReportingLevel.All.name(), carrier.get(Constants.HAWKULAR_APM_LEVEL));

    descendantDescendantOneSampling.finish();
    Assert.assertEquals(2, traceRecorder.getTraces().size());
}

Source File: TracingFilter.java From nakadi with MIT License

4 votes

@Override
protected void doFilterInternal(final HttpServletRequest request,
                                final HttpServletResponse response, final FilterChain filterChain)
        throws IOException, ServletException {
    final Long startTime = System.currentTimeMillis();
    final Map<String, String> requestHeaders = Collections.list(request.getHeaderNames())
            .stream()
            .collect(Collectors.toMap(h -> h, request::getHeader));

    final SpanContext spanContext = GlobalTracer.get()
            .extract(HTTP_HEADERS, new TextMapExtractAdapter(requestHeaders));
    final Span baseSpan;
    if (spanContext != null) {
        if (isCommitRequest(request.getRequestURI(), request.getMethod())) {
            baseSpan = TracingService.getNewSpanWithReference("commit_events",
                    startTime, spanContext);
        } else {
            baseSpan = TracingService.getNewSpanWithParent("all_requests",
                    startTime, spanContext);
        }
    } else {
        baseSpan = TracingService.getNewSpan("all_requests", startTime);
    }

    try {
        baseSpan
                .setTag("client_id", authorizationService.getSubject().map(Subject::getName).orElse("-"))
                .setTag("http.url", request.getRequestURI() +
                        Optional.ofNullable(request.getQueryString()).map(q -> "?" + q).orElse(""))
                .setTag("http.header.content_encoding",
                        Optional.ofNullable(request.getQueryString()).map(q -> "?" + q).orElse(""))
                .setTag("http.header.accept_encoding",
                        Optional.ofNullable(request.getQueryString()).map(q -> "?" + q).orElse(""))
                .setTag("http.header.user_agent",
                        Optional.ofNullable(request.getHeader("User-Agent")).orElse("-"));
        request.setAttribute("span", baseSpan);
        //execute request
        filterChain.doFilter(request, response);
        if (request.isAsyncStarted()) {
            final String flowId = FlowIdUtils.peek();
            request.getAsyncContext().addListener(new AsyncRequestListener(request, response, flowId, baseSpan));
        }
    } finally {
        if (!request.isAsyncStarted()) {
            traceRequest(request.getContentLength(), response.getStatus(), baseSpan);
        }
        final Map<String, String> spanContextToInject = new HashMap<>();
        GlobalTracer.get().inject(baseSpan.context(),
                HTTP_HEADERS, new TextMapInjectAdapter(spanContextToInject));
        response.setHeader(SPAN_CONTEXT, spanContextToInject.toString());
        baseSpan.finish();
    }
}

io.opentracing.propagation.TextMapInjectAdapter Java Examples