From 3810861bb1328112772dfdd607a6abd2ca2f2c90 Mon Sep 17 00:00:00 2001
From: zhannngchen <48427519+zhannngchen@users.noreply.github.com>
Date: Mon, 17 Jun 2024 20:41:38 +0800
Subject: [PATCH] [branch-2.1](cherry-pick) add _pk_index_meta's size to
 Segment::_meta_mem_usage (#36329) (#36399)

cherry-pick #36329

add _pk_index_meta's size to Segment::_meta_mem_usage to make memory
estimation more accurate.
---
 be/src/olap/rowset/segment_v2/segment.cpp | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/be/src/olap/rowset/segment_v2/segment.cpp b/be/src/olap/rowset/segment_v2/segment.cpp
index e121639ac9..f2ec504c90 100644
--- a/be/src/olap/rowset/segment_v2/segment.cpp
+++ b/be/src/olap/rowset/segment_v2/segment.cpp
@@ -116,6 +116,9 @@ Status Segment::_open() {
 
     // An estimated memory usage of a segment
     _meta_mem_usage += _footer_pb->ByteSizeLong();
+    if (_pk_index_meta != nullptr) {
+        _meta_mem_usage += _pk_index_meta->ByteSizeLong();
+    }
     _meta_mem_usage += sizeof(*this);
     _meta_mem_usage += _tablet_schema->num_columns() * config::estimated_mem_per_column_reader;