Kryo doesn't work since bump up a newer version

coderplay · coderplay · commit ed98c8dbb331 · 2015-07-16T11:22:33.000-07:00
diff --git a/lightningrpc-benchmark/src/main/java/info/minzhou/lightning/rpc/benchmark/RequestObjectSerializer.java b/lightningrpc-benchmark/src/main/java/info/minzhou/lightning/rpc/benchmark/RequestObjectSerializer.java
@@ -4,35 +4,23 @@
 import com.esotericsoftware.kryo.Serializer;
 import com.esotericsoftware.kryo.io.Input;
 import com.esotericsoftware.kryo.io.Output;
+import com.esotericsoftware.kryo.serializers.DefaultArraySerializers;
 
 /**
  * RequestObject Serializer
  * 
  */
 public class RequestObjectSerializer extends Serializer<RequestObject> {
-	/**
-	 * @param kryo
-	 * @param output
-	 * @param reqObject
-	 */
-	@Override
-	public void write(Kryo kryo, Output output, RequestObject reqObject) {
-		byte[] content = reqObject.getBytes();
-		output.write(content);
-	}
+  private DefaultArraySerializers.ByteArraySerializer delegate = new DefaultArraySerializers.ByteArraySerializer();
 
-	/**
-	 * @param kryo
-	 * @param input
-	 * @param type
-	 * @return
-	 */
-	public RequestObject create(Kryo kryo, Input input, Class<RequestObject> type) {
-		return new RequestObject(input.getBuffer().length - 1);
-	}
+  @Override
+  public void write(Kryo kryo, Output output, RequestObject reqObject) {
+    delegate.write(kryo, output, reqObject.getBytes());
+  }
 
   @Override
   public RequestObject read(Kryo kryo, Input input, Class<RequestObject> type) {
-    return kryo.readObjectOrNull(input, type);
+    byte[] bytes = delegate.read(kryo, input, byte[].class);
+    return new RequestObject(bytes);
   }
 }
diff --git a/lightningrpc-benchmark/src/main/java/info/minzhou/lightning/rpc/benchmark/ResponseObject.java b/lightningrpc-benchmark/src/main/java/info/minzhou/lightning/rpc/benchmark/ResponseObject.java
@@ -15,6 +15,9 @@ public ResponseObject(int size){
 		bytes = new byte[size];
 	}
 
+  public ResponseObject(byte[] bytes){
+    this.bytes = bytes;
+  }
 	public byte[] getBytes() {
 		return bytes;
 	}
diff --git a/lightningrpc-benchmark/src/main/java/info/minzhou/lightning/rpc/benchmark/ResponseObjectSerializer.java b/lightningrpc-benchmark/src/main/java/info/minzhou/lightning/rpc/benchmark/ResponseObjectSerializer.java
@@ -4,36 +4,25 @@
 import com.esotericsoftware.kryo.Serializer;
 import com.esotericsoftware.kryo.io.Input;
 import com.esotericsoftware.kryo.io.Output;
+import com.esotericsoftware.kryo.serializers.DefaultArraySerializers;
 
 /**
  * ResponseObject Serializer
  * 
  */
 public class ResponseObjectSerializer extends Serializer<ResponseObject> {
-	/**
-	 * @param kryo
-	 * @param output
-	 * @param resObject
-	 */
-	@Override
-	public void write(Kryo kryo, Output output, ResponseObject resObject) {
-		byte[] content = resObject.getBytes();
-		output.write(content);
-	}
 
-	/**
-	 * @param kryo
-	 * @param input
-	 * @param type
-	 * @return
-	 */
-	public ResponseObject create(Kryo kryo, Input input, Class<ResponseObject> type) {
-		return new ResponseObject(input.getBuffer().length - 1);
+  private DefaultArraySerializers.ByteArraySerializer delegate = new DefaultArraySerializers.ByteArraySerializer();
+
+  @Override
+	public void write(Kryo kryo, Output output, ResponseObject resObject) {
+    delegate.write(kryo, output, resObject.getBytes());
 	}
 
   @Override
   public ResponseObject read(Kryo kryo, Input input,
       Class<ResponseObject> type) {
-    return kryo.readObjectOrNull(input, type);
+    byte[] bytes = delegate.read(kryo, input, byte[].class);
+    return new ResponseObject(bytes);
   }
 }
diff --git a/lightningrpc-common/src/main/java/info/minzhou/lightning/rpc/protocol/KryoEncoder.java b/lightningrpc-common/src/main/java/info/minzhou/lightning/rpc/protocol/KryoEncoder.java
@@ -14,7 +14,7 @@ public class KryoEncoder implements Encoder {
 	 */
 	@Override
 	public byte[] encode(Object object) throws Exception {
-		Output output = new Output(256);
+		Output output = new Output(256, Integer.MAX_VALUE);
 		KryoUtils.getKryo().writeClassAndObject(output, object);
 		return output.toBytes();
 	}

Original file line number	Diff line number	Diff line change
`@@ -15,6 +15,9 @@ public ResponseObject(int size){`
`15`	`15`	`bytes = new byte[size];`
`16`	`16`	`}`
`17`	`17`
	`18`	`+ public ResponseObject(byte[] bytes){`
	`19`	`+ this.bytes = bytes;`
	`20`	`+ }`
`18`	`21`	`public byte[] getBytes() {`
`19`	`22`	`return bytes;`
`20`	`23`	`}`
Original file line number	Diff line number	Diff line change
`@@ -14,7 +14,7 @@ public class KryoEncoder implements Encoder {`
`14`	`14`	`*/`
`15`	`15`	`@Override`
`16`	`16`	`public byte[] encode(Object object) throws Exception {`
`17`		`- Output output = new Output(256);`
	`17`	`+ Output output = new Output(256, Integer.MAX_VALUE);`
`18`	`18`	`KryoUtils.getKryo().writeClassAndObject(output, object);`
`19`	`19`	`return output.toBytes();`
`20`	`20`	`}`