com.twitter.finatra.kafka.interceptors.PublishTimeProducerInterceptor.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of finatra-kafka_2.12 Show documentation

finatra-kafka

The newest version!

package com.twitter.finatra.kafka.interceptors

import com.google.common.primitives.Longs
import com.twitter.finatra.kafka.interceptors.PublishTimeProducerInterceptor._
import com.twitter.util.Time
import java.util
import org.apache.kafka.clients.producer.{ProducerInterceptor, ProducerRecord, RecordMetadata}

object PublishTimeProducerInterceptor {
  val PublishTimeHeaderName = "publish_time"
}

/**
 * An interceptor that puts a header on each Kafka record indicating when the record was published.
 */
class PublishTimeProducerInterceptor extends ProducerInterceptor[Any, Any] {

  override def onSend(record: ProducerRecord[Any, Any]): ProducerRecord[Any, Any] = {
    record
      .headers()
      .add(PublishTimeHeaderName, Longs.toByteArray(Time.now.inMillis))
    record
  }

  override def onAcknowledgement(metadata: RecordMetadata, exception: Exception): Unit = {}

  override def close(): Unit = {}

  override def configure(configs: util.Map[String, _]): Unit = {}
}