com.ing.baker.runtime.serialization.ProtoMap.scala Maven / Gradle / Ivy

Go to download
package com.ing.baker.runtime.serialization

import com.ing.baker.runtime.akka.actor.protobuf
import com.ing.baker.runtime.common.BakerException
import com.ing.baker.runtime.scaladsl._
import com.ing.baker.runtime.serialization.protomappings._
import com.ing.baker.{il, types}
import scalapb.GeneratedMessageCompanion

import java.util.Base64
import scala.util.{Success, Try}

trait ProtoMap[A, P <: scalapb.GeneratedMessage] {

  def companion: GeneratedMessageCompanion[P]

  def toProto(a: A): P

  def fromProto(message: P): Try[A]

  def toByteArray(a: A): Array[Byte] =
    toProto(a).toByteArray

  def fromByteArray(binary: Array[Byte]): Try[A] =
    Try(companion.parseFrom(binary)).flatMap(fromProto)

}

object ProtoMap {

  def encode64[A, P <: scalapb.GeneratedMessage](a: A)(implicit ev: ProtoMap[A, P]): String =
    new String(Base64.getEncoder.encode(ev.toByteArray(a)))

  def decode64[A, P <: scalapb.GeneratedMessage](str: String)(implicit ev: ProtoMap[A, P]): Try[A] =
    ev.fromByteArray(Base64.getDecoder.decode(str))

  def ctxToProto[A, P <: scalapb.GeneratedMessage](a: A)(implicit ev: ProtoMap[A, P]): P = ev.toProto(a)

  def ctxFromProto[A, P <: scalapb.GeneratedMessage](proto: P)(implicit ev: ProtoMap[A, P]): Try[A] = ev.fromProto(proto)

  def versioned[A](a: Option[A], name: String): Try[A] =
    Try(a.getOrElse(throw new IllegalStateException(s"Missing field '$name' from protobuf message, probably we received a different version of the message")))

  def versionedOptional[A](a: Option[A], default: A): A = a.getOrElse(default)

  type AnyRefMapping = ProtoMap[AnyRef, protobuf.SerializedData]

  implicit def compiledRecipeMapping: ProtoMap[il.CompiledRecipe, protobuf.CompiledRecipe] =
    new CompiledRecipeMapping()

  implicit def recipeInformationMapping: ProtoMap[RecipeInformation, protobuf.RecipeInformation] =
    new RecipeInformationMapping()

  implicit def bakerEventMapping: ProtoMap[BakerEvent, protobuf.BakerEvent] =
    new BakerEventMapping()

  implicit val bakerExceptionMapping: ProtoMap[BakerException, protobuf.BakerException] =
    new BakerExceptionMapping

  implicit val eventDescriptorMapping: ProtoMap[il.EventDescriptor, protobuf.EventDescriptor] =
    new EventDescriptorMapping

  implicit val ingredientDescriptorMapping: ProtoMap[il.IngredientDescriptor, protobuf.IngredientDescriptor] =
    new IngredientDescriptorMapping

  implicit val ingredientInstanceMapping: ProtoMap[IngredientInstance, protobuf.Ingredient] =
    new IngredientInstanceMapping

  implicit val bakerTypeMapping: ProtoMap[types.Type, protobuf.Type] =
    new BakerTypesMapping

  implicit val bakerValueMapping: ProtoMap[types.Value, protobuf.Value] =
    new BakerValuesMapping

  implicit val interactionFailureStrategyMapping: ProtoMap[il.failurestrategy.InteractionFailureStrategy, protobuf.InteractionFailureStrategy] =
    new InteractionFailureStrategyMapping

  implicit val eventOutputTransformerMapping: ProtoMap[il.EventOutputTransformer, protobuf.EventOutputTransformer] =
    new EventOutputTransformerMapping

  implicit val runtimeEventMapping: ProtoMap[EventInstance, protobuf.RuntimeEvent] =
    new RuntimeEventMapping

  implicit val recipeInstanceMetadataMapping: ProtoMap[RecipeInstanceMetadata, protobuf.RecipeInstanceMetadata] =
    new RecipeInstanceMetadataMapping

  implicit val processStateMapping: ProtoMap[RecipeInstanceState, protobuf.ProcessState] =
    new ProcessStateMapping

  implicit val recipeEventMetadataMapping: ProtoMap[RecipeEventMetadata, protobuf.RecipeEventMetadata] =
    new RecipeEventMetadataMapping

  implicit val eventMomentMapping: ProtoMap[EventMoment, protobuf.EventMoment] =
    new EventMomentMapping

  implicit val processEventResult: ProtoMap[SensoryEventResult, protobuf.SensoryEventResult] =
    new SensoryEventResultMapping

  def identityProtoMap[A <: scalapb.GeneratedMessage](companion0: GeneratedMessageCompanion[A]): ProtoMap[A, A] =
    new ProtoMap[A, A] {

      override def companion: GeneratedMessageCompanion[A] = companion0

      override def toProto(a: A): A = a

      override def fromProto(message: A): Try[A] = Success(message)
    }
}