apache#5

panxiaojun233 · Dec 29, 2020 · b67e42b · b67e42b
1 parent 3f6c810
commit b67e42b
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/dubbo-rpc/dubbo-rpc-triple/src/main/java/org/apache/dubbo/rpc/protocol/tri/Marshaller.java b/dubbo-rpc/dubbo-rpc-triple/src/main/java/org/apache/dubbo/rpc/protocol/tri/Marshaller.java
@@ -16,6 +16,7 @@ public class Marshaller {
     public static Marshaller marshaller = new Marshaller();
 
     public Object unmarshaller(Class<?> requestClass, ByteBuf in) {
+
         final Parser<?> parser = ProtoUtil.getParser(requestClass);
         Object result = null;
         try {

diff --git a/dubbo-rpc/dubbo-rpc-triple/src/main/java/org/apache/dubbo/rpc/protocol/tri/ProtoUtil.java b/dubbo-rpc/dubbo-rpc-triple/src/main/java/org/apache/dubbo/rpc/protocol/tri/ProtoUtil.java
@@ -1,6 +1,7 @@
 package org.apache.dubbo.rpc.protocol.tri;
 
 import java.lang.reflect.InvocationTargetException;
+import java.util.concurrent.ConcurrentHashMap;
 
 import com.google.protobuf.Message;
 import com.google.protobuf.Parser;
@@ -11,6 +12,8 @@
  * guohaoice@gmail.com
  */
 public class ProtoUtil {
+
+    private static ConcurrentHashMap<Class<?>, Message> instCache = new ConcurrentHashMap<>();
     /**
      * parse proto from netty {@link ByteBuf}
      *
@@ -26,12 +29,16 @@ public class ProtoUtil {
     //}
 
     public static Message defaultInst(Class<?> clz) {
-        Message defaultInst;
+        Message defaultInst = instCache.get(clz);
+        if (defaultInst != null) {
+            return defaultInst;
+        }
         try {
             defaultInst = (Message) clz.getMethod("getDefaultInstance").invoke(null);
         } catch (IllegalAccessException | InvocationTargetException | NoSuchMethodException e) {
             throw new RuntimeException("Create default protobuf instance failed ", e);
         }
+        instCache.put(clz, defaultInst);
         return defaultInst;
     }