brave.kafka.clients.KafkaTracing Java Examples

Consumer<String, String> createTracingConsumer(String... topics) {
  if (topics.length == 0) {
    topics = new String[] {testName.getMethodName()};
  }
  KafkaConsumer<String, String> consumer = kafka.helper().createStringConsumer();
  List<TopicPartition> assignments = new ArrayList<>();
  for (String topic : topics) {
    assignments.add(new TopicPartition(topic, 0));
  }
  consumer.assign(assignments);
  return KafkaTracing.create(tracing).consumer(consumer);
}
 

public static void main(String[] args) {

		final var config = ConfigFactory.load();
		/* START TRACING INSTRUMENTATION */
		final var sender = URLConnectionSender.newBuilder()
				.endpoint(config.getString("zipkin.endpoint")).build();
		final var reporter = AsyncReporter.builder(sender).build();
		final var tracing = Tracing.newBuilder().localServiceName("hello-consumer")
				.sampler(Sampler.ALWAYS_SAMPLE).spanReporter(reporter).build();
		final var kafkaTracing = KafkaTracing.newBuilder(tracing)
				.remoteServiceName("kafka").build();
		/* END TRACING INSTRUMENTATION */

		final var consumerConfigs = new Properties();
		consumerConfigs.setProperty(ConsumerConfig.BOOTSTRAP_SERVERS_CONFIG,
				config.getString("kafka.bootstrap-servers"));
		consumerConfigs.setProperty(ConsumerConfig.KEY_DESERIALIZER_CLASS_CONFIG,
				StringDeserializer.class.getName());
		consumerConfigs.setProperty(ConsumerConfig.VALUE_DESERIALIZER_CLASS_CONFIG,
				StringDeserializer.class.getName());
		consumerConfigs.setProperty(ConsumerConfig.GROUP_ID_CONFIG, "hello-consumer");
		consumerConfigs.setProperty(ConsumerConfig.AUTO_OFFSET_RESET_CONFIG,
				OffsetResetStrategy.EARLIEST.name().toLowerCase());
		final var kafkaConsumer = new KafkaConsumer<String, String>(consumerConfigs);
		final var tracingConsumer = kafkaTracing.consumer(kafkaConsumer);

		tracingConsumer.subscribe(Collections.singletonList("hello"));

		while (!Thread.interrupted()) {
			var records = tracingConsumer.poll(Duration.ofMillis(Long.MAX_VALUE));
			for (var record : records) {
				brave.Span span = kafkaTracing.nextSpan(record).name("print-hello")
						.start();
				span.annotate("starting printing");
				out.println(String.format("Record: %s", record));
				span.annotate("printing finished");
				span.finish();
			}
		}
	}
 

@Bean
@ConditionalOnMissingBean
KafkaTracing kafkaTracing(MessagingTracing messagingTracing,
		SleuthMessagingProperties properties) {
	return KafkaTracing.newBuilder(messagingTracing)
			.remoteServiceName(
					properties.getMessaging().getKafka().getRemoteServiceName())
			.build();
}
 

MySleuthKafka1664Aspect(KafkaTracing kafkaTracing, Tracer tracer) {
	super(kafkaTracing, tracer);
}
 

TracingKafkaClientSupplier(KafkaTracing kafkaTracing) {
  this.kafkaTracing = kafkaTracing;
}
 

Builder(KafkaTracing kafkaTracing) {
  if (kafkaTracing == null) throw new NullPointerException("kafkaTracing == null");
  this.kafkaTracing = kafkaTracing;
}
 

/** @since 5.10 */
public static Builder newBuilder(MessagingTracing messagingTracing) {
  return new Builder(KafkaTracing.create(messagingTracing));
}
 

/** @since 5.10 */
public static Builder newBuilder(Tracing tracing) {
  return new Builder(KafkaTracing.create(tracing));
}
 

/** @since 5.9 */
public static KafkaStreamsTracing create(KafkaTracing kafkaTracing) {
  return new Builder(kafkaTracing).build();
}
 

/** @since 5.10 */
public static KafkaStreamsTracing create(MessagingTracing messagingTracing) {
  return new Builder(KafkaTracing.create(messagingTracing)).build();
}
 

public static KafkaStreamsTracing create(Tracing tracing) {
  return create(KafkaTracing.create(tracing));
}
 

MySleuthKafkaAspect(KafkaTracing kafkaTracing, Tracer tracer) {
	super(kafkaTracing, tracer);
}
 

@Bean
SleuthKafkaAspect sleuthKafkaAspect(KafkaTracing kafkaTracing, Tracer tracer) {
	return new MySleuthKafkaAspect(kafkaTracing, tracer);
}
 

