From bff6ede94ee54c7a556bfedbab39c6aa243d5b11 Mon Sep 17 00:00:00 2001 From: xinghuayu007 <1450306854@qq.com> Date: Fri, 18 Jun 2021 11:40:00 +0800 Subject: [PATCH] add data size field for partition cache (#6026) Co-authored-by: wangxixu --- .../java/org/apache/doris/qe/cache/RowBatchBuilder.java | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/fe/fe-core/src/main/java/org/apache/doris/qe/cache/RowBatchBuilder.java b/fe/fe-core/src/main/java/org/apache/doris/qe/cache/RowBatchBuilder.java index c468bac4be..4f8b832dd6 100644 --- a/fe/fe-core/src/main/java/org/apache/doris/qe/cache/RowBatchBuilder.java +++ b/fe/fe-core/src/main/java/org/apache/doris/qe/cache/RowBatchBuilder.java @@ -167,13 +167,17 @@ public class RowBatchBuilder { Long key = entry.getKey(); PartitionRange.PartitionSingle partition = cachePartMap.get(key); partitionRowList = entry.getValue(); + int data_size = 0; + for (byte[] buf : partitionRowList) { + data_size += buf.length; + } updateRequest = updateRequest.toBuilder() .addValues(InternalService.PCacheValue.newBuilder() .setParam(InternalService.PCacheParam.newBuilder() .setPartitionKey(key) .setLastVersion(partition.getPartition().getVisibleVersion()) .setLastVersionTime(partition.getPartition().getVisibleVersionTime()) - .build()).addAllRows( + .build()).setDataSize(dataSize).addAllRows( partitionRowList.stream().map(row -> ByteString.copyFrom(row)) .collect(Collectors.toList()))).build(); }