diff --git a/net/src/main/java/com/zfoo/net/handler/codec/jprotobuf/JProtobufTcpCodecHandler.java b/net/src/main/java/com/zfoo/net/handler/codec/jprotobuf/JProtobufTcpCodecHandler.java index 55a4bbcf..0b5ea5ad 100644 --- a/net/src/main/java/com/zfoo/net/handler/codec/jprotobuf/JProtobufTcpCodecHandler.java +++ b/net/src/main/java/com/zfoo/net/handler/codec/jprotobuf/JProtobufTcpCodecHandler.java @@ -111,23 +111,16 @@ public class JProtobufTcpCodecHandler extends ByteToMessageCodec) ProtobufProxy.create(packet.getClass()); + byte[] bytes = protobufCodec.encode(packet); + // header(4byte) + protocolId(2byte) + buffer.writeInt(bytes.length + 2); + var protocolId = packet.protocolId(); // 写入协议号 ByteBufUtils.writeShort(buffer, protocolId); - // 写入protobuf协议 - var protobufCodec = (Codec) ProtobufProxy.create(packet.getClass()); - byte[] bytes = protobufCodec.encode(packet); - buffer.writeBytes(bytes); - - int length = buffer.readableBytes(); - int packetLength = length - PacketService.PACKET_HEAD_LENGTH; - - buffer.writerIndex(0); - buffer.writeInt(packetLength); - buffer.writerIndex(length); } }