com.itv.bucky.DeliveryUnmarshalHandler.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of bucky-core_2.13 Show documentation

com.itv

There is a newer version: 4.0.0-M1

package com.itv.bucky

import cats.Applicative
import com.itv.bucky.consume.{ConsumeAction, DeadLetter, Delivery}
import com.typesafe.scalalogging.StrictLogging

import scala.language.higherKinds

class DeliveryUnmarshalHandler[F[_], T, S](unmarshaller: DeliveryUnmarshaller[T])(handler: T => F[S], deserializationFailureAction: S)(
    implicit F: Applicative[F])
    extends (Delivery => F[S])
    with StrictLogging {
  override def apply(delivery: Delivery): F[S] = unmarshaller.unmarshal(delivery) match {
    case Right(message) => handler(message)
    case Left(throwable) =>
      F.point {
        logger.error(s"Cannot deserialize: ${delivery.body} (will $deserializationFailureAction)", throwable)
        deserializationFailureAction
      }
  }
}

class UnmarshalFailureAction[F[_], T](unmarshaller: DeliveryUnmarshaller[T])(implicit F: Applicative[F]) extends StrictLogging {
  def apply(f: T => F[ConsumeAction])(delivery: Delivery): F[ConsumeAction] =
    unmarshaller.unmarshal(delivery) match {
      case Right(message) => f(message)
      case Left(throwable) =>
        F.point {
          logger.error(s"Cannot deserialize: ${delivery.body}", throwable)
          DeadLetter
        }
    }
}