[Enhancement] Optimize performance of get physical partition from ola…

…p table (#47198) Signed-off-by: meegoo <[email protected]> (cherry picked from commit 70c621c)
StarRocks · Aug 8, 2024 · 00b2bbf · 00b2bbf
1 parent 9560005
commit 00b2bbf
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 0 deletions.
diff --git a/fe/fe-core/src/main/java/com/starrocks/backup/RestoreJob.java b/fe/fe-core/src/main/java/com/starrocks/backup/RestoreJob.java
@@ -1009,6 +1009,7 @@ public Partition resetPartitionForRestore(OlapTable localTbl, OlapTable remoteTb
                 physicalPartition.setIdForRestore(newPhysicalPartId);
                 physicalPartition.setParentId(newPartId);
                 remotePart.addSubPartition(physicalPartition);
+                remoteTbl.addPhysicalPartition(physicalPartition);
             }
             // save version info for creating replicas
             long visibleVersion = physicalPartition.getVisibleVersion();

diff --git a/fe/fe-core/src/main/java/com/starrocks/catalog/OlapTable.java b/fe/fe-core/src/main/java/com/starrocks/catalog/OlapTable.java
@@ -1291,6 +1291,10 @@ public void addPartition(Partition partition) {
         }
     }
 
+    public void addPhysicalPartition(PhysicalPartition physicalPartition) {
+        physicalPartitionIdToPartitionId.put(physicalPartition.getId(), physicalPartition.getParentId());
+    }
+
     // This is a private method.
     // Call public "dropPartitionAndReserveTablet" and "dropPartition"
     private void dropPartition(long dbId, String partitionName, boolean isForceDrop, boolean reserveTablets) {

diff --git a/fe/fe-core/src/main/java/com/starrocks/server/LocalMetastore.java b/fe/fe-core/src/main/java/com/starrocks/server/LocalMetastore.java
@@ -1623,6 +1623,7 @@ public void addSubPartitions(Database db, OlapTable table, Partition partition,
             for (PhysicalPartition subPartition : subPartitions) {
                 // add sub partition
                 partition.addSubPartition(subPartition);
+                olapTable.addPhysicalPartition(subPartition);
             }
 
             // add partition log
@@ -1641,6 +1642,7 @@ public void replayAddSubPartition(PhysicalPartitionPersistInfoV2 info) throws Dd
             Partition partition = olapTable.getPartition(info.getPartitionId());
             PhysicalPartition physicalPartition = info.getPhysicalPartition();
             partition.addSubPartition(physicalPartition);
+            olapTable.addPhysicalPartition(physicalPartition);
 
             if (!isCheckpointThread()) {
                 // add to inverted index