Versions Compared

Key

  • This line was added.
  • This line was removed.
  • Formatting was changed.

...

Resetting the offsets of a consumer group is a relatively common operation. It's usually done to allow an application to reprocess data or otherwise to skip/drop records. Currently (thanks to KIP-122) it can be done using the kafka-consumer-group.sh tool however it would be nice to be able to to it directly with the AdminClient. Applications can more easily depend on the AdminClient API rather than running the ConsumerGroupsCommand class or worse reimplementing the logic. The ability to retrieve committed offsets is already available in the AdminClient and has proved very useful. Additionally KIP-496 is adding the ability to delete committed offsets. This justifies integrating the consumer group reset/alter functionality directly in the AdminClient.

...

Code Block
languagejava
titleAdminClient.java
    /**
     * <p>Resets<p>Alters offsets for the specified group.
     *
     * <p>This is a convenience method for {@link #resetConsumerGroupOffsets#alterConsumerGroupOffsets(String, Map, ResetOffsetsOptionsAlterOffsetsOptions)} with default options.
     * See the overload for more details.
     *
     * @param groupId The group for which to resetalter offsets.
     * @param offsets A map of offsets by partition with associated metadata.
     * @return The ResetOffsetsResultAlterOffsetsResult.
     */
    public ResetOffsetsResultAlterOffsetsResult resetConsumerGroupOffsetsalterConsumerGroupOffsets(String groupId, Map<TopicPartition, OffsetAndMetadata> offsets) {
        return resetOffsetsalterOffsets(groupId, offsets, new ResetOffsetsOptionsAlterOffsetsOptions());
    }

    /**
     * <p>Resets<p>Alters offsets for the specified group.
     *
     * <p>This operation is not transactional so it may succeed for some partitions while fail for others.
     *
     * @param groupId The group for which to resetalter offsets.
     * @param offsets A map of offsets by partition with associated metadata.
     * @param options The options to use when resettingaltering the offsets.
     * @return The ResetOffsetsResultAlterOffsetsResult.
     */
    public abstract ResetOffsetsResultAlterOffsetsResult resetConsumerGroupOffsetsalterConsumerGroupOffsets(String groupId, Map<TopicPartition, OffsetAndMetadata> offsets, ResetOffsetsOptionsAlterOffsetsOptions options);

    /**
     * <p>List offset for the specified partitions.
     *
     * <p>This is a convenience method for {@link #listOffsets(Collection, ListOffsetsOptions)}
     *
     * @param topicPartitionOffsets The mapping from partition to the OffsetSpec to look up.
     * @return The ListOffsetsResult.
     */
    public ListOffsetsResult listOffsets(Map<TopicPartition, OffsetSpec> topicPartitionOffsets) {
        return listOffsets(topicPartitionOffsets, new ListOffsetsOptions());
    }


    /**
     * <p>List offset for the specified partitions.
     *
     * @param topicPartitionOffsets The mapping from partition to the OffsetSpec to look up.
     * @param options The options to use when retrieving the offsets
     * @return The ListOffsetsResult.
     */
    public abstract ListOffsetsResult listOffsets(Map<TopicPartition, OffsetSpec> topicPartitionOffsets, ListOffsetsOptions options);

...

We will also add companion objects under org.apache.kafka.clients.admin for these methods:

  • ResetOffsetsResultAlterOffsetsResult:


Code Block
languagejava
titleCommitOffsetsResult.java
/**
 * The result of the {@link AdminClient#commitOffsetsAdminClient#alterOffsets(String, Map)} call.
 *
 * The API of this class is evolving, see {@link AdminClient} for details.
 */
@InterfaceStability.Evolving
public class ResetOffsetsResultAlterOffsetsResult {

    public ResettOffsetsResultAlterOffsetsResult(Map<TopicPartition, KafkaFuture<Void>> futures) {}

    public Map<TopicPartition, KafkaFuture<Void>> values() {}

