perf[golang]: compatible field of inside protocol class

zfoo-project · Oct 15, 2023 · 034f4ac · 034f4ac
1 parent 8f9897c
commit 034f4ac
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/protocol/src/main/java/com/zfoo/protocol/serializer/go/GenerateGoUtils.java b/protocol/src/main/java/com/zfoo/protocol/serializer/go/GenerateGoUtils.java
@@ -192,7 +192,7 @@ private static String readObject(ProtocolRegistration registration) {
             if (field.isAnnotationPresent(Compatible.class)) {
                 goBuilder.append(TAB ).append("if buffer.CompatibleRead(beforeReadIndex, length) {").append(LS);
                 var compatibleReadObject = goSerializer(fieldRegistration.serializer()).readObject(goBuilder, 2, field, fieldRegistration);
-                goBuilder.append(TAB + TAB).append(StringUtils.format("packet.{} = {}", field.getName(), compatibleReadObject)).append(LS);
+                goBuilder.append(TAB + TAB).append(StringUtils.format("packet.{} = {}", StringUtils.capitalize(field.getName()), compatibleReadObject)).append(LS);
                 goBuilder.append(TAB).append("}").append(LS);
                 continue;
             }

diff --git a/protocol/src/main/resources/go/ByteBuffer.go b/protocol/src/main/resources/go/ByteBuffer.go
@@ -30,8 +30,7 @@ func (byteBuffer *ByteBuffer) AdjustPadding(predictionLength int, beforeWriteInd
 		byteBuffer.WriteInt(length)
 		byteBuffer.SetWriteOffset(currentWriteIndex)
 	} else {
-		byteBuffer.SetWriteOffset(currentWriteIndex - length)
-		var byteArray = byteBuffer.ReadUBytes(length)
+		var byteArray = byteBuffer.buffer[(currentWriteIndex - length):currentWriteIndex]
 		byteBuffer.SetWriteOffset(beforeWriteIndex)
 		byteBuffer.WriteInt(length)
 		byteBuffer.WriteUBytes(byteArray)