Fix allocate size by add 9 bytes

apache · Jul 22, 2024 · ed0c1fa · ed0c1fa
1 parent b5cac30
commit ed0c1fa
Showing 1 changed file with 5 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,11 @@ 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(