endless.runtime.akka.serializer.CommandSerializer.scala Maven / Gradle / Ivy

Go to download
package endless.runtime.akka.serializer

import akka.actor.typed.ActorRefResolver
import akka.actor.typed.scaladsl.adapter.ClassicActorSystemOps
import akka.serialization.{BaseSerializer, SerializerWithStringManifest}
import cats.syntax.show.*
import com.google.protobuf.ByteString
import endless.runtime.akka.data.Command
import endless.runtime.akka.serializer.CommandSerializer.ManifestKey

import java.io.NotSerializableException

/** Internal command carrier serializer, configured in reference.conf
  * @param system
  *   actor system
  */
class CommandSerializer(val system: akka.actor.ExtendedActorSystem)
    extends SerializerWithStringManifest
    with BaseSerializer {
  private implicit val actorRefResolver: ActorRefResolver = ActorRefResolver(system.toTyped)

  def manifest(o: AnyRef): String = o match {
    case _: Command => ManifestKey
    case _ =>
      throw new IllegalArgumentException(
        show"Can't serialize object of type ${o.getClass.getName} in [${getClass.getName}]"
      )
  }

  def toBinary(o: AnyRef): Array[Byte] = o match {
    case command: Command =>
      proto.command
        .Command(
          command.id,
          ByteString.copyFrom(command.payload),
          actorRefResolver.toSerializationFormat(command.replyTo)
        )
        .toByteArray
    case _ =>
      throw new IllegalArgumentException(s"Cannot serialize object of type [${o.getClass.getName}]")
  }

  def fromBinary(bytes: Array[Byte], manifest: String): AnyRef = manifest match {
    case ManifestKey =>
      val protoCommand = proto.command.Command.parseFrom(bytes)
      Command(protoCommand.id, protoCommand.payload.toByteArray)(
        actorRefResolver.resolveActorRef(protoCommand.replyTo)
      )
    case _ =>
      throw new NotSerializableException(
        s"Unimplemented deserialization of message with manifest [$manifest] in [${getClass.getName}]"
      )
  }
}

object CommandSerializer {
  val ManifestKey: String = "Command"
}