From 7818077686ee1fccba30fcb243e23cdfbd9098f7 Mon Sep 17 00:00:00 2001
From: nathan <nathan@dorkbox.com>
Date: Sun, 24 Sep 2017 16:47:41 +0200
Subject: [PATCH] SerializationManager register() methods now support chaining
 methods

---
 src/dorkbox/util/SerializationManager.java        | 15 ++++++++-------
 .../DefaultStorageSerializationManager.java       |  9 ++++++---
 2 files changed, 14 insertions(+), 10 deletions(-)

diff --git a/src/dorkbox/util/SerializationManager.java b/src/dorkbox/util/SerializationManager.java
index fefdf7f..e069522 100644
--- a/src/dorkbox/util/SerializationManager.java
+++ b/src/dorkbox/util/SerializationManager.java
@@ -15,14 +15,16 @@
  */
 package dorkbox.util;
 
+import java.io.IOException;
+
+import org.slf4j.Logger;
+
 import com.esotericsoftware.kryo.Kryo;
 import com.esotericsoftware.kryo.Serializer;
 import com.esotericsoftware.kryo.io.Input;
 import com.esotericsoftware.kryo.io.Output;
-import io.netty.buffer.ByteBuf;
-import org.slf4j.Logger;
 
-import java.io.IOException;
+import io.netty.buffer.ByteBuf;
 
 public
 interface SerializationManager {
@@ -38,7 +40,7 @@ interface SerializationManager {
      * order classes are registered is important when using this method. The
      * order must be the same at deserialization as it was for serialization.
      */
-    void register(Class<?> clazz);
+    SerializationManager register(Class<?> clazz);
 
     /**
      * Registers the class using the lowest, next available integer ID and the
@@ -50,8 +52,7 @@ interface SerializationManager {
      * order classes are registered is important when using this method. The
      * order must be the same at deserialization as it was for serialization.
      */
-    void register(Class<?> clazz, Serializer<?> serializer);
-
+    SerializationManager register(Class<?> clazz, Serializer<?> serializer);
 
     /**
      * Registers the class using the specified ID and serializer. If the ID is
@@ -65,7 +66,7 @@ interface SerializationManager {
      *           0-8 are used by default for primitive types and String, but
      *           these IDs can be repurposed.
      */
-    void register(Class<?> clazz, Serializer<?> serializer, int id);
+    SerializationManager register(Class<?> clazz, Serializer<?> serializer, int id);
 
     /**
      * Waits until a kryo is available to write, using CAS operations to prevent having to synchronize.
diff --git a/src/dorkbox/util/storage/DefaultStorageSerializationManager.java b/src/dorkbox/util/storage/DefaultStorageSerializationManager.java
index cecc2a0..1883810 100644
--- a/src/dorkbox/util/storage/DefaultStorageSerializationManager.java
+++ b/src/dorkbox/util/storage/DefaultStorageSerializationManager.java
@@ -36,20 +36,23 @@ class DefaultStorageSerializationManager implements SerializationManager {
 
     @Override
     public
-    void register(final Class<?> clazz) {
+    SerializationManager register(final Class<?> clazz) {
         kryo.register(clazz);
+        return this;
     }
 
     @Override
     public
-    void register(final Class<?> clazz, final Serializer<?> serializer) {
+    SerializationManager register(final Class<?> clazz, final Serializer<?> serializer) {
         kryo.register(clazz, serializer);
+        return this;
     }
 
     @Override
     public
-    void register(final Class<?> type, final Serializer<?> serializer, final int id) {
+    SerializationManager register(final Class<?> type, final Serializer<?> serializer, final int id) {
         kryo.register(type, serializer, id);
+        return this;
     }
 
     @Override