    public KafkaFuture<Void> all() {}
}


  • AlterOffsetsOptionsCommitOffsetsOptions


    Code Block
    languagejava
    titleCommitOffsetsOptions.java
    /**
     * Options for the {@link AdminClient#resetOffsets(String, org.apache.kafka.clients.consumer.OffsetAndMetadata, ResetOffsetsOptionsAdminClient#alterOffsets() call.
     *
     * The API of this class is evolving, see {@link AdminClient} for details.
     */
    @InterfaceStability.Evolving
    public class ResetOffsetsOptionsAlterOffsetsOptions extends AbstractOptions<ResetOffsetsOptions>AbstractOptions<AlterOffsetsOptions> {
    }


  • ListOffsetsResult


    Code Block
    languagejava
    titleListOffsetsResult.java
    /**
     * The result of the {@link AdminClient#listOffsets(String)} call.
     *
     * The API of this class is evolving, see {@link AdminClient} for details.
     */
    @InterfaceStability.Evolving
    public class ListOffsetsResult {
    
        public ListOffsetsResult(Map<TopicPartition, KafkaFuture<ListOffsetsResultInfo>> futures) {}
    
        /**
         * Return a map from TopicPartition to futures which can be used to retrieve the offsets
         */
        public Map<TopicPartition, KafkaFuture<ListOffsetsResultInfo>> values() {}
    
        /**
         * Return a future which succeeds only if offsets for all specified partitions have been successfully
         * retrieved.
         */
        public KafkaFuture<Map<TopicPartition, ListOffsetsResultInfo>> all() {}
    
        static public class ListOffsetsResultInfo {
    
            ListOffsetsResultInfo(long offset, long timestamp, Optional<Integer> leaderEpoch) {}
    
            long offset() {}
            long timestamp() {}
            Optional<Integer> leaderEpoch() {}
        }
    }


  • ListOffsetsOptions



    Code Block
    languagejava
    titleListOffsetsOptions.java
    /**
     * Options for {@link AdminClient#listOffsets()}.
     *
     * The API of this class is evolving, see {@link AdminClient} for details.
     */
    @InterfaceStability.Evolving
    public class ListOffsetsOptions extends AbstractOptions<ListOffsetsOptions> {
    
        private final String isolationLevel;
    
        // isolationLevel default to read_uncommitted
        public ListOffsetsOptions() {}
        public ListOffsetsOptions(String isolationLevel) {}
    
        public String isolationLevel() {}
    }


  • OffsetSpec


    Code Block
    languagejava
    /**  
     * This class allows to specify the desired query when retrieving offsets
     */
    public class OffsetSpec {
    
        public OffsetSpec latest() {}
    
        public OffsetSpec earliest() {}
    
        public OffsetSpec forTimestamp(long timestamp) {}
    }


...

The proposal of this KIP is to add methods to reset alter offsets for a group in the AdminClient. Alongside, methods to retrieve log offsets (ListOffset API) so users can easily retrieve start and end offsets or offsets for a timestamp for partitions will be added.

The 4 new methods in AdminClient:

  • public ResettOffsetsResult resetConsumerGroupOffsetsAlterOffsetsResult alterConsumerGroupOffsets(String groupId, Map<TopicPartition, OffsetAndMetadata> offsets) {}
  • public abstract ResettOffsetsResult resetConsumerGroupOffsetsAlterOffsetsResult alterConsumerGroupOffsets(String groupId, Map<TopicPartition, OffsetAndMetadata> offsets, ResetOffsetsOptions AlterOffsetsOptions options);
  • public ListOffsetsResult listOffsets(Map<TopicPartition, OffsetSpec> topicPartitionOffsets) {}
  • public abstract ListOffsetsResult listOffsets(Map<TopicPartition, OffsetSpec> topicPartitionOffsets, ListOffsetsOptions options);

Regarding the implementation, instead of relying on KafkaConsumer to perform the resetalter, I propose to instead directly send OffsetCommit requests. This is principally for 2 reasons

  • Be more performant. In order to reset alter offsets, a KafkaConsumer has to be created for each group. It first needs to subscribe or be assigned topic/partitions before being able to reset alter offsets. Additionally we've need to pause consumption so calls to poll to perform the reset alter operation don't return records. That would require more connections/disconnections to the cluster, more network requests.
  • Avoid having the AdminClient depend on KafkaConsumer. Currently AdminClient does not depend on another client.

...