From 98fc98537b3938ccc8c0b39183aa3bb75c9bbe18 Mon Sep 17 00:00:00 2001 From: yiguolei Date: Wed, 23 Oct 2024 18:38:49 +0800 Subject: [PATCH] f --- be/src/runtime/workload_group/workload_group_manager.cpp | 2 +- be/src/runtime/workload_group/workload_group_manager.h | 2 ++ 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/be/src/runtime/workload_group/workload_group_manager.cpp b/be/src/runtime/workload_group/workload_group_manager.cpp index 80340e7cf7ecf1..a3dec1f7b76cd2 100644 --- 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 index 0f706af7839bb9..065528c66ecd3e 100644 --- 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 requestor, + WorkloadGroupPtr wg, int64_t need_free_mem); bool handle_single_query_(std::shared_ptr query_ctx, size_t size_to_reserve, Status paused_reason); int64_t revoke_memory_from_other_group_(std::shared_ptr requestor,