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 3ff15b47..b06fd248 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 @@ -231,6 +231,9 @@ public abstract class GeneratePbUtils { public static String buildMessage(PbGenerateOperation pbGenerateOperation, List protos, Proto proto, PbMessage pbMessage) { var builder = new StringBuilder(); + // package definition + builder.append(StringUtils.format("package {};", pbGenerateOperation.getJavaPackage())).append(LS).append(LS); + // import other class var imports = buildMessageImports(pbGenerateOperation, protos, proto, pbMessage); builder.append(imports);