2 files changed, 47 insertions, 11 deletions
diff --git a/shared/src/main/java/kz/ilotterytea/frogartha/domain/Identity.java b/shared/src/main/java/kz/ilotterytea/frogartha/domain/Identity.java
index 9a72268..03c0ef4 100644
--- a/shared/src/main/java/kz/ilotterytea/frogartha/domain/Identity.java
+++ b/shared/src/main/java/kz/ilotterytea/frogartha/domain/Identity.java
@@ -1,9 +1,15 @@
 package kz.ilotterytea.frogartha.domain;
 
-import java.io.Serializable;
+import com.github.czyzby.websocket.serialization.SerializationException;
+import com.github.czyzby.websocket.serialization.Transferable;
+import com.github.czyzby.websocket.serialization.impl.Deserializer;
+import com.github.czyzby.websocket.serialization.impl.Serializer;
 
-public class Identity implements Serializable {
-    private final String username;
+public class Identity implements Transferable<Identity> {
+    private String username;
+
+    public Identity() {
+    }
 
     public Identity(String username) {
         this.username = username;
@@ -27,4 +33,14 @@ public class Identity implements Serializable {
             "username='" + username + '\'' +
             '}';
     }
+
+    @Override
+    public void serialize(Serializer serializer) throws SerializationException {
+        serializer.serializeString(username);
+    }
+
+    @Override
+    public Identity deserialize(Deserializer deserializer) throws SerializationException {
+        return new Identity(deserializer.deserializeString());
+    }
 }
diff --git a/shared/src/main/java/kz/ilotterytea/frogartha/domain/server/Acknowledge.java b/shared/src/main/java/kz/ilotterytea/frogartha/domain/server/Acknowledge.java
index f82c1c9..8ca0f66 100644
--- a/shared/src/main/java/kz/ilotterytea/frogartha/domain/server/Acknowledge.java
+++ b/shared/src/main/java/kz/ilotterytea/frogartha/domain/server/Acknowledge.java
@@ -1,22 +1,42 @@
 package kz.ilotterytea.frogartha.domain.server;
 
-import java.io.Serializable;
+import com.github.czyzby.websocket.serialization.SerializationException;
+import com.github.czyzby.websocket.serialization.Transferable;
+import com.github.czyzby.websocket.serialization.impl.Deserializer;
+import com.github.czyzby.websocket.serialization.impl.Serializer;
 
-public class Acknowledge implements Serializable {
-    private final Object payload;
+public class Acknowledge implements Transferable<Acknowledge> {
+    public enum AcknowledgeCode {
+        IDENTIFIED
+    }
+
+    private AcknowledgeCode code;
 
-    public Acknowledge(Object payload) {
-        this.payload = payload;
+    public Acknowledge() {
     }
 
-    public Object getPayload() {
-        return payload;
+    public Acknowledge(AcknowledgeCode code) {
+        this.code = code;
+    }
+
+    public AcknowledgeCode getCode() {
+        return code;
     }
 
     @Override
     public String toString() {
         return "Acknowledge{" +
-            "payload=" + payload +
+            "code=" + code.toString() +
             '}';
     }
+
+    @Override
+    public void serialize(Serializer serializer) throws SerializationException {
+        serializer.serializeInt(code.ordinal());
+    }
+
+    @Override
+    public Acknowledge deserialize(Deserializer deserializer) throws SerializationException {
+        return new Acknowledge(AcknowledgeCode.values()[deserializer.deserializeInt()]);
+    }
 }