Merge pull request #27 from TangMinXuan/master

[add] Add support for protostuff

pom.xml

@@ -23,6 +23,8 @@
         <junit.platform.version>1.5.2</junit.platform.version>
         <!--logging-->
         <slf4j.version>1.7.25</slf4j.version>
+        <!-- protostuff -->
+        <protostuff.version>1.7.2</protostuff.version>
     </properties>
     <modules>
         <module>rpc-framework-simple</module>

rpc-framework-simple/pom.xml

@@ -41,5 +41,16 @@
             <artifactId>spring-context</artifactId>
             <version>${spring.version}</version>
         </dependency>
+        <!-- protostuff -->
+        <dependency>
+            <groupId>io.protostuff</groupId>
+            <artifactId>protostuff-core</artifactId>
+            <version>${protostuff.version}</version>
+        </dependency>
+        <dependency>
+            <groupId>io.protostuff</groupId>
+            <artifactId>protostuff-runtime</artifactId>
+            <version>${protostuff.version}</version>
+        </dependency>
     </dependencies>
 </project>

rpc-framework-simple/src/main/java/github/javaguide/serialize/protostuff/ProtostuffSerializer.java

+package github.javaguide.serialize.protostuff;
+
+import github.javaguide.remoting.dto.RpcMessage;
+import github.javaguide.serialize.Serializer;
+import io.protostuff.LinkedBuffer;
+import io.protostuff.ProtostuffIOUtil;
+import io.protostuff.Schema;
+import io.protostuff.runtime.RuntimeSchema;
+
+/**
+ * @author TangMinXuan
+ * @createTime 2020年11月09日 20:13
+ */
+public class ProtostuffSerializer implements Serializer {
+
+    /**
+     * Avoid re applying buffer space every time serialization
+     */
+    private static LinkedBuffer buffer = LinkedBuffer.allocate(LinkedBuffer.DEFAULT_BUFFER_SIZE);
+
+    @SuppressWarnings("unchecked")
+    @Override
+    public byte[] serialize(Object obj) {
+        Class<?> clazz = obj.getClass();
+        Schema schema = RuntimeSchema.getSchema(clazz);
+        byte[] bytes;
+        try {
+            bytes = ProtostuffIOUtil.toByteArray(obj, schema, buffer);
+        } finally {
+            buffer.clear();
+        }
+        return bytes;
+    }
+
+    @Override
+    public <T> T deserialize(byte[] bytes, Class<T> clazz) {
+        Schema<T> schema = RuntimeSchema.getSchema(clazz);
+        T obj = schema.newMessage();
+        ProtostuffIOUtil.mergeFrom(bytes, obj, schema);
+        return obj;
+    }
+}

rpc-framework-simple/src/main/resources/META-INF/extensions/github.javaguide.serialize.Serializer

 kyro=github.javaguide.serialize.kyro.KryoSerializer
+protostuff=github.javaguide.serialize.protostuff.ProtostuffSerializer
\ No newline at end of file