5

apache · Sep 15, 2024 · 074c450 · 074c450
1 parent 3de0ae1
commit 074c450
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 4 deletions.
diff --git a/be/src/runtime/exec_env_init.cpp b/be/src/runtime/exec_env_init.cpp
@@ -600,7 +600,7 @@ void ExecEnv::init_mem_tracker() {
             MemTrackerLimiter::create_shared(MemTrackerLimiter::Type::GLOBAL, "PointQueryExecutor");
     _query_cache_mem_tracker =
             MemTrackerLimiter::create_shared(MemTrackerLimiter::Type::GLOBAL, "QueryCache");
-    _block_compression_mem_tracker = _block_compression_mem_tracker =
+    _block_compression_mem_tracker =
             MemTrackerLimiter::create_shared(MemTrackerLimiter::Type::GLOBAL, "BlockCompression");
     _rowid_storage_reader_tracker =
             MemTrackerLimiter::create_shared(MemTrackerLimiter::Type::GLOBAL, "RowIdStorageReader");

diff --git a/be/src/runtime/memory/lru_cache_policy.h b/be/src/runtime/memory/lru_cache_policy.h
@@ -117,7 +117,7 @@ class LRUCachePolicy : public CachePolicy {
         if (value != nullptr) {
             mem_tracker()->consume(tracking_bytes_with_handle);
             ((LRUCacheValueBase*)value)
-                    ->set_tracking_bytes(tracking_bytes_with_handle, _mem_tracker.get());
+                    ->set_tracking_bytes(tracking_bytes_with_handle, _mem_tracker);
         }
         return _cache->insert(key, value, charge, priority);
     }

diff --git a/be/src/runtime/memory/lru_cache_value_base.h b/be/src/runtime/memory/lru_cache_value_base.h
@@ -31,14 +31,15 @@ class LRUCacheValueBase {
         }
     }
 
-    void set_tracking_bytes(size_t tracking_bytes, MemTrackerLimiter* mem_tracker) {
+    void set_tracking_bytes(size_t tracking_bytes,
+                            const std::shared_ptr<MemTrackerLimiter>& mem_tracker) {
         this->_tracking_bytes = tracking_bytes;
         this->_mem_tracker = mem_tracker;
     }
 
 protected:
     size_t _tracking_bytes = 0;
-    MemTrackerLimiter* _mem_tracker = nullptr;
+    std::shared_ptr<MemTrackerLimiter> _mem_tracker;
 };
 
 } // namespace doris