• Home
• pl.touk.nussknacker
• nussknacker-flink-kafka-components-utils_2.13
• 1.18.0
• source code
• ConsumerGroupDeterminer.scala

×

Project download

Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $

You can buy this project and download/modify it how often you want.

pl.touk.nussknacker.engine.kafka.ConsumerGroupDeterminer.scala Maven / Gradle / Ivy

package pl.touk.nussknacker.engine.kafka

import pl.touk.nussknacker.engine.api.process.ProcessName
import pl.touk.nussknacker.engine.flink.api.process.FlinkCustomNodeContext

class ConsumerGroupDeterminer(consumerGroupNamingStrategy: ConsumerGroupNamingStrategy.Value) {

  def consumerGroup(nodeContext: FlinkCustomNodeContext): String = {
    consumerGroup(nodeContext.metaData.name, nodeContext.nodeId)
  }

  def consumerGroup(processName: ProcessName, nodeId: String): String = {
    consumerGroupNamingStrategy match {
      case ConsumerGroupNamingStrategy.ProcessId       => processName.value
      case ConsumerGroupNamingStrategy.ProcessIdNodeId => processName.value + "-" + nodeId
    }
  }

}

object ConsumerGroupDeterminer {

  def apply(config: KafkaConfig): ConsumerGroupDeterminer =
    new ConsumerGroupDeterminer(
      config.consumerGroupNamingStrategy.getOrElse(ConsumerGroupNamingStrategy.ProcessIdNodeId)
    )

}