2018-05-29 18:14:31 +00:00
|
|
|
#pragma once
|
2018-09-05 21:01:43 +00:00
|
|
|
|
2018-05-29 18:14:31 +00:00
|
|
|
#include <memory>
|
2018-06-15 13:45:19 +00:00
|
|
|
#include <string>
|
2018-05-29 18:14:31 +00:00
|
|
|
|
2018-09-05 21:01:43 +00:00
|
|
|
#include <Common/ThreadStatus.h>
|
2018-05-29 18:14:31 +00:00
|
|
|
|
|
|
|
namespace ProfileEvents
|
|
|
|
{
|
2018-05-31 15:54:08 +00:00
|
|
|
class Counters;
|
2018-05-29 18:14:31 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
class MemoryTracker;
|
|
|
|
|
|
|
|
|
|
|
|
namespace DB
|
|
|
|
{
|
|
|
|
|
2018-06-19 20:30:35 +00:00
|
|
|
class Context;
|
2018-05-29 18:14:31 +00:00
|
|
|
class QueryStatus;
|
2018-05-31 15:54:08 +00:00
|
|
|
struct Progress;
|
2018-06-15 17:32:35 +00:00
|
|
|
class InternalTextLogsQueue;
|
2018-05-29 18:14:31 +00:00
|
|
|
|
|
|
|
class CurrentThread
|
|
|
|
{
|
|
|
|
public:
|
|
|
|
|
2018-06-20 17:49:52 +00:00
|
|
|
/// Handler to current thread
|
2018-05-29 18:14:31 +00:00
|
|
|
static ThreadStatusPtr get();
|
2018-09-05 21:01:43 +00:00
|
|
|
|
2018-06-20 17:49:52 +00:00
|
|
|
/// Group to which belongs current thread
|
2018-06-19 20:30:35 +00:00
|
|
|
static ThreadGroupStatusPtr getGroup();
|
2018-05-29 18:14:31 +00:00
|
|
|
|
2018-06-06 20:57:07 +00:00
|
|
|
/// A logs queue used by TCPHandler to pass logs to a client
|
2018-06-19 20:30:35 +00:00
|
|
|
static void attachInternalTextLogsQueue(const std::shared_ptr<InternalTextLogsQueue> & logs_queue);
|
2018-06-15 17:32:35 +00:00
|
|
|
static std::shared_ptr<InternalTextLogsQueue> getInternalTextLogsQueue();
|
2018-06-06 20:57:07 +00:00
|
|
|
|
2018-06-20 17:49:52 +00:00
|
|
|
/// Makes system calls to update ProfileEvents that contain info from rusage and taskstats
|
|
|
|
static void updatePerformanceCounters();
|
|
|
|
|
|
|
|
static ProfileEvents::Counters & getProfileEvents();
|
|
|
|
static MemoryTracker & getMemoryTracker();
|
|
|
|
|
|
|
|
/// Update read and write rows (bytes) statistics (used in system.query_thread_log)
|
|
|
|
static void updateProgressIn(const Progress & value);
|
|
|
|
static void updateProgressOut(const Progress & value);
|
|
|
|
|
|
|
|
/// Query management:
|
|
|
|
|
|
|
|
/// Call from master thread as soon as possible (e.g. when thread accepted connection)
|
|
|
|
static void initializeQuery();
|
|
|
|
|
2018-06-19 20:30:35 +00:00
|
|
|
/// Sets query_context for current thread group
|
|
|
|
static void attachQueryContext(Context & query_context);
|
|
|
|
|
2018-06-20 17:49:52 +00:00
|
|
|
/// You must call one of these methods when create a query child thread:
|
2018-08-17 18:57:07 +00:00
|
|
|
/// Add current thread to a group associated with the thread group
|
2018-06-19 20:30:35 +00:00
|
|
|
static void attachTo(const ThreadGroupStatusPtr & thread_group);
|
2018-05-29 18:14:31 +00:00
|
|
|
/// Is useful for a ThreadPool tasks
|
2018-06-19 20:30:35 +00:00
|
|
|
static void attachToIfDetached(const ThreadGroupStatusPtr & thread_group);
|
2018-05-29 18:14:31 +00:00
|
|
|
|
2018-06-19 20:30:35 +00:00
|
|
|
/// Update ProfileEvents and dumps info to system.query_thread_log
|
|
|
|
static void finalizePerformanceCounters();
|
2018-06-01 19:39:32 +00:00
|
|
|
|
2018-06-15 13:45:19 +00:00
|
|
|
/// Returns a non-empty string if the thread is attached to a query
|
|
|
|
static std::string getCurrentQueryID();
|
|
|
|
|
2018-05-31 15:54:08 +00:00
|
|
|
/// Non-master threads call this method in destructor automatically
|
2018-05-29 18:14:31 +00:00
|
|
|
static void detachQuery();
|
2018-06-13 19:01:07 +00:00
|
|
|
static void detachQueryIfNotDetached();
|
2018-06-20 15:21:42 +00:00
|
|
|
|
|
|
|
/// Initializes query with current thread as master thread in constructor, and detaches it in desstructor
|
|
|
|
struct QueryScope
|
|
|
|
{
|
2018-06-20 17:49:52 +00:00
|
|
|
explicit QueryScope(Context & query_context);
|
2018-06-20 15:21:42 +00:00
|
|
|
~QueryScope();
|
|
|
|
};
|
2018-09-05 21:01:43 +00:00
|
|
|
|
|
|
|
public:
|
|
|
|
/// Implicitly finalizes current thread in the destructor
|
|
|
|
class ThreadScope
|
|
|
|
{
|
|
|
|
public:
|
|
|
|
void (*deleter)() = nullptr;
|
|
|
|
|
|
|
|
ThreadScope() = default;
|
|
|
|
~ThreadScope()
|
|
|
|
{
|
|
|
|
if (deleter)
|
|
|
|
deleter();
|
|
|
|
|
|
|
|
/// std::terminate on exception: this is Ok.
|
|
|
|
}
|
|
|
|
};
|
|
|
|
|
|
|
|
using ThreadScopePtr = std::shared_ptr<ThreadScope>;
|
|
|
|
static ThreadScopePtr getScope();
|
|
|
|
|
|
|
|
private:
|
|
|
|
static void defaultThreadDeleter();
|
2018-05-29 18:14:31 +00:00
|
|
|
};
|
|
|
|
|
|
|
|
}
|
|
|
|
|