f

yiguolei · Sep 20, 2024 · a6333c8 · a6333c8
1 parent ca4895a
commit a6333c8
Showing 1 changed file with 8 additions and 8 deletions.
diff --git a/be/src/runtime/workload_group/workload_group.h b/be/src/runtime/workload_group/workload_group.h
@@ -135,18 +135,18 @@ class WorkloadGroup : public std::enable_shared_from_this<WorkloadGroup> {
                                 _spill_high_watermark.load(std::memory_order_relaxed) / 100));
     }
 
-    void update_load_mem_usage(int64_t active_bytes, int64_t write_bytes, int64_t flush_bytes) {
+    void update_load_mem_usage(int64_t active_bytes, int64_t queue_bytes, int64_t flush_bytes) {
         std::unique_lock<std::shared_mutex> wlock(_mutex);
-        _active_mem_usage = _active_mem_usage;
-        _queue_mem_usage = _queue_mem_usage;
-        _flush_mem_usage = _flush_mem_usage;
+        _active_mem_usage = active_bytes;
+        _queue_mem_usage = queue_bytes;
+        _flush_mem_usage = flush_bytes;
     }
 
-    void get_load_mem_usage(int64_t* active_bytes, int64_t* write_bytes, int64_t* flush_bytes) {
+    void get_load_mem_usage(int64_t* active_bytes, int64_t* queue_bytes, int64_t* flush_bytes) {
         std::shared_lock<std::shared_mutex> r_lock(_mutex);
-        *active_bytes += writer->active_memtable_mem_consumption();
-        *write_bytes += writer->mem_consumption(MemType::WRITE_FINISHED);
-        *flush_bytes += writer->mem_consumption(MemType::FLUSH);
+        *active_bytes += _active_mem_usage;
+        *queue_bytes += _queue_mem_usage;
+        *flush_bytes += _flush_mem_usage;
     }
 
     std::string debug_string() const;