@Bean
SleuthKafkaAspect sleuthKafkaAspect(KafkaTracing kafkaTracing, Tracer tracer) {
	return new MySleuthKafka1664Aspect(kafkaTracing, tracer);
}
 

MessageListenerMethodInterceptor(KafkaTracing kafkaTracing, Tracer tracer) {
	this.kafkaTracing = kafkaTracing;
	this.tracer = tracer;
}
 

SleuthKafkaAspect(KafkaTracing kafkaTracing, Tracer tracer) {
	this.kafkaTracing = kafkaTracing;
	this.tracer = tracer;
	this.recordMessageConverter = ReflectionUtils.findField(
			MessagingMessageListenerAdapter.class, "recordMessageConverter");
}
 

@Bean
// for tests
@ConditionalOnMissingBean
SleuthKafkaAspect sleuthKafkaAspect(KafkaTracing kafkaTracing, Tracer tracer) {
	return new SleuthKafkaAspect(kafkaTracing, tracer);
}
 

public static void main(String[] args) {

		final var config = ConfigFactory.load();
		final var kafkaBootstrapServers = config.getString("kafka.bootstrap-servers");
		final var schemaRegistryUrl = config.getString("schema-registry.url");

		/* START TRACING INSTRUMENTATION */
		final var sender = URLConnectionSender.newBuilder()
				.endpoint(config.getString("zipkin.endpoint")).build();
		final var reporter = AsyncReporter.builder(sender).build();
		final var tracing = Tracing.newBuilder()
				.localServiceName("twitter-console-consumer")
				.sampler(Sampler.ALWAYS_SAMPLE).spanReporter(reporter).build();
		final var kafkaTracing = KafkaTracing.newBuilder(tracing)
				.remoteServiceName("kafka").build();
		/* END TRACING INSTRUMENTATION */

		final var consumerConfigs = new Properties();
		consumerConfigs.setProperty(ConsumerConfig.BOOTSTRAP_SERVERS_CONFIG,
				kafkaBootstrapServers);
		consumerConfigs.setProperty(ConsumerConfig.KEY_DESERIALIZER_CLASS_CONFIG,
				StringDeserializer.class.getName());
		consumerConfigs.setProperty(ConsumerConfig.VALUE_DESERIALIZER_CLASS_CONFIG,
				SpecificAvroDeserializer.class.getName());
		consumerConfigs.put(AbstractKafkaAvroSerDeConfig.SCHEMA_REGISTRY_URL_CONFIG,
				schemaRegistryUrl);
		consumerConfigs.setProperty(ConsumerConfig.GROUP_ID_CONFIG, "twitter-console");
		final var kafkaConsumer = new KafkaConsumer<String, Tweet>(consumerConfigs);
		final var tracingConsumer = kafkaTracing.consumer(kafkaConsumer);

		tracingConsumer.subscribe(
				Collections.singletonList(config.getString("topics.input-tweets")));

		while (!Thread.interrupted()) {
			final var records = tracingConsumer.poll(Duration.ofMillis(Long.MAX_VALUE));
			for (var record : records) {
				var span = kafkaTracing.nextSpan(record).name("print-hello").start();
				try (var ignored = tracing.tracer().withSpanInScope(span)) {
					span.annotate("starting printing");
					out.println(String.format("Record: %s", record));
					span.annotate("printing finished");
				}
				catch (RuntimeException | Error e) {
					span.error(e);
					throw e;
				}
				finally {
					span.finish();
				}
			}
		}

	}
 

/**
 * Default constructor.
 * @param kafkaTracing The kafka tracing
 */
public BraveKafkaProducerTracingInstrumentation(KafkaTracing kafkaTracing) {
    this.kafkaTracing = kafkaTracing;
}
 

/**
 * Default constructor.
 * @param kafkaTracing The kafka tracing
 */
public BraveKafkaConsumerTracingInstrumentation(KafkaTracing kafkaTracing) {
    this.kafkaTracing = kafkaTracing;
}
 

/**
 * Creates a default {@link KafkaTracing} using {@link MessagingTracing} or {@link Tracing}.
 *
 * @param tracing The {@link Tracing}
 * @param messagingTracing The {@link MessagingTracing} if exists
 * @return a new instance of {@link KafkaTracing}
 */
@Requires(missingBeans = KafkaTracing.class)
@Singleton
KafkaTracing kafkaTracing(Tracing tracing, @Nullable MessagingTracing messagingTracing) {
  return messagingTracing == null ? KafkaTracing.create(tracing) : KafkaTracing.create(messagingTracing);
}