From 2cfef3e625fad682b0248dd0a1757b74486366bf Mon Sep 17 00:00:00 2001
From: sun <jaysunxiao@gmail.com>
Date: Fri, 20 Oct 2023 18:57:02 +0800
Subject: [PATCH] perf[protobuf]: protobuf generate

---
 .../protocol/serializer/protobuf/GenerateProtobufUtils.java  | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/protocol/src/main/java/com/zfoo/protocol/serializer/protobuf/GenerateProtobufUtils.java b/protocol/src/main/java/com/zfoo/protocol/serializer/protobuf/GenerateProtobufUtils.java
index 3cf94ec6..41fd0925 100644
--- a/protocol/src/main/java/com/zfoo/protocol/serializer/protobuf/GenerateProtobufUtils.java
+++ b/protocol/src/main/java/com/zfoo/protocol/serializer/protobuf/GenerateProtobufUtils.java
@@ -21,6 +21,7 @@ import com.zfoo.protocol.collection.CollectionUtils;
 import com.zfoo.protocol.exception.RunException;
 import com.zfoo.protocol.generate.GenerateOperation;
 import com.zfoo.protocol.generate.GenerateProtocolNote;
+import com.zfoo.protocol.generate.GenerateProtocolPath;
 import com.zfoo.protocol.model.Pair;
 import com.zfoo.protocol.registration.IProtocolRegistration;
 import com.zfoo.protocol.registration.ProtocolRegistration;
@@ -126,8 +127,8 @@ public abstract class GenerateProtobufUtils {
     public static void createProtocols(List<IProtocolRegistration> allGenerateProtocols) {
         var packageMap = new HashMap<String, List<IProtocolRegistration>>();
         for (var protocolRegistration : allGenerateProtocols) {
-            var clazz = protocolRegistration.protocolConstructor().getDeclaringClass();
-            String packageName = clazz.getPackageName();
+            var protocolId = protocolRegistration.protocolId();
+            String packageName = GenerateProtocolPath.getProtocolPath(protocolId);
             var packagList = packageMap.computeIfAbsent(packageName, k -> new ArrayList<>());
             packagList.add(protocolRegistration);
         }