fix[pb]: the note of class generated by the protobuf protocol is inco…

…rrect
zfoo-project · Jan 23, 2024 · 59b15db · 59b15db
1 parent 9b325f0
commit 59b15db
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/protocol/src/main/java/com/zfoo/protocol/serializer/protobuf/GeneratePbUtils.java b/protocol/src/main/java/com/zfoo/protocol/serializer/protobuf/GeneratePbUtils.java
@@ -252,6 +252,10 @@ private static String buildMessageImports(PbGenerateOperation pbGenerateOperatio
         var imports = new HashSet<String>();
         imports.add(Protocol.class.getName());
 
+        if (CollectionUtils.isNotEmpty(pbMessage.getComments())) {
+            imports.add(Note.class.getName());
+        }
+
         var pbFields = pbMessage.getFields();
         for (var pbField : pbFields) {
             if (CollectionUtils.isNotEmpty(pbField.getComments())) {