From 24a8e43277fc5345cc4cbbe5ce02f57c7d7942b2 Mon Sep 17 00:00:00 2001
From: To-om <toom@thehive-project.org>
Date: Fri, 11 Dec 2020 18:22:25 +0100
Subject: [PATCH] #1708 Register serializers

---
 .../src/main/resources/play/reference-overrides.conf  | 11 +++++++++++
 cortex/connector/src/main/resources/reference.conf    |  2 +-
 thehive/conf/play/reference-overrides.conf            |  4 ++++
 3 files changed, 16 insertions(+), 1 deletion(-)
 create mode 100644 cortex/connector/src/main/resources/play/reference-overrides.conf

diff --git a/cortex/connector/src/main/resources/play/reference-overrides.conf b/cortex/connector/src/main/resources/play/reference-overrides.conf
new file mode 100644
index 0000000000..346ac6866c
--- /dev/null
+++ b/cortex/connector/src/main/resources/play/reference-overrides.conf
@@ -0,0 +1,11 @@
+akka {
+  actor {
+    serializers {
+      cortex-schema-updater = "org.thp.thehive.connector.cortex.models.SchemaUpdaterSerializer"
+    }
+
+    serialization-bindings {
+      "org.thp.thehive.connector.cortex.models.SchemaUpdaterMessage" = cortex-schema-updater
+    }
+  }
+}
diff --git a/cortex/connector/src/main/resources/reference.conf b/cortex/connector/src/main/resources/reference.conf
index f604462606..c106915e85 100644
--- a/cortex/connector/src/main/resources/reference.conf
+++ b/cortex/connector/src/main/resources/reference.conf
@@ -22,4 +22,4 @@ cortex = {
   //    # HTTP client configuration (SSL and proxy)
   //    #  ws {}
   //  }]
-}
\ No newline at end of file
+}
diff --git a/thehive/conf/play/reference-overrides.conf b/thehive/conf/play/reference-overrides.conf
index 4581bd5806..01917c745a 100644
--- a/thehive/conf/play/reference-overrides.conf
+++ b/thehive/conf/play/reference-overrides.conf
@@ -22,10 +22,14 @@ akka.actor {
   serializers {
     stream = "org.thp.thehive.services.StreamSerializer"
     notification = "org.thp.thehive.services.notification.NotificationSerializer"
+    thehive-schema-updater = "org.thp.thehive.models.SchemaUpdaterSerializer"
+    flow = "org.thp.thehive.services.FlowSerializer"
   }
 
   serialization-bindings {
     "org.thp.thehive.services.StreamMessage" = stream
     "org.thp.thehive.services.notification.NotificationMessage" = notification
+    "org.thp.thehive.models.SchemaUpdaterMessage" = thehive-schema-updater
+    "org.thp.thehive.services.FlowMessage" = flow
   }
 }