dev.responsive.kafka.internal.utils.ExceptionSupplier Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of kafka-client Show documentation

artifact for kafka-client

The newest version!

/*
 * Copyright 2024 Responsive Computing, Inc.
 *
 * This source code is licensed under the Responsive Business Source License Agreement v1.0
 * available at:
 *
 * https://www.responsive.dev/legal/responsive-bsl-10
 *
 * This software requires a valid Commercial License Key for production use. Trial and commercial
 * licenses can be obtained at https://www.responsive.dev
 */

package dev.responsive.kafka.internal.utils;

import dev.responsive.kafka.internal.config.ResponsiveStreamsConfig;
import java.util.Map;
import org.apache.kafka.clients.consumer.CommitFailedException;
import org.apache.kafka.common.errors.ProducerFencedException;
import org.apache.kafka.streams.StreamsConfig;

public class ExceptionSupplier {

  private final boolean eosEnabled;

  public ExceptionSupplier(final boolean eosEnabled) {
    this.eosEnabled = eosEnabled;
  }

  public static ExceptionSupplier fromConfig(final Map props) {
    final String processingGuarantee = ResponsiveStreamsConfig.streamsConfig(props)
        .getString(StreamsConfig.PROCESSING_GUARANTEE_CONFIG);

    // only eos-v2 is allowed in Responsive, so we can ignore eos-v1 here
    return new ExceptionSupplier(processingGuarantee.equals(StreamsConfig.EXACTLY_ONCE_V2));
  }

  public RuntimeException commitFencedException(final String message) {
    if (eosEnabled) {
      return new ProducerFencedException(message);
    } else {
      return new CommitFailedException(message);
    }
  }

  public boolean eosEnabled() {
    return eosEnabled;
  }
}