From 472550511e084d814c8a8d91e2aafa6756506271 Mon Sep 17 00:00:00 2001
From: Marcin Chrzanowski <marcin.j.chrzanowski@gmail.com>
Date: Sat, 11 Jan 2020 18:53:54 +0100
Subject: Register messages with serializer

---
 .../java/pl/edu/mimuw/cloudatlas/agent/modules/UDUPSerializer.java  | 6 ++++++
 1 file changed, 6 insertions(+)

(limited to 'src/main/java/pl/edu/mimuw/cloudatlas/agent/modules')

diff --git a/src/main/java/pl/edu/mimuw/cloudatlas/agent/modules/UDUPSerializer.java b/src/main/java/pl/edu/mimuw/cloudatlas/agent/modules/UDUPSerializer.java
index 0f7b99d..f858468 100644
--- a/src/main/java/pl/edu/mimuw/cloudatlas/agent/modules/UDUPSerializer.java
+++ b/src/main/java/pl/edu/mimuw/cloudatlas/agent/modules/UDUPSerializer.java
@@ -15,6 +15,7 @@ import java.net.UnknownHostException;
 import java.rmi.Remote;
 import java.util.ArrayList;
 import java.util.HashSet;
+import java.util.HashMap;
 import java.util.LinkedHashMap;
 
 /**
@@ -125,7 +126,11 @@ public class UDUPSerializer {
 
         // messages in chronological order so it's easier to keep track
         kryo.register(AgentMessage.class);
+        kryo.register(AttributesMessage.class);
         kryo.register(GetStateMessage.class);
+        kryo.register(HejkaMessage.class);
+        kryo.register(NoCoTamMessage.class);
+        kryo.register(QueryMessage.class);
         kryo.register(QurnikMessage.class);
         kryo.register(RemikMessage.class);
         kryo.register(RemoveZMIMessage.class);
@@ -150,6 +155,7 @@ public class UDUPSerializer {
         // other
         kryo.register(byte[].class);
         kryo.register(LinkedHashMap.class);
+        kryo.register(HashMap.class);
         kryo.register(ModuleType.class);
     }
 
-- 
cgit v1.2.3