feat(KryoUtils): Add new methods

GeorgeV220 · Jul 3, 2023 · 4a42228 · 4a42228
1 parent 25e9c99
commit 4a42228
Showing 1 changed file with 13 additions and 5 deletions.
diff --git a/utilities/src/main/java/com/georgev22/library/utilities/KryoUtils.java b/utilities/src/main/java/com/georgev22/library/utilities/KryoUtils.java
@@ -4,7 +4,6 @@
 import com.esotericsoftware.kryo.Serializer;
 import com.esotericsoftware.kryo.io.Input;
 import com.esotericsoftware.kryo.io.Output;
-import com.esotericsoftware.kryo.serializers.DefaultSerializers;
 import com.esotericsoftware.kryo.util.MapReferenceResolver;
 import com.esotericsoftware.kryo.util.ObjectMap;
 import com.georgev22.library.maps.*;
@@ -39,11 +38,19 @@ public static <T> T deserialize(byte[] bytes) {
     }
 
     public static void registerClass(Class<?> clazz) {
-        kryo.register(clazz);
+        kryo.register(clazz, getKryo().getDefaultSerializer(clazz));
     }
 
     public static void registerClass(Class<?> clazz, int id) {
-        kryo.register(clazz, id);
+        kryo.register(clazz, getKryo().getDefaultSerializer(clazz), id);
+    }
+
+    public static <T> void registerClass(Class<?> clazz, Serializer<T> serializer) {
+        kryo.register(clazz, serializer);
+    }
+
+    public static <T> void registerClass(Class<?> clazz, Serializer<T> serializer, int id) {
+        kryo.register(clazz, serializer, id);
     }
 
     public static <T> void setDefaultSerializer(Class<? extends Serializer<T>> serializerClass) {
@@ -108,9 +115,10 @@ public static <T> void setDefaultSerializer(Class<? extends Serializer<T>> seria
         kryo.register(UnmodifiableObjectMap.class);
         kryo.register(UnmodifiableObjectMap[].class);
 
+        return kryo;
+    }
 
-        kryo.setDefaultSerializer(DefaultSerializers.StringSerializer.class);
-
+    public static Kryo getKryo() {
         return kryo;
     }
 }