From 59b15dba6d200a5b0d7caaa37e0249ef6eab130e Mon Sep 17 00:00:00 2001
From: godotg <jaysunxiao@gmail.com>
Date: Tue, 23 Jan 2024 16:57:55 +0800
Subject: [PATCH] fix[pb]: the note of class generated by the protobuf protocol
 is incorrect

---
 .../zfoo/protocol/serializer/protobuf/GeneratePbUtils.java    | 4 ++++
 1 file changed, 4 insertions(+)

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
index 9d28380b..45806397 100644
--- 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 @@ public abstract class GeneratePbUtils {
         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())) {