com.akkaserverless.javasdk.impl.EntityExceptions.scala Maven / Gradle / Ivy

Go to download
/*
 * Copyright 2021 Lightbend Inc.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package com.akkaserverless.javasdk.impl

import com.akkaserverless.javasdk.{ eventsourcedentity, valueentity }
import com.akkaserverless.protocol.entity.Command
import com.akkaserverless.protocol.event_sourced_entity.EventSourcedInit
import com.akkaserverless.protocol.replicated_entity.ReplicatedEntityInit
import com.akkaserverless.protocol.value_entity.ValueEntityInit

object EntityExceptions {

  final case class EntityException(
      entityId: String,
      commandId: Long,
      commandName: String,
      message: String,
      cause: Option[Throwable])
      extends RuntimeException(message, cause.orNull) {
    def this(entityId: String, commandId: Long, commandName: String, message: String) =
      this(entityId, commandId, commandName, message, None)
  }

  object EntityException {
    def apply(message: String): EntityException =
      EntityException(entityId = "", commandId = 0, commandName = "", message, None)

    def apply(command: Command, message: String): EntityException =
      EntityException(command.entityId, command.id, command.name, message, None)

    def apply(command: Command, message: String, cause: Option[Throwable]): EntityException =
      EntityException(command.entityId, command.id, command.name, message, cause)

    def apply(context: valueentity.CommandContext, message: String): EntityException =
      EntityException(context.entityId, context.commandId, context.commandName, message, None)

    def apply(context: valueentity.CommandContext, message: String, cause: Option[Throwable]): EntityException =
      EntityException(context.entityId, context.commandId, context.commandName, message, cause)

    def apply(context: eventsourcedentity.CommandContext, message: String): EntityException =
      EntityException(context.entityId, context.commandId, context.commandName, message, None)

    def apply(context: eventsourcedentity.CommandContext, message: String, cause: Option[Throwable]): EntityException =
      EntityException(context.entityId, context.commandId, context.commandName, message, cause)
  }

  object ProtocolException {
    def apply(message: String): EntityException =
      EntityException(entityId = "", commandId = 0, commandName = "", "Protocol error: " + message, None)

    def apply(command: Command, message: String): EntityException =
      EntityException(command.entityId, command.id, command.name, "Protocol error: " + message, None)

    def apply(entityId: String, message: String): EntityException =
      EntityException(entityId, commandId = 0, commandName = "", "Protocol error: " + message, None)

    def apply(init: ValueEntityInit, message: String): EntityException =
      ProtocolException(init.entityId, message)

    def apply(init: EventSourcedInit, message: String): EntityException =
      ProtocolException(init.entityId, message)

    def apply(init: ReplicatedEntityInit, message: String): EntityException =
      ProtocolException(init.entityId, message)
  }

  def failureMessageForLog(cause: Throwable): String = cause match {
    case EntityException(entityId, commandId, commandName, _, _) =>
      val commandDescription = if (commandId != 0) s" for command [$commandName]" else ""
      val entityDescription = if (entityId.nonEmpty) s"entity [$entityId]" else "entity"
      s"Terminating $entityDescription due to unexpected failure$commandDescription"
    case _ => "Terminating entity due to unexpected failure"
  }
}