com.evolutiongaming.kafka.journal.PartitionOffset.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of kafka-journal_2.13 Show documentation

kafka-journal

The newest version!

package com.evolutiongaming.kafka.journal

import cats.{Eq, Order, Show}
import com.datastax.driver.core.{GettableByNameData, SettableData}
import com.evolutiongaming.kafka.journal.util.SkafkaHelper._
import com.evolutiongaming.scassandra.syntax._
import com.evolutiongaming.scassandra.{DecodeRow, EncodeRow}
import com.evolutiongaming.skafka.consumer.ConsumerRecord
import com.evolutiongaming.skafka.{Offset, Partition}

final case class PartitionOffset(
  partition: Partition = Partition.min,
  offset: Offset = Offset.min
) {
  override def toString = s"$partition:$offset"
}

object PartitionOffset {

  val empty: PartitionOffset = PartitionOffset()

  implicit val encodeRowPartitionOffset: EncodeRow[PartitionOffset] = new EncodeRow[PartitionOffset] {

    def apply[B <: SettableData[B]](data: B, value: PartitionOffset) = {
      data
        .encode("partition", value.partition)
        .encode("offset", value.offset)
    }
  }

  implicit val decodeRowPartitionOffset: DecodeRow[PartitionOffset] = (data: GettableByNameData) => {
    PartitionOffset(
      partition = data.decode[Partition]("partition"),
      offset = data.decode[Offset]("offset"))
  }


  implicit val eqPartitionOffset: Eq[PartitionOffset] = Eq.fromUniversalEquals

  implicit val showPartitionOffset: Show[PartitionOffset] = Show.fromToString[PartitionOffset]


  implicit val orderPartitionOffset: Order[PartitionOffset] = Order.whenEqual(
    Order.by { a: PartitionOffset => a.partition },
    Order.by { a: PartitionOffset => a.offset })


  def apply(record: ConsumerRecord[_, _]): PartitionOffset = {
    PartitionOffset(
      partition = record.topicPartition.partition,
      offset = record.offset)
  }
}