override protobuf serializer copy method (#314)

twitter · Aug 11, 2018 · 5515ba8 · 5515ba8
1 parent 84ab358
commit 5515ba8
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/chill-protobuf/src/main/java/com/twitter/chill/protobuf/ProtobufSerializer.java b/chill-protobuf/src/main/java/com/twitter/chill/protobuf/ProtobufSerializer.java
@@ -76,5 +76,16 @@ public Message read(Kryo kryo, Input input, Class<Message> pbClass) {
       throw new RuntimeException("Could not create " + pbClass, e);
     }
   }
+
+  @Override
+  public Message copy(Kryo kryo, Message original) {
+    try {
+      byte[] bytes = original.toByteArray();
+      return (Message)getParse(original.getClass()).invoke(null, bytes);
+    } catch (Exception exception) {
+      throw new RuntimeException("Copy message error.", exception);
+    }
+  }
+
 }