From c782be6406468911719baea78f1134e4be343b42 Mon Sep 17 00:00:00 2001
From: Xinyi Zou <zouxinyi02@gmail.com>
Date: Mon, 27 Feb 2023 16:23:50 +0800
Subject: [PATCH] 1

---
 be/src/olap/push_handler.cpp | 6 ++++--
 be/src/olap/push_handler.h   | 1 +
 2 files changed, 5 insertions(+), 2 deletions(-)

diff --git a/be/src/olap/push_handler.cpp b/be/src/olap/push_handler.cpp
index b174bbb6ea8a45..78c081056c80fe 100644
--- a/be/src/olap/push_handler.cpp
+++ b/be/src/olap/push_handler.cpp
@@ -819,8 +819,10 @@ Status PushBrokerReader::init(const Schema* schema, const TBrokerScanRange& t_sc
     }
     _runtime_profile = _runtime_state->runtime_profile();
     _runtime_profile->set_name("PushBrokerReader");
-    _mem_pool.reset(new MemPool());
-    _tuple_buffer_pool.reset(new MemPool());
+    _mem_tracker = std::make_unique<MemTracker>(
+            fmt::format("PushBrokerReader#InstanceId={}", print_id(params.fragment_instance_id)));
+    _mem_pool.reset(new MemPool(_mem_tracker.get()));
+    _tuple_buffer_pool.reset(new MemPool(_mem_tracker.get()));
 
     _counter.reset(new ScannerCounter());
 
diff --git a/be/src/olap/push_handler.h b/be/src/olap/push_handler.h
index f24d8376515f7b..4330aba9e5ef62 100644
--- a/be/src/olap/push_handler.h
+++ b/be/src/olap/push_handler.h
@@ -205,6 +205,7 @@ class PushBrokerReader {
     const Schema* _schema;
     std::unique_ptr<RuntimeState> _runtime_state;
     RuntimeProfile* _runtime_profile;
+    std::unique_ptr<MemTracker> _mem_tracker;
     std::unique_ptr<MemPool> _mem_pool;
     std::unique_ptr<MemPool> _tuple_buffer_pool;
     std::unique_ptr<ScannerCounter> _counter;