f

yiguolei · Oct 23, 2024 · 98fc985 · 98fc985
1 parent ed70529
commit 98fc985
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/be/src/runtime/workload_group/workload_group_manager.cpp b/be/src/runtime/workload_group/workload_group_manager.cpp
@@ -456,7 +456,7 @@ int64_t WorkloadGroupMgr::flush_memtable_from_current_group_(
     int64_t memtable_active_bytes = 0;
     int64_t memtable_queue_bytes = 0;
     int64_t memtable_flush_bytes = 0;
-    memtable_limiter->get_workload_group_memtable_usage(
+    memtable_limiter->get_workload_group_memtable_usage(wg->id(),
             &memtable_active_bytes, &memtable_queue_bytes, &memtable_flush_bytes);
     // TODO: should add a signal in memtable limiter to prevent new batch
     // For example, streamload, it will not reserve many memory, but it will occupy many memtable memory.

diff --git a/be/src/runtime/workload_group/workload_group_manager.h b/be/src/runtime/workload_group/workload_group_manager.h
@@ -108,6 +108,8 @@ class WorkloadGroupMgr {
 private:
     int64_t cancel_top_query_in_overcommit_group_(int64_t need_free_mem, int64_t lower_bound,
                                                   RuntimeProfile* profile);
+    int64_t flush_memtable_from_current_group_(std::shared_ptr<QueryContext> requestor,
+                                               WorkloadGroupPtr wg, int64_t need_free_mem);
     bool handle_single_query_(std::shared_ptr<QueryContext> query_ctx, size_t size_to_reserve,
                               Status paused_reason);
     int64_t revoke_memory_from_other_group_(std::shared_ptr<QueryContext> requestor,