[Fix](autobucket) Fix autobucket partition size by using getAllDataSi…

…ze including cooldown size (#23557)
apache · Aug 28, 2023 · f70638e · f70638e
1 parent 6e82178
commit f70638e
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/fe/fe-core/src/main/java/org/apache/doris/catalog/Partition.java b/fe/fe-core/src/main/java/org/apache/doris/catalog/Partition.java
@@ -253,6 +253,11 @@ public List<MaterializedIndex> getMaterializedIndices(IndexExtState extState) {
         return indices;
     }
 
+    public long getAllDataSize() {
+        return getDataSize() + getRemoteDataSize();
+    }
+
+    // this is local data size
     public long getDataSize() {
         long dataSize = 0;
         for (MaterializedIndex mIndex : getMaterializedIndices(IndexExtState.VISIBLE)) {

diff --git a/fe/fe-core/src/main/java/org/apache/doris/clone/DynamicPartitionScheduler.java b/fe/fe-core/src/main/java/org/apache/doris/clone/DynamicPartitionScheduler.java
@@ -210,7 +210,7 @@ private static int getBucketsNum(DynamicPartitionProperty property, OlapTable ta
         ArrayList<Long> partitionSizeArray = Lists.newArrayList();
         for (Partition partition : partitions) {
             if (partition.getVisibleVersion() >= 2) {
-                partitionSizeArray.add(partition.getDataSize());
+                partitionSizeArray.add(partition.getAllDataSize());
             }
         }