mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-11-19 14:11:58 +00:00
92 lines
2.4 KiB
C++
92 lines
2.4 KiB
C++
#include <memory>
|
|
|
|
#include "CurrentThread.h"
|
|
#include <common/logger_useful.h>
|
|
#include <Common/ThreadStatus.h>
|
|
#include <Interpreters/ProcessList.h>
|
|
#include <Interpreters/Context.h>
|
|
#include <Poco/Ext/ThreadNumber.h>
|
|
#include <Poco/Logger.h>
|
|
|
|
|
|
namespace DB
|
|
{
|
|
|
|
namespace ErrorCodes
|
|
{
|
|
extern const int LOGICAL_ERROR;
|
|
}
|
|
|
|
/// Order of current_thread and current_thread_scope matters
|
|
thread_local ThreadStatusPtr current_thread = ThreadStatus::create();
|
|
thread_local CurrentThread::ThreadScopePtr current_thread_scope = std::make_shared<CurrentThread::ThreadScope>();
|
|
|
|
void CurrentThread::updatePerformanceCounters()
|
|
{
|
|
get()->updatePerformanceCounters();
|
|
}
|
|
|
|
ThreadStatusPtr CurrentThread::get()
|
|
{
|
|
#ifndef NDEBUG
|
|
if (!current_thread || current_thread.use_count() <= 0)
|
|
throw Exception("Thread #" + std::to_string(Poco::ThreadNumber::get()) + " status was not initialized", ErrorCodes::LOGICAL_ERROR);
|
|
|
|
if (Poco::ThreadNumber::get() != current_thread->thread_number)
|
|
throw Exception("Current thread has different thread number", ErrorCodes::LOGICAL_ERROR);
|
|
#endif
|
|
|
|
return current_thread;
|
|
}
|
|
|
|
CurrentThread::ThreadScopePtr CurrentThread::getScope()
|
|
{
|
|
return current_thread_scope;
|
|
}
|
|
|
|
ProfileEvents::Counters & CurrentThread::getProfileEvents()
|
|
{
|
|
return current_thread->performance_counters;
|
|
}
|
|
|
|
MemoryTracker & CurrentThread::getMemoryTracker()
|
|
{
|
|
return current_thread->memory_tracker;
|
|
}
|
|
|
|
void CurrentThread::updateProgressIn(const Progress & value)
|
|
{
|
|
current_thread->progress_in.incrementPiecewiseAtomically(value);
|
|
}
|
|
|
|
void CurrentThread::updateProgressOut(const Progress & value)
|
|
{
|
|
current_thread->progress_out.incrementPiecewiseAtomically(value);
|
|
}
|
|
|
|
void CurrentThread::attachInternalTextLogsQueue(const std::shared_ptr<InternalTextLogsQueue> & logs_queue)
|
|
{
|
|
get()->attachInternalTextLogsQueue(logs_queue);
|
|
}
|
|
|
|
std::shared_ptr<InternalTextLogsQueue> CurrentThread::getInternalTextLogsQueue()
|
|
{
|
|
/// NOTE: this method could be called at early server startup stage
|
|
/// NOTE: this method could be called in ThreadStatus destructor, therefore we make use_count() check just in case
|
|
|
|
if (!current_thread || current_thread.use_count() <= 0)
|
|
return nullptr;
|
|
|
|
if (current_thread->getCurrentState() == ThreadStatus::ThreadState::Died)
|
|
return nullptr;
|
|
|
|
return current_thread->getInternalTextLogsQueue();
|
|
}
|
|
|
|
ThreadGroupStatusPtr CurrentThread::getGroup()
|
|
{
|
|
return get()->getThreadGroup();
|
|
}
|
|
|
|
}
|