Added ability to provide customizations to Kryo via a consumer (#255)

orbit · Jun 15, 2017 · 6c2ba59 · 6c2ba59
1 parent a00fe88
commit 6c2ba59
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/actors/runtime/src/main/java/cloud/orbit/actors/runtime/KryoSerializer.java b/actors/runtime/src/main/java/cloud/orbit/actors/runtime/KryoSerializer.java
@@ -68,6 +68,7 @@
 import java.util.GregorianCalendar;
 import java.util.HashMap;
 import java.util.UUID;
+import java.util.function.Consumer;
 
 /**
  * Kryo based message serializer and object cloning implementation.
@@ -79,6 +80,10 @@ public class KryoSerializer implements ExecutionObjectCloner, MessageSerializer
     private final KryoPool kryoPool;
 
     public KryoSerializer() {
+        this(kryo -> {});
+    }
+
+    public KryoSerializer(Consumer<Kryo> kryoConsumer) {
         KryoFactory factory = new KryoFactory() {
             @Override
             public Kryo create() {
@@ -127,6 +132,8 @@ public Registration writeClass(Output output, Class type) {
 
                 kryo.register(ReferenceReplacement.class, new ReferenceReplacementSerializer());
 
+                kryoConsumer.accept(kryo);
+
                 return kryo;
             }
         };