Backport #73081 to 24.9: Fix possible overestimate memory tracking

This commit is contained in:
robot-clickhouse 2024-12-11 14:09:35 +00:00
parent a4f1a32c0b
commit 454695e67f

View File

@ -206,8 +206,6 @@ bool ThreadStatus::isQueryCanceled() const
ThreadStatus::~ThreadStatus()
{
flushUntrackedMemory();
/// It may cause segfault if query_context was destroyed, but was not detached
auto query_context_ptr = query_context.lock();
assert((!query_context_ptr && getQueryId().empty()) || (query_context_ptr && getQueryId() == query_context_ptr->getCurrentQueryId()));
@ -218,6 +216,9 @@ ThreadStatus::~ThreadStatus()
chassert(!check_current_thread_on_destruction || current_thread == this);
/// Flush untracked_memory **right before** switching the current_thread to avoid losing untracked_memory in deleter (detachFromGroup)
flushUntrackedMemory();
/// Only change current_thread if it's currently being used by this ThreadStatus
/// For example, PushingToViews chain creates and deletes ThreadStatus instances while running in the main query thread
if (current_thread == this)