Fix allocate size by add 9 bytes

apache · Jul 22, 2024 · 5da4167 · 5da4167
1 parent b5cac30
commit 5da4167
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/...lient/src/main/java/org/apache/uniffle/client/impl/grpc/ShuffleServerGrpcNettyClient.java b/...lient/src/main/java/org/apache/uniffle/client/impl/grpc/ShuffleServerGrpcNettyClient.java
@@ -166,7 +166,15 @@ public RssSendShuffleDataResponse sendShuffleData(RssSendShuffleDataRequest requ
  0L,
  stb.getValue(),
  System.currentTimeMillis());
- int allocateSize = size + sendShuffleDataRequest.encodedLength();
+ // headerEncodedLength = messageEncodedLength + messageTypeEncodedLength + bodyLength +
+ // messageEncodedLength
+ // {@link org.apache.uniffle.common.netty.MessageEncoder#encode}
+ int headerEncodedLength =
+ Integer.BYTES
+ + sendShuffleDataRequest.type().encodedLength()
+ + Integer.BYTES
+ + sendShuffleDataRequest.encodedLength();
+ int allocateSize = headerEncodedLength + size;
  int finalBlockNum = blockNum;
  try {
  RetryUtils.retryWithCondition(