diff --git a/thehive/app/org/thp/thehive/services/IntegrityCheckActor.scala b/thehive/app/org/thp/thehive/services/IntegrityCheckActor.scala index 401dbf8dcc..e864d75cd2 100644 --- a/thehive/app/org/thp/thehive/services/IntegrityCheckActor.scala +++ b/thehive/app/org/thp/thehive/services/IntegrityCheckActor.scala @@ -20,10 +20,10 @@ import scala.util.Success sealed trait IntegrityCheckMessage case class EntityAdded(name: String) extends IntegrityCheckMessage +case class NeedCheck(name: String) extends IntegrityCheckMessage +case class Check(name: String) extends IntegrityCheckMessage class IntegrityCheckActor() extends Actor { - case class NeedCheck(name: String) - case class Check(name: String) lazy val logger: Logger = Logger(getClass) lazy val injector: Injector = GuiceAkkaExtension(context.system).injector diff --git a/thehive/app/org/thp/thehive/services/IntegrityCheckSerializer.scala b/thehive/app/org/thp/thehive/services/IntegrityCheckSerializer.scala index 9375fc7fa7..4ab8dc9650 100644 --- a/thehive/app/org/thp/thehive/services/IntegrityCheckSerializer.scala +++ b/thehive/app/org/thp/thehive/services/IntegrityCheckSerializer.scala @@ -12,12 +12,16 @@ class IntegrityCheckSerializer extends Serializer { override def toBinary(o: AnyRef): Array[Byte] = o match { case EntityAdded(name) => 0.toByte +: name.getBytes + case NeedCheck(name) => 1.toByte +: name.getBytes + case Check(name) => 2.toByte +: name.getBytes case _ => throw new NotSerializableException } override def fromBinary(bytes: Array[Byte], manifest: Option[Class[_]]): AnyRef = bytes(0) match { case 0 => EntityAdded(new String(bytes.tail)) + case 1 => NeedCheck(new String(bytes.tail)) + case 2 => Check(new String(bytes.tail)) case _ => throw new NotSerializableException } } diff --git a/thehive/conf/play/reference-overrides.conf b/thehive/conf/play/reference-overrides.conf index 01917c745a..afb748cb43 100644 --- a/thehive/conf/play/reference-overrides.conf +++ b/thehive/conf/play/reference-overrides.conf @@ -24,6 +24,7 @@ akka.actor { notification = "org.thp.thehive.services.notification.NotificationSerializer" thehive-schema-updater = "org.thp.thehive.models.SchemaUpdaterSerializer" flow = "org.thp.thehive.services.FlowSerializer" + integrity = "org.thp.thehive.services.IntegrityCheckSerializer" } serialization-bindings { @@ -31,5 +32,6 @@ akka.actor { "org.thp.thehive.services.notification.NotificationMessage" = notification "org.thp.thehive.models.SchemaUpdaterMessage" = thehive-schema-updater "org.thp.thehive.services.FlowMessage" = flow + "org.thp.thehive.services.IntegrityCheckMessage" = integrity } }