io.opentelemetry.instrumentation.kafka.internal.KafkaReceiveRequest Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of opentelemetry-kafka-clients-common Show documentation

Instrumentation of Java libraries using OpenTelemetry.

There is a newer version: 2.9.0-alpha

Show newest version

/*
 * Copyright The OpenTelemetry Authors
 * SPDX-License-Identifier: Apache-2.0
 */

package io.opentelemetry.instrumentation.kafka.internal;

import javax.annotation.Nullable;
import org.apache.kafka.clients.consumer.Consumer;
import org.apache.kafka.clients.consumer.ConsumerRecords;

/**
 * This class is internal and is hence not for public use. Its APIs are unstable and can change at
 * any time.
 */
public class KafkaReceiveRequest extends AbstractKafkaConsumerRequest {

  private final ConsumerRecords records;

  public static KafkaReceiveRequest create(
      ConsumerRecords records, @Nullable Consumer consumer) {
    return create(records, KafkaUtil.getConsumerGroup(consumer), KafkaUtil.getClientId(consumer));
  }

  public static KafkaReceiveRequest create(
      KafkaConsumerContext consumerContext, ConsumerRecords records) {
    String consumerGroup = consumerContext != null ? consumerContext.getConsumerGroup() : null;
    String clientId = consumerContext != null ? consumerContext.getClientId() : null;
    return create(records, consumerGroup, clientId);
  }

  public static KafkaReceiveRequest create(
      ConsumerRecords records, String consumerGroup, String clientId) {
    return new KafkaReceiveRequest(records, consumerGroup, clientId);
  }

  private KafkaReceiveRequest(
      ConsumerRecords records, String consumerGroup, String clientId) {
    super(consumerGroup, clientId);
    this.records = records;
  }

  public ConsumerRecords getRecords() {
    return records;
  }
}